feat: Add APIs for querying workspaces (#61)

* Add SQL migration * Add query functions for workspaces * Add create routes * Add tests for codersdk * Add workspace parameter route * Add workspace query * Move workspace function * Add querying for workspace history * Fix query * Fix syntax error * Move workspace routes * Fix version * Add CLI tests * Fix syntax error * Remove error * Fix history error * Add new user test * Fix test * Lower target to 70% * Improve comments * Add comment
2026-09-24 15:04:27 +08:00 · 2022-01-25 19:52:58 +00:00
parent 139828d594
commit 5b01f615eb
27 changed files with 2511 additions and 81 deletions
@@ -18,9 +18,13 @@ func New() database.Store {
 		organizationMembers: make([]database.OrganizationMember, 0),
 		users:               make([]database.User, 0),

-		project:          make([]database.Project, 0),
-		projectHistory:   make([]database.ProjectHistory, 0),
-		projectParameter: make([]database.ProjectParameter, 0),
+		project:           make([]database.Project, 0),
+		projectHistory:    make([]database.ProjectHistory, 0),
+		projectParameter:  make([]database.ProjectParameter, 0),
+		workspace:         make([]database.Workspace, 0),
+		workspaceResource: make([]database.WorkspaceResource, 0),
+		workspaceHistory:  make([]database.WorkspaceHistory, 0),
+		workspaceAgent:    make([]database.WorkspaceAgent, 0),
 	}
 }

@@ -33,9 +37,13 @@ type fakeQuerier struct {
 	users               []database.User

 	// New tables
-	project          []database.Project
-	projectHistory   []database.ProjectHistory
-	projectParameter []database.ProjectParameter
+	project           []database.Project
+	projectHistory    []database.ProjectHistory
+	projectParameter  []database.ProjectParameter
+	workspace         []database.Workspace
+	workspaceResource []database.WorkspaceResource
+	workspaceHistory  []database.WorkspaceHistory
+	workspaceAgent    []database.WorkspaceAgent
 }

 // InTx doesn't rollback data properly for in-memory yet.
@@ -74,6 +82,103 @@ func (q *fakeQuerier) GetUserCount(_ context.Context) (int64, error) {
 	return int64(len(q.users)), nil
 }

