Skip to content

Commit 1122da1

Browse files
kylecarbskylecarbs
committed
feat: Add parameter and jobs database schema
This modifies a prior migration which is typically forbidden, but because we're pre-production deployment I felt grouping would be helpful to future contributors. This adds database functions that are required for the provisioner daemon and job queue logic.
1 parent a193a08 commit 1122da1

File tree

11 files changed

+1169
-62
lines changed

11 files changed

+1169
-62
lines changed

‎database/databasefake/databasefake.go

+198-16
Original file line numberDiff line numberDiff line change
@@ -18,13 +18,16 @@ func New() database.Store {
1818
organizationMembers: make([]database.OrganizationMember, 0),
1919
users: make([]database.User, 0),
2020

21-
project: make([]database.Project, 0),
22-
projectHistory: make([]database.ProjectHistory, 0),
23-
projectParameter: make([]database.ProjectParameter, 0),
24-
workspace: make([]database.Workspace, 0),
25-
workspaceResource: make([]database.WorkspaceResource, 0),
26-
workspaceHistory: make([]database.WorkspaceHistory, 0),
27-
workspaceAgent: make([]database.WorkspaceAgent, 0),
21+
parameterValue: make([]database.ParameterValue, 0),
22+
project: make([]database.Project, 0),
23+
projectHistory: make([]database.ProjectHistory, 0),
24+
projectParameter: make([]database.ProjectParameter, 0),
25+
provisionerDaemons: make([]database.ProvisionerDaemon, 0),
26+
provisionerJobs: make([]database.ProvisionerJob, 0),
27+
workspace: make([]database.Workspace, 0),
28+
workspaceResource: make([]database.WorkspaceResource, 0),
29+
workspaceHistory: make([]database.WorkspaceHistory, 0),
30+
workspaceAgent: make([]database.WorkspaceAgent, 0),
2831
}
2932
}
3033

@@ -37,20 +40,48 @@ type fakeQuerier struct {
3740
users []database.User
3841

3942
// New tables
40-
project []database.Project
41-
projectHistory []database.ProjectHistory
42-
projectParameter []database.ProjectParameter
43-
workspace []database.Workspace
44-
workspaceResource []database.WorkspaceResource
45-
workspaceHistory []database.WorkspaceHistory
46-
workspaceAgent []database.WorkspaceAgent
43+
parameterValue []database.ParameterValue
44+
project []database.Project
45+
projectHistory []database.ProjectHistory
46+
projectParameter []database.ProjectParameter
47+
provisionerDaemons []database.ProvisionerDaemon
48+
provisionerJobs []database.ProvisionerJob
49+
workspace []database.Workspace
50+
workspaceResource []database.WorkspaceResource
51+
workspaceHistory []database.WorkspaceHistory
52+
workspaceAgent []database.WorkspaceAgent
4753
}
4854

4955
// InTx doesn't rollback data properly for in-memory yet.
5056
func (q *fakeQuerier) InTx(fn func(database.Store) error) error {
5157
return fn(q)
5258
}
5359

