Movatterモバイル変換

[0]ホーム
URL:

Skip to content

Navigation Menu

Sign in
Appearance settings

Search code, repositories, users, issues, pull requests...

Provide feedback

We read every piece of feedback, and take your input very seriously.

Saved searches

Use saved searches to filter your results more quickly

 Sign up
Appearance settings

Commit96a37bb

Browse files
committed
make gen impl
1 parentd98cbb1 commit96a37bb

File tree

3 files changed

+44
-4
lines changed

3 files changed

+44
-4
lines changed

‎coderd/database/dbauthz/dbauthz.go

Lines changed: 16 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1439,7 +1439,14 @@ func (q *querier) FetchMemoryResourceMonitorsByAgentID(ctx context.Context, agen
14391439
}
14401440

14411441
func (q*querier)FetchMemoryResourceMonitorsCreatedAfter(ctx context.Context,createdAt time.Time) ([]database.WorkspaceAgentMemoryResourceMonitor,error) {
1442-
panic("not implemented")
1442+
// Ideally, we would return a list of monitors that the user has access to. However, that check would need to
1443+
// be implemented similarly to GetWorkspaces, which is more complex than what we're doing here. Since this query
1444+
// was introduced for telemetry, we perform a simpler check.
1445+
iferr:=q.authorizeContext(ctx,policy.ActionRead,rbac.ResourceWorkspaceAgentResourceMonitor);err!=nil {
1446+
returnnil,err
1447+
}
1448+
1449+
returnq.db.FetchMemoryResourceMonitorsCreatedAfter(ctx,createdAt)
14431450
}
14441451

14451452
func (q*querier)FetchNewMessageMetadata(ctx context.Context,arg database.FetchNewMessageMetadataParams) (database.FetchNewMessageMetadataRow,error) {
@@ -1464,7 +1471,14 @@ func (q *querier) FetchVolumesResourceMonitorsByAgentID(ctx context.Context, age
14641471
}
14651472

14661473
func (q*querier)FetchVolumesResourceMonitorsCreatedAfter(ctx context.Context,createdAt time.Time) ([]database.WorkspaceAgentVolumeResourceMonitor,error) {
1467-
panic("not implemented")
1474+
// Ideally, we would return a list of monitors that the user has access to. However, that check would need to
1475+
// be implemented similarly to GetWorkspaces, which is more complex than what we're doing here. Since this query
1476+
// was introduced for telemetry, we perform a simpler check.
1477+
iferr:=q.authorizeContext(ctx,policy.ActionRead,rbac.ResourceWorkspaceAgentResourceMonitor);err!=nil {
1478+
returnnil,err
1479+
}
1480+
1481+
returnq.db.FetchVolumesResourceMonitorsCreatedAfter(ctx,createdAt)
14681482
}
14691483

14701484
func (q*querier)GetAPIKeyByID(ctx context.Context,idstring) (database.APIKey,error) {

‎coderd/database/dbauthz/dbauthz_test.go

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4919,6 +4919,14 @@ func (s *MethodTestSuite) TestResourcesMonitor() {
49194919
}).Asserts(rbac.ResourceWorkspaceAgentResourceMonitor,policy.ActionUpdate)
49204920
}))
49214921

4922+
s.Run("FetchMemoryResourceMonitorsCreatedAfter",s.Subtest(func(db database.Store,check*expects) {
4923+
check.Args(dbtime.Now()).Asserts(rbac.ResourceWorkspaceAgentResourceMonitor,policy.ActionRead)
4924+
}))
4925+
4926+
s.Run("FetchVolumesResourceMonitorsCreatedAfter",s.Subtest(func(db database.Store,check*expects) {
4927+
check.Args(dbtime.Now()).Asserts(rbac.ResourceWorkspaceAgentResourceMonitor,policy.ActionRead)
4928+
}))
4929+
49224930
s.Run("FetchMemoryResourceMonitorsByAgentID",s.Subtest(func(db database.Store,check*expects) {
49234931
agt,w:=createAgent(s.T(),db)
49244932

‎coderd/database/dbmem/dbmem.go

Lines changed: 20 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2392,7 +2392,16 @@ func (q *FakeQuerier) FetchMemoryResourceMonitorsByAgentID(_ context.Context, ag
23922392
}
23932393

23942394
func (q*FakeQuerier)FetchMemoryResourceMonitorsCreatedAfter(ctx context.Context,createdAt time.Time) ([]database.WorkspaceAgentMemoryResourceMonitor,error) {
2395-
panic("not implemented")
2395+
q.mutex.RLock()
2396+
deferq.mutex.RUnlock()
2397+
2398+
monitors:= []database.WorkspaceAgentMemoryResourceMonitor{}
2399+
for_,monitor:=rangeq.workspaceAgentMemoryResourceMonitors {
2400+
ifmonitor.CreatedAt.After(createdAt) {
2401+
monitors=append(monitors,monitor)
2402+
}
2403+
}
2404+
returnmonitors,nil
23962405
}
23972406

23982407
func (q*FakeQuerier)FetchNewMessageMetadata(_ context.Context,arg database.FetchNewMessageMetadataParams) (database.FetchNewMessageMetadataRow,error) {
@@ -2440,7 +2449,16 @@ func (q *FakeQuerier) FetchVolumesResourceMonitorsByAgentID(_ context.Context, a
24402449
}
24412450

24422451
func (q*FakeQuerier)FetchVolumesResourceMonitorsCreatedAfter(ctx context.Context,createdAt time.Time) ([]database.WorkspaceAgentVolumeResourceMonitor,error) {
2443-
panic("not implemented")
2452+
q.mutex.RLock()
2453+
deferq.mutex.RUnlock()
2454+
2455+
monitors:= []database.WorkspaceAgentVolumeResourceMonitor{}
2456+
for_,monitor:=rangeq.workspaceAgentVolumeResourceMonitors {
2457+
ifmonitor.CreatedAt.After(createdAt) {
2458+
monitors=append(monitors,monitor)
2459+
}
2460+
}
2461+
returnmonitors,nil
24442462
}
24452463

24462464
func (q*FakeQuerier)GetAPIKeyByID(_ context.Context,idstring) (database.APIKey,error) {

0 commit comments

Comments
 (0)
[8]ページ先頭
©2009-2025 Movatter.jp