@@ -26,21 +26,22 @@ func New() database.Store {
26
26
organizations :make ([]database.Organization ,0 ),
27
27
users :make ([]database.User ,0 ),
28
28
29
- auditLogs :make ([]database.AuditLog ,0 ),
30
- files :make ([]database.File ,0 ),
31
- gitSSHKey :make ([]database.GitSSHKey ,0 ),
32
- parameterSchemas :make ([]database.ParameterSchema ,0 ),
33
- parameterValues :make ([]database.ParameterValue ,0 ),
34
- provisionerDaemons :make ([]database.ProvisionerDaemon ,0 ),
35
- provisionerJobAgents :make ([]database.WorkspaceAgent ,0 ),
36
- provisionerJobLogs :make ([]database.ProvisionerJobLog ,0 ),
37
- provisionerJobResources :make ([]database.WorkspaceResource ,0 ),
38
- provisionerJobs :make ([]database.ProvisionerJob ,0 ),
39
- templateVersions :make ([]database.TemplateVersion ,0 ),
40
- templates :make ([]database.Template ,0 ),
41
- workspaceBuilds :make ([]database.WorkspaceBuild ,0 ),
42
- workspaceApps :make ([]database.WorkspaceApp ,0 ),
43
- workspaces :make ([]database.Workspace ,0 ),
29
+ auditLogs :make ([]database.AuditLog ,0 ),
30
+ files :make ([]database.File ,0 ),
31
+ gitSSHKey :make ([]database.GitSSHKey ,0 ),
32
+ parameterSchemas :make ([]database.ParameterSchema ,0 ),
33
+ parameterValues :make ([]database.ParameterValue ,0 ),
34
+ provisionerDaemons :make ([]database.ProvisionerDaemon ,0 ),
35
+ provisionerJobAgents :make ([]database.WorkspaceAgent ,0 ),
36
+ provisionerJobLogs :make ([]database.ProvisionerJobLog ,0 ),
37
+ provisionerJobResources :make ([]database.WorkspaceResource ,0 ),
38
+ provisionerJobResourceMetadata :make ([]database.WorkspaceResourceMetadatum ,0 ),
39
+ provisionerJobs :make ([]database.ProvisionerJob ,0 ),
40
+ templateVersions :make ([]database.TemplateVersion ,0 ),
41
+ templates :make ([]database.Template ,0 ),
42
+ workspaceBuilds :make ([]database.WorkspaceBuild ,0 ),
43
+ workspaceApps :make ([]database.WorkspaceApp ,0 ),
44
+ workspaces :make ([]database.Workspace ,0 ),
44
45
},
45
46
}
46
47
}
@@ -74,21 +75,22 @@ type data struct {
74
75
users []database.User
75
76
76
77
// New tables
77
- auditLogs []database.AuditLog
78
- files []database.File
79
- gitSSHKey []database.GitSSHKey
80
- parameterSchemas []database.ParameterSchema
81
- parameterValues []database.ParameterValue
82
- provisionerDaemons []database.ProvisionerDaemon
83
- provisionerJobAgents []database.WorkspaceAgent
84
- provisionerJobLogs []database.ProvisionerJobLog
85
- provisionerJobResources []database.WorkspaceResource
86
- provisionerJobs []database.ProvisionerJob
87
- templateVersions []database.TemplateVersion
88
- templates []database.Template
89
- workspaceBuilds []database.WorkspaceBuild
90
- workspaceApps []database.WorkspaceApp
91
- workspaces []database.Workspace
78
+ auditLogs []database.AuditLog
79
+ files []database.File
80
+ gitSSHKey []database.GitSSHKey
81
+ parameterSchemas []database.ParameterSchema
82
+ parameterValues []database.ParameterValue
83
+ provisionerDaemons []database.ProvisionerDaemon
84
+ provisionerJobAgents []database.WorkspaceAgent
85
+ provisionerJobLogs []database.ProvisionerJobLog
86
+ provisionerJobResources []database.WorkspaceResource
87
+ provisionerJobResourceMetadata []database.WorkspaceResourceMetadatum
88
+ provisionerJobs []database.ProvisionerJob
89
+ templateVersions []database.TemplateVersion
90
+ templates []database.Template
91
+ workspaceBuilds []database.WorkspaceBuild
92
+ workspaceApps []database.WorkspaceApp
93
+ workspaces []database.Workspace
92
94
93
95
deploymentID string
94
96
}
@@ -1331,6 +1333,34 @@ func (q *fakeQuerier) GetWorkspaceResourcesCreatedAfter(_ context.Context, after
1331
1333
return resources ,nil
1332
1334
}
1333
1335
1336
+ func (q * fakeQuerier )GetWorkspaceResourceMetadataByResourceID (_ context.Context ,id uuid.UUID ) ([]database.WorkspaceResourceMetadatum ,error ) {
1337
+ q .mutex .RLock ()
1338
+ defer q .mutex .RUnlock ()
1339
+
1340
+ metadata := make ([]database.WorkspaceResourceMetadatum ,0 )
1341
+ for _ ,metadatum := range q .provisionerJobResourceMetadata {
1342
+ if metadatum .WorkspaceResourceID .String ()== id .String () {
1343
+ metadata = append (metadata ,metadatum )
1344
+ }
1345
+ }
1346
+ return metadata ,nil
1347
+ }
1348
+
1349
+ func (q * fakeQuerier )GetWorkspaceResourceMetadataByResourceIDs (_ context.Context ,ids []uuid.UUID ) ([]database.WorkspaceResourceMetadatum ,error ) {
1350
+ q .mutex .RLock ()
1351
+ defer q .mutex .RUnlock ()
1352
+
1353
+ metadata := make ([]database.WorkspaceResourceMetadatum ,0 )
1354
+ for _ ,metadatum := range q .provisionerJobResourceMetadata {
1355
+ for _ ,id := range ids {
1356
+ if metadatum .WorkspaceResourceID .String ()== id .String () {
1357
+ metadata = append (metadata ,metadatum )
1358
+ }
1359
+ }
1360
+ }
1361
+ return metadata ,nil
1362
+ }
1363
+
1334
1364
func (q * fakeQuerier )GetProvisionerJobsByIDs (_ context.Context ,ids []uuid.UUID ) ([]database.ProvisionerJob ,error ) {
1335
1365
q .mutex .RLock ()
1336
1366
defer q .mutex .RUnlock ()
@@ -1659,6 +1689,21 @@ func (q *fakeQuerier) InsertWorkspaceResource(_ context.Context, arg database.In
1659
1689
return resource ,nil
1660
1690
}
1661
1691
1692
+ func (q * fakeQuerier )InsertWorkspaceResourceMetadata (_ context.Context ,arg database.InsertWorkspaceResourceMetadataParams ) (database.WorkspaceResourceMetadatum ,error ) {
1693
+ q .mutex .Lock ()
1694
+ defer q .mutex .Unlock ()
1695
+
1696
+ //nolint:gosimple
1697
+ metadatum := database.WorkspaceResourceMetadatum {
1698
+ WorkspaceResourceID :arg .WorkspaceResourceID ,
1699
+ Key :arg .Key ,
1700
+ Value :arg .Value ,
1701
+ Sensitive :arg .Sensitive ,
1702
+ }
1703
+ q .provisionerJobResourceMetadata = append (q .provisionerJobResourceMetadata ,metadatum )
1704
+ return metadatum ,nil
1705
+ }
1706
+
1662
1707
func (q * fakeQuerier )InsertUser (_ context.Context ,arg database.InsertUserParams ) (database.User ,error ) {
1663
1708
q .mutex .Lock ()
1664
1709
defer q .mutex .Unlock ()