60+
func (q *fakeQuerier) AcquireProvisionerJob(_ context.Context, arg database.AcquireProvisionerJobParams) (database.ProvisionerJob, error) {
61+
for index, provisionerJob := range q.provisionerJobs {
62+
if provisionerJob.StartedAt.Valid {
63+
continue
64+
}
65+
found := false
66+
for _, provisionerType := range arg.Types {
67+
if provisionerJob.Provisioner != provisionerType {
68+
continue
69+
}
70+
found = true
71+
break
72+
}
73+
if !found {
74+
continue
75+
}
76+
provisionerJob.StartedAt = arg.StartedAt
77+
provisionerJob.UpdatedAt = arg.StartedAt.Time
78+
provisionerJob.WorkerID = arg.WorkerID
79+
q.provisionerJobs[index] = provisionerJob
80+
return provisionerJob, nil
81+
}
82+
return database.ProvisionerJob{}, sql.ErrNoRows
83+
}
84+
5485
func (q *fakeQuerier) GetAPIKeyByID(_ context.Context, id string) (database.APIKey, error) {
5586
for _, apiKey := range q.apiKeys {
5687
if apiKey.ID == id {
@@ -97,6 +128,15 @@ func (q *fakeQuerier) GetWorkspaceAgentsByResourceIDs(_ context.Context, ids []u
97128
return agents, nil
98129
}
99130

131+
func (q *fakeQuerier) GetWorkspaceByID(_ context.Context, id uuid.UUID) (database.Workspace, error) {
132+
for _, workspace := range q.workspace {
133+
if workspace.ID.String() == id.String() {
134+
return workspace, nil
135+
}
136+
}
137+
return database.Workspace{}, sql.ErrNoRows
138+
}
139+
100140
func (q *fakeQuerier) GetWorkspaceByUserIDAndName(_ context.Context, arg database.GetWorkspaceByUserIDAndNameParams) (database.Workspace, error) {
101141
for _, workspace := range q.workspace {
102142
if workspace.OwnerID != arg.OwnerID {
@@ -123,6 +163,15 @@ func (q *fakeQuerier) GetWorkspaceResourcesByHistoryID(_ context.Context, worksp
123163
return resources, nil
124164
}
125165

166+
func (q *fakeQuerier) GetWorkspaceHistoryByID(_ context.Context, id uuid.UUID) (database.WorkspaceHistory, error) {
167+
for _, history := range q.workspaceHistory {
168+
if history.ID.String() == id.String() {
169+
return history, nil
170+
}
171+
}
172+
return database.WorkspaceHistory{}, sql.ErrNoRows
173+
}
174+
126175
func (q *fakeQuerier) GetWorkspaceHistoryByWorkspaceIDWithoutAfter(_ context.Context, workspaceID uuid.UUID) (database.WorkspaceHistory, error) {
127176
for _, workspaceHistory := range q.workspaceHistory {
128177
if workspaceHistory.WorkspaceID.String() != workspaceID.String() {
@@ -179,6 +228,15 @@ func (q *fakeQuerier) GetWorkspacesByUserID(_ context.Context, ownerID string) (
179228
return workspaces, nil
180229
}
181230

231+
func (q *fakeQuerier) GetOrganizationByID(_ context.Context, id string) (database.Organization, error) {
232+
for _, organization := range q.organizations {
233+
if organization.ID == id {
234+
return organization, nil
235+
}
236+
}
237+
return database.Organization{}, sql.ErrNoRows
238+
}
239+
182240
func (q *fakeQuerier) GetOrganizationByName(_ context.Context, name string) (database.Organization, error) {
183241
for _, organization := range q.organizations {
184242
if organization.Name == name {
@@ -207,6 +265,23 @@ func (q *fakeQuerier) GetOrganizationsByUserID(_ context.Context, userID string)
207265
return organizations, nil
208266
}
209267

268+
func (q *fakeQuerier) GetParameterValuesByScope(_ context.Context, arg database.GetParameterValuesByScopeParams) ([]database.ParameterValue, error) {
269+
parameterValues := make([]database.ParameterValue, 0)
270+
for _, parameterValue := range q.parameterValue {
271+
if parameterValue.Scope != arg.Scope {
272+
continue
273+
}
274+
if parameterValue.ScopeID != arg.ScopeID {
275+
continue
276+
}
277+
parameterValues = append(parameterValues, parameterValue)
278+
}
279+
if len(parameterValues) == 0 {
280+
return nil, sql.ErrNoRows
281+
}
282+
return parameterValues, nil
283+
}
284+
210285
func (q *fakeQuerier) GetProjectByID(_ context.Context, id uuid.UUID) (database.Project, error) {
211286
for _, project := range q.project {
212287
if project.ID.String() == id.String() {
@@ -253,6 +328,20 @@ func (q *fakeQuerier) GetProjectHistoryByID(_ context.Context, projectHistoryID
253328
return database.ProjectHistory{}, sql.ErrNoRows
254329
}
255330

331+
func (q *fakeQuerier) GetProjectParametersByHistoryID(_ context.Context, projectHistoryID uuid.UUID) ([]database.ProjectParameter, error) {
332+
parameters := make([]database.ProjectParameter, 0)
333+
for _, projectParameter := range q.projectParameter {
334+
if projectParameter.ProjectHistoryID.String() != projectHistoryID.String() {
335+
continue
336+
}
337+
parameters = append(parameters, projectParameter)
338+
}
339+
if len(parameters) == 0 {
340+
return nil, sql.ErrNoRows
341+
}
342+
return parameters, nil
343+
}
344+
256345
func (q *fakeQuerier) GetProjectsByOrganizationIDs(_ context.Context, ids []string) ([]database.Project, error) {
257346
projects := make([]database.Project, 0)
258347
for _, project := range q.project {
@@ -282,6 +371,26 @@ func (q *fakeQuerier) GetOrganizationMemberByUserID(_ context.Context, arg datab
282371
return database.OrganizationMember{}, sql.ErrNoRows
283372
}
284373

374+
func (q *fakeQuerier) GetProvisionerDaemonByID(_ context.Context, id uuid.UUID) (database.ProvisionerDaemon, error) {
375+
for _, provisionerDaemon := range q.provisionerDaemons {
376+
if provisionerDaemon.ID.String() != id.String() {
377+
continue
378+
}
379+
return provisionerDaemon, nil
380+
}
381+
return database.ProvisionerDaemon{}, sql.ErrNoRows
382+
}
383+
384+
func (q *fakeQuerier) GetProvisionerJobByID(_ context.Context, id uuid.UUID) (database.ProvisionerJob, error) {
385+
for _, provisionerJob := range q.provisionerJobs {
386+
if provisionerJob.ID.String() != id.String() {
387+
continue
388+
}
389+
return provisionerJob, nil
390+
}
391+
return database.ProvisionerJob{}, sql.ErrNoRows
392+
}
393+
285394
func (q *fakeQuerier) InsertAPIKey(_ context.Context, arg database.InsertAPIKeyParams) (database.APIKey, error) {
286395
//nolint:gosimple
287396
key := database.APIKey{
@@ -329,6 +438,23 @@ func (q *fakeQuerier) InsertOrganizationMember(_ context.Context, arg database.I
329438
return organizationMember, nil
330439
}
331440

441+
func (q *fakeQuerier) InsertParameterValue(ctx context.Context, arg database.InsertParameterValueParams) (database.ParameterValue, error) {
442+
parameterValue := database.ParameterValue{
443+
ID: arg.ID,
444+
Name: arg.Name,
445+
CreatedAt: arg.CreatedAt,
446+
UpdatedAt: arg.UpdatedAt,
447+
Scope: arg.Scope,
448+
ScopeID: arg.ScopeID,
449+
SourceScheme: arg.SourceScheme,
450+
SourceValue: arg.SourceValue,
451+
DestinationScheme: arg.DestinationScheme,
452+
DestinationValue: arg.DestinationValue,
453+
}
454+
q.parameterValue = append(q.parameterValue, parameterValue)
455+
return parameterValue, nil
456+
}
457+
332458
func (q *fakeQuerier) InsertProject(_ context.Context, arg database.InsertProjectParams) (database.Project, error) {
333459
project := database.Project{
334460
ID: arg.ID,
@@ -367,9 +493,11 @@ func (q *fakeQuerier) InsertProjectParameter(_ context.Context, arg database.Ins
367493
ProjectHistoryID: arg.ProjectHistoryID,
368494
Name: arg.Name,
369495
Description: arg.Description,
370-
DefaultSource: arg.DefaultSource,
496+
DefaultSourceScheme: arg.DefaultSourceScheme,
497+
DefaultSourceValue: arg.DefaultSourceValue,
371498
AllowOverrideSource: arg.AllowOverrideSource,
372-
DefaultDestination: arg.DefaultDestination,
499+
DefaultDestinationScheme: arg.DefaultDestinationScheme,
500+
DefaultDestinationValue: arg.DefaultDestinationValue,
373501
AllowOverrideDestination: arg.AllowOverrideDestination,
374502
DefaultRefresh: arg.DefaultRefresh,
375503
RedisplayValue: arg.RedisplayValue,
@@ -382,6 +510,32 @@ func (q *fakeQuerier) InsertProjectParameter(_ context.Context, arg database.Ins
382510
return param, nil
383511
}
384512

513+
func (q *fakeQuerier) InsertProvisionerDaemon(_ context.Context, arg database.InsertProvisionerDaemonParams) (database.ProvisionerDaemon, error) {
514+
daemon := database.ProvisionerDaemon{
515+
ID: arg.ID,
516+
CreatedAt: arg.CreatedAt,
517+
Name: arg.Name,
518+
Provisioners: arg.Provisioners,
519+
}
520+
q.provisionerDaemons = append(q.provisionerDaemons, daemon)
521+
return daemon, nil
522+
}
523+
524+
func (q *fakeQuerier) InsertProvisionerJob(_ context.Context, arg database.InsertProvisionerJobParams) (database.ProvisionerJob, error) {
525+
job := database.ProvisionerJob{
526+
ID: arg.ID,
527+
CreatedAt: arg.CreatedAt,
528+
UpdatedAt: arg.UpdatedAt,
529+
InitiatorID: arg.InitiatorID,
530+
Provisioner: arg.Provisioner,
531+
ProjectID: arg.ProjectID,
532+
Type: arg.Type,
533+
Input: arg.Input,
534+
}
535+
q.provisionerJobs = append(q.provisionerJobs, job)
536+
return job, nil
537+
}
538+
385539
func (q *fakeQuerier) InsertUser(_ context.Context, arg database.InsertUserParams) (database.User, error) {
386540
user := database.User{
387541
ID: arg.ID,
@@ -470,6 +624,34 @@ func (q *fakeQuerier) UpdateAPIKeyByID(_ context.Context, arg database.UpdateAPI
470624
return sql.ErrNoRows
471625
}
472626

627+
func (q *fakeQuerier) UpdateProvisionerDaemonByID(_ context.Context, arg database.UpdateProvisionerDaemonByIDParams) error {
628+
for index, daemon := range q.provisionerDaemons {
629+
if arg.ID.String() != daemon.ID.String() {
630+
continue
631+
}
632+
daemon.UpdatedAt = arg.UpdatedAt
633+
daemon.Provisioners = arg.Provisioners
634+
q.provisionerDaemons[index] = daemon
635+
return nil
636+
}
637+
return sql.ErrNoRows
638+
}
639+
640+
func (q *fakeQuerier) UpdateProvisionerJobByID(_ context.Context, arg database.UpdateProvisionerJobByIDParams) error {
641+
for index, job := range q.provisionerJobs {
642+
if arg.ID.String() != job.ID.String() {
643+
continue
644+
}
645+
job.CompletedAt = arg.CompletedAt
646+
job.CancelledAt = arg.CancelledAt
647+
job.UpdatedAt = arg.UpdatedAt
648+
job.Error = arg.Error
649+
q.provisionerJobs[index] = job
650+
return nil
651+
}
652+
return sql.ErrNoRows
653+
}
654+
473655
func (q *fakeQuerier) UpdateWorkspaceHistoryByID(_ context.Context, arg database.UpdateWorkspaceHistoryByIDParams) error {
474656
for index, workspaceHistory := range q.workspaceHistory {
475657
if workspaceHistory.ID.String() != arg.ID.String() {

0 commit comments

Comments
 (0)