Function getScaletestPrebuildWorkspaces

cli/exp_scaletest.go:549–583

getScaletestPrebuildWorkspaces returns all prebuild workspaces that belong to scaletest templates. It uses getScaletestPrebuildsTemplates to scope the query so that legitimate (non-scaletest) prebuilds on the deployment are not caught in the cleanup. If template is non-empty only workspaces for that

(ctx context.Context, client *codersdk.Client, template string)

// caught in the cleanup. If template is non-empty only workspaces for that
// template are returned.
func getScaletestPrebuildWorkspaces(ctx context.Context, client *codersdk.Client, template string) ([]codersdk.Workspace, error) {
	const pageSize = 100

	templates, err := getScaletestPrebuildsTemplates(ctx, client, template)
	if err != nil {
		return nil, xerrors.Errorf("list scaletest prebuild templates: %w", err)
	}

	seen := make(map[uuid.UUID]struct{})
	var result []codersdk.Workspace

	for _, tmpl := range templates {
		for page := 0; ; page++ {
			resp, err := client.Workspaces(ctx, codersdk.WorkspaceFilter{
				Template: tmpl.Name,
				Offset:   page * pageSize,
				Limit:    pageSize,
			})
			if err != nil {
				return nil, xerrors.Errorf("list workspaces for template %q (page %d): %w", tmpl.Name, page, err)
			}
			for _, ws := range resp.Workspaces {
				if _, ok := seen[ws.ID]; !ok {
					seen[ws.ID] = struct{}{}
					result = append(result, ws)
				}
			}
			if len(resp.Workspaces) < pageSize {
				break
			}
		}
	}

	return result, nil
}

// getScaletestPrebuildsTemplates returns all templates created by the scaletest
// prebuilds runner (identified by prebuilds.TemplatePrefix). If template is