+func (q *fakeQuerier) GetWorkspaceAgentsByResourceIDs(_ context.Context, ids []uuid.UUID) ([]database.WorkspaceAgent, error) {
+	agents := make([]database.WorkspaceAgent, 0)
+	for _, workspaceAgent := range q.workspaceAgent {
+		for _, id := range ids {
+			if workspaceAgent.WorkspaceResourceID.String() == id.String() {
+				agents = append(agents, workspaceAgent)
+			}
+		}
+	}
+	if len(agents) == 0 {
+		return nil, sql.ErrNoRows
+	}
+	return agents, nil
+}
+
+func (q *fakeQuerier) GetWorkspaceByUserIDAndName(_ context.Context, arg database.GetWorkspaceByUserIDAndNameParams) (database.Workspace, error) {
+	for _, workspace := range q.workspace {
+		if workspace.OwnerID != arg.OwnerID {
+			continue
+		}
+		if !strings.EqualFold(workspace.Name, arg.Name) {
+			continue
+		}
+		return workspace, nil
+	}
+	return database.Workspace{}, sql.ErrNoRows
+}
+
+func (q *fakeQuerier) GetWorkspaceResourcesByHistoryID(_ context.Context, workspaceHistoryID uuid.UUID) ([]database.WorkspaceResource, error) {
+	resources := make([]database.WorkspaceResource, 0)
+	for _, workspaceResource := range q.workspaceResource {
+		if workspaceResource.WorkspaceHistoryID.String() == workspaceHistoryID.String() {
+			resources = append(resources, workspaceResource)
+		}
+	}
+	if len(resources) == 0 {
+		return nil, sql.ErrNoRows
+	}
+	return resources, nil
+}
+
+func (q *fakeQuerier) GetWorkspaceHistoryByWorkspaceIDWithoutAfter(_ context.Context, workspaceID uuid.UUID) (database.WorkspaceHistory, error) {
+	for _, workspaceHistory := range q.workspaceHistory {
+		if workspaceHistory.WorkspaceID.String() != workspaceID.String() {
+			continue
+		}
+		if !workspaceHistory.AfterID.Valid {
+			return workspaceHistory, nil
+		}
+	}
+	return database.WorkspaceHistory{}, sql.ErrNoRows
+}
+
+func (q *fakeQuerier) GetWorkspaceHistoryByWorkspaceID(_ context.Context, workspaceID uuid.UUID) ([]database.WorkspaceHistory, error) {
+	history := make([]database.WorkspaceHistory, 0)
+	for _, workspaceHistory := range q.workspaceHistory {
+		if workspaceHistory.WorkspaceID.String() == workspaceID.String() {
+			history = append(history, workspaceHistory)
+		}
+	}
+	if len(history) == 0 {
+		return nil, sql.ErrNoRows
+	}
+	return history, nil
+}
+
+func (q *fakeQuerier) GetWorkspacesByProjectAndUserID(_ context.Context, arg database.GetWorkspacesByProjectAndUserIDParams) ([]database.Workspace, error) {
+	workspaces := make([]database.Workspace, 0)
+	for _, workspace := range q.workspace {
+		if workspace.OwnerID != arg.OwnerID {
+			continue
+		}
+		if workspace.ProjectID.String() != arg.ProjectID.String() {
+			continue
+		}
+		workspaces = append(workspaces, workspace)
+	}
+	if len(workspaces) == 0 {
+		return nil, sql.ErrNoRows
+	}
+	return workspaces, nil
+}
+
+func (q *fakeQuerier) GetWorkspacesByUserID(_ context.Context, ownerID string) ([]database.Workspace, error) {
+	workspaces := make([]database.Workspace, 0)
+	for _, workspace := range q.workspace {
+		if workspace.OwnerID != ownerID {
+			continue
+		}
+		workspaces = append(workspaces, workspace)
+	}
+	if len(workspaces) == 0 {
+		return nil, sql.ErrNoRows
+	}
+	return workspaces, nil
+}
+
 func (q *fakeQuerier) GetOrganizationByName(_ context.Context, name string) (database.Organization, error) {
 	for _, organization := range q.organizations {
 		if organization.Name == name {
@@ -102,6 +207,15 @@ func (q *fakeQuerier) GetOrganizationsByUserID(_ context.Context, userID string)
 	return organizations, nil
 }

+func (q *fakeQuerier) GetProjectByID(_ context.Context, id uuid.UUID) (database.Project, error) {
+	for _, project := range q.project {
+		if project.ID.String() == id.String() {
+			return project, nil
+		}
+	}
+	return database.Project{}, sql.ErrNoRows
+}
+
 func (q *fakeQuerier) GetProjectByOrganizationAndName(_ context.Context, arg database.GetProjectByOrganizationAndNameParams) (database.Project, error) {
 	for _, project := range q.project {
 		if project.OrganizationID != arg.OrganizationID {
@@ -129,6 +243,16 @@ func (q *fakeQuerier) GetProjectHistoryByProjectID(_ context.Context, projectID
 	return history, nil
 }

+func (q *fakeQuerier) GetProjectHistoryByID(_ context.Context, projectHistoryID uuid.UUID) (database.ProjectHistory, error) {
+	for _, projectHistory := range q.projectHistory {
+		if projectHistory.ID.String() != projectHistoryID.String() {
+			continue
+		}
+		return projectHistory, nil
+	}
+	return database.ProjectHistory{}, sql.ErrNoRows
+}
+
 func (q *fakeQuerier) GetProjectsByOrganizationIDs(_ context.Context, ids []string) ([]database.Project, error) {
 	projects := make([]database.Project, 0)
 	for _, project := range q.project {
@@ -273,6 +397,63 @@ func (q *fakeQuerier) InsertUser(_ context.Context, arg database.InsertUserParam
 	return user, nil
 }

+func (q *fakeQuerier) InsertWorkspace(_ context.Context, arg database.InsertWorkspaceParams) (database.Workspace, error) {
+	//nolint:gosimple
+	workspace := database.Workspace{
+		ID:        arg.ID,
+		CreatedAt: arg.CreatedAt,
+		UpdatedAt: arg.UpdatedAt,
+		OwnerID:   arg.OwnerID,
+		ProjectID: arg.ProjectID,
+		Name:      arg.Name,
+	}
+	q.workspace = append(q.workspace, workspace)
+	return workspace, nil
+}
+
+func (q *fakeQuerier) InsertWorkspaceAgent(_ context.Context, arg database.InsertWorkspaceAgentParams) (database.WorkspaceAgent, error) {
+	//nolint:gosimple
+	workspaceAgent := database.WorkspaceAgent{
+		ID:                  arg.ID,
+		CreatedAt:           arg.CreatedAt,
+		UpdatedAt:           arg.UpdatedAt,
+		WorkspaceResourceID: arg.WorkspaceResourceID,
+		InstanceMetadata:    arg.InstanceMetadata,
+		ResourceMetadata:    arg.ResourceMetadata,
+	}
+	q.workspaceAgent = append(q.workspaceAgent, workspaceAgent)
+	return workspaceAgent, nil
+}
+
+func (q *fakeQuerier) InsertWorkspaceHistory(_ context.Context, arg database.InsertWorkspaceHistoryParams) (database.WorkspaceHistory, error) {
+	workspaceHistory := database.WorkspaceHistory{
+		ID:               arg.ID,
+		CreatedAt:        arg.CreatedAt,
+		UpdatedAt:        arg.UpdatedAt,
+		WorkspaceID:      arg.WorkspaceID,
+		ProjectHistoryID: arg.ProjectHistoryID,
+		BeforeID:         arg.BeforeID,
+		Transition:       arg.Transition,
+		Initiator:        arg.Initiator,
+		ProvisionJobID:   arg.ProvisionJobID,
+	}
+	q.workspaceHistory = append(q.workspaceHistory, workspaceHistory)
+	return workspaceHistory, nil
+}
+
+func (q *fakeQuerier) InsertWorkspaceResource(_ context.Context, arg database.InsertWorkspaceResourceParams) (database.WorkspaceResource, error) {
+	workspaceResource := database.WorkspaceResource{
+		ID:                  arg.ID,
+		CreatedAt:           arg.CreatedAt,
+		WorkspaceHistoryID:  arg.WorkspaceHistoryID,
+		Type:                arg.Type,
+		Name:                arg.Name,
+		WorkspaceAgentToken: arg.WorkspaceAgentToken,
+	}
+	q.workspaceResource = append(q.workspaceResource, workspaceResource)
+	return workspaceResource, nil
+}
+
 func (q *fakeQuerier) UpdateAPIKeyByID(_ context.Context, arg database.UpdateAPIKeyByIDParams) error {
 	for index, apiKey := range q.apiKeys {
 		if apiKey.ID != arg.ID {
@@ -288,3 +469,16 @@ func (q *fakeQuerier) UpdateAPIKeyByID(_ context.Context, arg database.UpdateAPI
 	}
 	return sql.ErrNoRows
 }
+
+func (q *fakeQuerier) UpdateWorkspaceHistoryByID(_ context.Context, arg database.UpdateWorkspaceHistoryByIDParams) error {
+	for index, workspaceHistory := range q.workspaceHistory {
+		if workspaceHistory.ID.String() != arg.ID.String() {
+			continue
+		}
+		workspaceHistory.UpdatedAt = arg.UpdatedAt
+		workspaceHistory.AfterID = arg.AfterID
+		q.workspaceHistory[index] = workspaceHistory
+		return nil
+	}
+	return sql.ErrNoRows
+}