Method scaletestCleanup

cli/exp_scaletest.go:606–805

()

}

func (r *RootCmd) scaletestCleanup() *serpent.Command {
	var template string
	cleanupStrategy := newScaletestCleanupStrategy()
	cmd := &serpent.Command{
		Use:   "cleanup",
		Short: "Cleanup scaletest workspaces, then cleanup scaletest users.",
		Long:  "The strategy flags will apply to each stage of the cleanup process.",
		Handler: func(inv *serpent.Invocation) error {
			client, err := r.InitClient(inv)
			if err != nil {
				return err
			}

			ctx := inv.Context()

			me, err := RequireAdmin(ctx, client)
			if err != nil {
				return err
			}

			client.HTTPClient = &http.Client{
				Transport: &codersdk.HeaderTransport{
					Transport: http.DefaultTransport,
					Header: map[string][]string{
						codersdk.BypassRatelimitHeader: {"true"},
					},
				},
			}

			if template != "" {
				_, err := parseTemplate(ctx, client, me.OrganizationIDs, template)
				if err != nil {
					return xerrors.Errorf("parse template: %w", err)
				}
			}

			cliui.Infof(inv.Stdout, "Pausing prebuilds reconciler...")
			setPrebuild := func(val bool) error {
				return client.PutPrebuildsSettings(ctx, codersdk.PrebuildsSettings{ReconciliationPaused: val})
			}
			if err = setPrebuild(true); err != nil {
				return xerrors.Errorf("pause prebuilds reconciler: %w", err)
			}
			defer func() {
				cliui.Infof(inv.Stdout, "Resuming prebuilds reconciler...")
				if resumeErr := setPrebuild(false); resumeErr != nil {
					cliui.Errorf(inv.Stderr, "Failed to resume prebuilds reconciler: %+v\n", resumeErr)
				}
			}()

			cliui.Infof(inv.Stdout, "Fetching scaletest prebuild workspaces...")
			prebuildWorkspaces, err := getScaletestPrebuildWorkspaces(ctx, client, template)
			if err != nil {
				return err
			}

			cliui.Errorf(inv.Stderr, "Found %d scaletest prebuild workspaces\n", len(prebuildWorkspaces))
			if len(prebuildWorkspaces) != 0 {