synced/coder
1
0
Fork 0
mirror of https://github.com/coder/coder.git synced 2025-07-15 22:20:27 +00:00
Code Issues Packages Projects Releases Wiki Activity

feat: store coder_workspace_tags in the database (#13294)

Browse Source
This commit is contained in:
Marcin Tojek
2024-05-20 15:30:19 +02:00
committed by GitHub
parent 45b45f1107
commit b8b80fe6d2
30 changed files with 1052 additions and 397 deletions
Download Patch File Download Diff File Expand all files Collapse all files

38
coderd/database/dbmem/dbmem.go
View File

@ -163,6 +163,7 @@ type data struct {
templateVersions []database.TemplateVersionTable
templateVersionParameters []database.TemplateVersionParameter
templateVersionVariables []database.TemplateVersionVariable
templateVersionWorkspaceTags []database.TemplateVersionWorkspaceTag
templates []database.TemplateTable
templateUsageStats []database.TemplateUsageStat
workspaceAgents []database.WorkspaceAgent
@ -4177,6 +4178,24 @@ func (q *FakeQuerier) GetTemplateVersionVariables(_ context.Context, templateVer
return variables, nil
}
func (q *FakeQuerier) GetTemplateVersionWorkspaceTags(_ context.Context, templateVersionID uuid.UUID) ([]database.TemplateVersionWorkspaceTag, error) {
q.mutex.RLock()
defer q.mutex.RUnlock()
workspaceTags := make([]database.TemplateVersionWorkspaceTag, 0)
for _, workspaceTag := range q.templateVersionWorkspaceTags {
if workspaceTag.TemplateVersionID != templateVersionID {
continue
}
workspaceTags = append(workspaceTags, workspaceTag)
}
sort.Slice(workspaceTags, func(i, j int) bool {
return workspaceTags[i].Key < workspaceTags[j].Key
})
return workspaceTags, nil
}
func (q *FakeQuerier) GetTemplateVersionsByIDs(_ context.Context, ids []uuid.UUID) ([]database.TemplateVersion, error) {
q.mutex.RLock()
defer q.mutex.RUnlock()
@ -6352,6 +6371,25 @@ func (q *FakeQuerier) InsertTemplateVersionVariable(_ context.Context, arg datab
return variable, nil
}
func (q *FakeQuerier) InsertTemplateVersionWorkspaceTag(_ context.Context, arg database.InsertTemplateVersionWorkspaceTagParams) (database.TemplateVersionWorkspaceTag, error) {
err := validateDatabaseType(arg)
if err != nil {
return database.TemplateVersionWorkspaceTag{}, err
}
q.mutex.Lock()
defer q.mutex.Unlock()
//nolint:gosimple
workspaceTag := database.TemplateVersionWorkspaceTag{
TemplateVersionID: arg.TemplateVersionID,
Key: arg.Key,
Value: arg.Value,
}
q.templateVersionWorkspaceTags = append(q.templateVersionWorkspaceTags, workspaceTag)
return workspaceTag, nil
}
func (q *FakeQuerier) InsertUser(_ context.Context, arg database.InsertUserParams) (database.User, error) {
if err := validateDatabaseType(arg); err != nil {
return database.User{}, err