Exclude template active versions

Emyrk · Emyrk · commit 1de0a5dfb41b · 2023-10-05T15:46:17.000-05:00
diff --git a/coderd/database/dbfake/dbfake.go b/coderd/database/dbfake/dbfake.go
@@ -5199,10 +5199,14 @@ func (q *FakeQuerier) PruneUnusedTemplateVersions(ctx context.Context, arg datab
 			Version: b.TemplateVersionID,
 		}
 	}
+
 	usedVersions := make(map[uuid.UUID]bool)
 	for _, l := range latest {
 		usedVersions[l.Version] = true
 	}
+	for _, tpl := range q.templates {
+		usedVersions[tpl.ActiveVersionID] = true
+	}
 
 	var deleted []uuid.UUID
 	for i, v := range q.templateVersions {
diff --git a/coderd/database/queries.sql.go b/coderd/database/queries.sql.go
diff --git a/coderd/database/queries/templateversions.sql b/coderd/database/queries/templateversions.sql
@@ -150,6 +150,8 @@ UPDATE
 SET
 	deleted = true,
 	updated_at = @updated_at
+FROM
+    (SELECT active_version_id FROM templates WHERE id = @template_id) AS active_version
 WHERE
 	-- Actively used template versions (meaning the latest build is using
 	-- the version) are never pruned. A "restart" command on the workspace,
@@ -165,6 +167,8 @@ WHERE
 			workspace_builds
 		ORDER BY build_number DESC
 	)
+  	-- Also never delete the active template version
+	AND template_versions.id != ANY(active_version)
     -- Ignore already deleted versions.
 	AND template_versions.deleted = false
 	-- Scope a prune to a single template.

Original file line number	Diff line number	Diff line change
`@@ -5199,10 +5199,14 @@ func (q *FakeQuerier) PruneUnusedTemplateVersions(ctx context.Context, arg datab`
`5199`	`5199`	`Version: b.TemplateVersionID,`
`5200`	`5200`	`}`
`5201`	`5201`	`}`
	`5202`	`+`
`5202`	`5203`	`usedVersions := make(map[uuid.UUID]bool)`
`5203`	`5204`	`for _, l := range latest {`
`5204`	`5205`	`usedVersions[l.Version] = true`
`5205`	`5206`	`}`
	`5207`	`+ for _, tpl := range q.templates {`
	`5208`	`+ usedVersions[tpl.ActiveVersionID] = true`
	`5209`	`+ }`
`5206`	`5210`
`5207`	`5211`	`var deleted []uuid.UUID`
`5208`	`5212`	`for i, v := range q.templateVersions {`