Skip to content

Commit ca8498b

Browse files
committed
add dbmem impl + tests
1 parent f62203c commit ca8498b

File tree

6 files changed

+608
-90
lines changed

6 files changed

+608
-90
lines changed

coderd/database/dbauthz/dbauthz.go

+3-3
Original file line numberDiff line numberDiff line change
@@ -1419,7 +1419,7 @@ func (q *querier) GetDeploymentWorkspaceAgentStats(ctx context.Context, createdA
14191419
}
14201420

14211421
func (q *querier) GetDeploymentWorkspaceAgentUsageStats(ctx context.Context, createdAt time.Time) (database.GetDeploymentWorkspaceAgentUsageStatsRow, error) {
1422-
panic("not implemented")
1422+
return q.db.GetDeploymentWorkspaceAgentUsageStats(ctx, createdAt)
14231423
}
14241424

14251425
func (q *querier) GetDeploymentWorkspaceStats(ctx context.Context) (database.GetDeploymentWorkspaceStatsRow, error) {
@@ -2365,11 +2365,11 @@ func (q *querier) GetWorkspaceAgentStatsAndLabels(ctx context.Context, createdAf
23652365
}
23662366

23672367
func (q *querier) GetWorkspaceAgentUsageStats(ctx context.Context, createdAt time.Time) ([]database.GetWorkspaceAgentUsageStatsRow, error) {
2368-
panic("not implemented")
2368+
return q.db.GetWorkspaceAgentUsageStats(ctx, createdAt)
23692369
}
23702370

23712371
func (q *querier) GetWorkspaceAgentUsageStatsAndLabels(ctx context.Context, createdAt time.Time) ([]database.GetWorkspaceAgentUsageStatsAndLabelsRow, error) {
2372-
panic("not implemented")
2372+
return q.db.GetWorkspaceAgentUsageStatsAndLabels(ctx, createdAt)
23732373
}
23742374

23752375
// GetWorkspaceAgentsByResourceIDs

coderd/database/dbgen/dbgen.go

+2
Original file line numberDiff line numberDiff line change
@@ -800,6 +800,7 @@ func WorkspaceAgentStat(t testing.TB, db database.Store, orig database.Workspace
800800
SessionCountReconnectingPTY: []int64{takeFirst(orig.SessionCountReconnectingPTY, 0)},
801801
SessionCountSSH: []int64{takeFirst(orig.SessionCountSSH, 0)},
802802
ConnectionMedianLatencyMS: []float64{takeFirst(orig.ConnectionMedianLatencyMS, 0)},
803+
Usage: []bool{takeFirst(orig.Usage, false)},
803804
}
804805
err := db.InsertWorkspaceAgentStats(genCtx, params)
805806
require.NoError(t, err, "insert workspace agent stat")
@@ -822,6 +823,7 @@ func WorkspaceAgentStat(t testing.TB, db database.Store, orig database.Workspace
822823
SessionCountJetBrains: params.SessionCountJetBrains[0],
823824
SessionCountReconnectingPTY: params.SessionCountReconnectingPTY[0],
824825
SessionCountSSH: params.SessionCountSSH[0],
826+
Usage: params.Usage[0],
825827
}
826828
}
827829

0 commit comments

Comments
 (0)