opensource/coder
3
0
Fork 0
mirror of https://github.com/coder/coder.git synced 2026-09-24 15:04:27 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat(coderd/telemetry): add telemetry for database Tasks (#20279)

Browse Source 
Adds Tasks to telemetry snapshots

Co-authored-by: Mathias Fredriksson <mafredri@gmail.com>
This commit is contained in:
Cian Johnston
2025-10-17 10:48:56 +01:00
committed by GitHub
co-authored by Mathias Fredriksson
parent 14e80022c9
commit dc6e50d6b7
5 changed files with 102 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
coderd/database/dbgen/dbgen.go
+7 -2
View File
@@ -1561,7 +1561,7 @@ func AIBridgeToolUsage(t testing.TB, db database.Store, seed database.InsertAIBr
return toolUsage
}
func Task(t testing.TB, db database.Store, orig database.TaskTable) database.TaskTable {
func Task(t testing.TB, db database.Store, orig database.TaskTable) database.Task {
t.Helper()
parameters := orig.TemplateParameters
@@ -1581,7 +1581,12 @@ func Task(t testing.TB, db database.Store, orig database.TaskTable) database.Tas
})
require.NoError(t, err, "failed to insert task")
return task
// Return the Task from the view instead of the TaskTable
fetched, err := db.GetTaskByID(genCtx, task.ID)
require.NoError(t, err, "failed to fetch task")
require.Equal(t, task.ID, fetched.ID)
return fetched
}
func TaskWorkspaceApp(t testing.TB, db database.Store, orig database.TaskWorkspaceApp) database.TaskWorkspaceApp {
coderd/database/querier_test.go
+2 -2
View File
@@ -6709,7 +6709,7 @@ func TestTasksWithStatusView(t *testing.T) {
buildTransition database.WorkspaceTransition,
agentState database.WorkspaceAgentLifecycleState,
appHealths []database.WorkspaceAppHealth,
) database.TaskTable {
) database.Task {
t.Helper()
template := dbgen.Template(t, db, database.Template{
@@ -7455,7 +7455,7 @@ func TestListTasks(t *testing.T) {
})
// Helper function to create a task
createTask := func(orgID, ownerID uuid.UUID) database.TaskTable {
createTask := func(orgID, ownerID uuid.UUID) database.Task {
ws := dbgen.Workspace(t, db, database.WorkspaceTable{
OrganizationID: orgID,
OwnerID: ownerID,