Skip to content

Commit d8568a3

Browse files
defelmnqdefelmnq
committed
work on fake db
1 parent de9e73a commit d8568a3

File tree

7 files changed

+58
-24
lines changed

7 files changed

+58
-24
lines changed

‎coderd/database/dbauthz/dbauthz.go

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1391,8 +1391,8 @@ func (q *querier) FavoriteWorkspace(ctx context.Context, id uuid.UUID) error {
13911391
return update(q.log, q.auth, fetch, q.db.FavoriteWorkspace)(ctx, id)
13921392
}
13931393

1394-
func (q *querier) FetchAgentResourceMonitorsByAgentID(ctx context.Context, agentID uuid.UUID) (database.WorkspaceAgentResourceMonitor, error) {
1395-
panic("not implemented")
1394+
func (q *querier) FetchAgentResourceMonitorsByAgentID(ctx context.Context, agentID uuid.UUID) ([]database.WorkspaceAgentResourceMonitor, error) {
1395+
return q.db.FetchAgentResourceMonitorsByAgentID(ctx, agentID)
13961396
}
13971397

13981398
func (q *querier) FetchNewMessageMetadata(ctx context.Context, arg database.FetchNewMessageMetadataParams) (database.FetchNewMessageMetadataRow, error) {
@@ -3219,7 +3219,7 @@ func (q *querier) InsertWorkspaceAgentMetadata(ctx context.Context, arg database
32193219
}
32203220

32213221
func (q *querier) InsertWorkspaceAgentResourceMonitor(ctx context.Context, arg database.InsertWorkspaceAgentResourceMonitorParams) (database.WorkspaceAgentResourceMonitor, error) {
3222-
panic("not implemented")
3222+
return q.db.InsertWorkspaceAgentResourceMonitor(ctx, arg)
32233223
}
32243224

32253225
func (q *querier) InsertWorkspaceAgentScriptTimings(ctx context.Context, arg database.InsertWorkspaceAgentScriptTimingsParams) (database.WorkspaceAgentScriptTiming, error) {

‎coderd/database/dbmem/dbmem.go

Lines changed: 21 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -227,6 +227,7 @@ type data struct {
227227
workspaceAgentScriptTimings []database.WorkspaceAgentScriptTiming
228228
workspaceAgentScripts []database.WorkspaceAgentScript
229229
workspaceAgentStats []database.WorkspaceAgentStat
230+
workspaceAgentResourceMonitors []database.WorkspaceAgentResourceMonitor
230231
workspaceApps []database.WorkspaceApp
231232
workspaceAppStatsLastInsertID int64
232233
workspaceAppStats []database.WorkspaceAppStat
@@ -2355,8 +2356,18 @@ func (q *FakeQuerier) FavoriteWorkspace(_ context.Context, arg uuid.UUID) error
23552356
return nil
23562357
}
23572358

2358-
func (q *FakeQuerier) FetchAgentResourceMonitorsByAgentID(ctx context.Context, agentID uuid.UUID) (database.WorkspaceAgentResourceMonitor, error) {
2359-
panic("not implemented")
2359+
func (q *FakeQuerier) FetchAgentResourceMonitorsByAgentID(ctx context.Context, agentID uuid.UUID) ([]database.WorkspaceAgentResourceMonitor, error) {
2360+
q.mutex.RLock()
2361+
defer q.mutex.RUnlock()
2362+
2363+
monitors := make([]database.WorkspaceAgentResourceMonitor, 0)
2364+
for _, monitor := range q.workspaceAgentResourceMonitors {
2365+
if monitor.AgentID == agentID {
2366+
monitors = append(monitors, monitor)
2367+
}
2368+
}
2369+
2370+
return monitors, nil
23602371
}
23612372

23622373
func (q *FakeQuerier) FetchNewMessageMetadata(_ context.Context, arg database.FetchNewMessageMetadataParams) (database.FetchNewMessageMetadataRow, error) {
@@ -8535,7 +8546,14 @@ func (q *FakeQuerier) InsertWorkspaceAgentResourceMonitor(ctx context.Context, a
85358546
return database.WorkspaceAgentResourceMonitor{}, err
85368547
}
85378548

8538-
panic("not implemented")
8549+
q.mutex.Lock()
8550+
defer q.mutex.Unlock()
8551+
8552+
monitor := database.WorkspaceAgentResourceMonitor(arg)
8553+
8554+
q.workspaceAgentResourceMonitors = append(q.workspaceAgentResourceMonitors, monitor)
8555+
8556+
return monitor, nil
85398557
}
85408558

85418559
func (q *FakeQuerier) InsertWorkspaceAgentScriptTimings(_ context.Context, arg database.InsertWorkspaceAgentScriptTimingsParams) (database.WorkspaceAgentScriptTiming, error) {

‎coderd/database/dbmetrics/querymetrics.go

Lines changed: 1 addition & 1 deletion
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

‎coderd/database/dbmock/dbmock.go

Lines changed: 2 additions & 2 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

‎coderd/database/querier.go

Lines changed: 1 addition & 1 deletion
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

‎coderd/database/queries.sql.go

Lines changed: 29 additions & 13 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

‎coderd/database/queries/workspaceagentresourcemonitors.sql

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
-- name: FetchAgentResourceMonitorsByAgentID :one
1+
-- name: FetchAgentResourceMonitorsByAgentID :many
22
SELECT
33
*
44
FROM

0 commit comments

Comments
 (0)