coder
diff --git a/‎coderd/database/dbauthz/dbauthz.go
Lines changed: 21 additions & 0 deletions b/‎coderd/database/dbauthz/dbauthz.go
Lines changed: 21 additions & 0 deletions
diff --git a/‎coderd/database/dbmetrics/querymetrics.go
Lines changed: 21 additions & 0 deletions b/‎coderd/database/dbmetrics/querymetrics.go
Lines changed: 21 additions & 0 deletions
diff --git a/‎coderd/database/dbmock/dbmock.go
Lines changed: 45 additions & 0 deletions b/‎coderd/database/dbmock/dbmock.go
Lines changed: 45 additions & 0 deletions
diff --git a/‎coderd/database/querier.go
Lines changed: 3 additions & 0 deletions b/‎coderd/database/querier.go
Lines changed: 3 additions & 0 deletions
diff --git a/‎coderd/database/queries.sql.go
Lines changed: 196 additions & 0 deletions b/‎coderd/database/queries.sql.go
Lines changed: 196 additions & 0 deletions
diff --git a/‎coderd/database/queries/workspaceagents.sql
Lines changed: 14 additions & 0 deletions b/‎coderd/database/queries/workspaceagents.sql
Lines changed: 14 additions & 0 deletions
@@ -3433,6 +3433,13 @@ func (q *querier) GetWorkspaceAgentsByWorkspaceAndBuildNumber(ctx context.Contex
 	return q.db.GetWorkspaceAgentsByWorkspaceAndBuildNumber(ctx, arg)
 }
 
+func (q *querier) GetWorkspaceAgentsByWorkspaceIDAndBuildNumber(ctx context.Context, arg database.GetWorkspaceAgentsByWorkspaceIDAndBuildNumberParams) ([]database.WorkspaceAgent, error) {
+	if err := q.authorizeContext(ctx, policy.ActionRead, rbac.ResourceSystem); err != nil {
+		return nil, err
+	}
+	return q.db.GetWorkspaceAgentsByWorkspaceIDAndBuildNumber(ctx, arg)
+}
+
 func (q *querier) GetWorkspaceAgentsCreatedAfter(ctx context.Context, createdAt time.Time) ([]database.WorkspaceAgent, error) {
 	if err := q.authorizeContext(ctx, policy.ActionRead, rbac.ResourceSystem); err != nil {
 		return nil, err
@@ -3745,6 +3752,20 @@ func (q *querier) GetWorkspacesEligibleForTransition(ctx context.Context, now ti
 	return q.db.GetWorkspacesEligibleForTransition(ctx, now)
 }
 
+func (q *querier) GetWorkspacesForAgentMetrics(ctx context.Context, deleted bool) ([]database.GetWorkspacesForAgentMetricsRow, error) {
+	if err := q.authorizeContext(ctx, policy.ActionRead, rbac.ResourceSystem); err != nil {
+		return nil, err
+	}
+	return q.db.GetWorkspacesForAgentMetrics(ctx, deleted)
+}
+
+func (q *querier) GetWorkspacesForWorkspaceMetrics(ctx context.Context, deleted bool) ([]database.GetWorkspacesForWorkspaceMetricsRow, error) {
+	if err := q.authorizeContext(ctx, policy.ActionRead, rbac.ResourceSystem); err != nil {
+		return nil, err
+	}
+	return q.db.GetWorkspacesForWorkspaceMetrics(ctx, deleted)
+}
+
 func (q *querier) InsertAPIKey(ctx context.Context, arg database.InsertAPIKeyParams) (database.APIKey, error) {
 	// TODO(Cian): ideally this would be encoded in the policy, but system users are just members and we
 	// don't currently have a capability to conditionally deny creating resources by owner ID in a role.
 
@@ -365,3 +365,17 @@ WHERE
 	id = $1
 	AND parent_id IS NOT NULL
 	AND deleted = FALSE;
+
+-- name: GetWorkspaceAgentsByWorkspaceIDAndBuildNumber :many
+SELECT
+    workspace_agents.*
+FROM
+    workspace_agents
+JOIN
+    workspace_resources ON workspace_agents.resource_id = workspace_resources.id
+JOIN
+    workspace_builds ON workspace_resources.job_id = workspace_builds.job_id
+WHERE
+    workspace_builds.workspace_id = @workspace_id :: uuid
+    AND workspace_builds.build_number = @build_number :: integer
+    AND workspace_agents.deleted = FALSE;