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)
pFad - Phonifier reborn

Pfad - The Proxy pFad of © 2024 Garber Painting. All rights reserved.

Note: This service is not intended for secure transactions such as banking, social media, email, or purchasing. Use at your own risk. We assume no liability whatsoever for broken pages.


Alternative Proxies:

Alternative Proxy

pFad Proxy

pFad v3 Proxy

pFad v4 Proxy