From 9cdde0b278279f9e981177b444c1ea3a6a0485ce Mon Sep 17 00:00:00 2001
From: Nathan Coad <nathan@thecoads.com>
Date: Wed, 2 Oct 2024 21:23:54 +1000
Subject: [PATCH] cleanup templates

---
 db/queries/query.sql            |  5 +++++
 db/queries/query.sql.go         | 11 +++++++++++
 server/handler/updateCleanup.go |  2 +-
 3 files changed, 17 insertions(+), 1 deletion(-)

diff --git a/db/queries/query.sql b/db/queries/query.sql
index 1dd2344..9d941ea 100644
--- a/db/queries/query.sql
+++ b/db/queries/query.sql
@@ -58,6 +58,11 @@ DELETE FROM "Inventory"
 WHERE "Vcenter" = sqlc.arg('vc')
 RETURNING *;
 
+-- name: InventoryCleanupTemplates :exec
+DELETE FROM "Inventory"
+WHERE "IsTemplate" = "TRUE"
+RETURNING *;
+
 -- name: CreateUpdate :one
 INSERT INTO "Updates" (
     "InventoryId", "Name", "EventKey", "EventId", "UpdateTime", "UpdateType", "NewVcpus", "NewRam", "NewResourcePool", "NewProvisionedDisk", "UserName", "PlaceholderChange", "RawChangeString"
diff --git a/db/queries/query.sql.go b/db/queries/query.sql.go
index f8449ba..462f485 100644
--- a/db/queries/query.sql.go
+++ b/db/queries/query.sql.go
@@ -644,6 +644,17 @@ func (q *Queries) InventoryCleanup(ctx context.Context, arg InventoryCleanupPara
 	return err
 }
 
+const inventoryCleanupTemplates = `-- name: InventoryCleanupTemplates :exec
+DELETE FROM "Inventory"
+WHERE "IsTemplate" = "TRUE"
+RETURNING Iid, Name, Vcenter, VmId, EventKey, CloudId, CreationTime, DeletionTime, ResourcePool, VmType, Datacenter, Cluster, Folder, ProvisionedDisk, InitialVcpus, InitialRam, IsTemplate, PoweredOn, SrmPlaceholder, VmUuid
+`
+
+func (q *Queries) InventoryCleanupTemplates(ctx context.Context) error {
+	_, err := q.db.ExecContext(ctx, inventoryCleanupTemplates)
+	return err
+}
+
 const inventoryCleanupVcenter = `-- name: InventoryCleanupVcenter :exec
 DELETE FROM "Inventory"
 WHERE "Vcenter" = ?1
diff --git a/server/handler/updateCleanup.go b/server/handler/updateCleanup.go
index 69ec92b..a2b190e 100644
--- a/server/handler/updateCleanup.go
+++ b/server/handler/updateCleanup.go
@@ -26,7 +26,7 @@ func (h *Handler) UpdateCleanup(w http.ResponseWriter, r *http.Request) {
 		err := h.Database.Queries().CleanupUpdates(context.Background(), params)
 	*/
 
-	err := h.Database.Queries().CleanupUpdatesNullVm(context.Background())
+	err := h.Database.Queries().InventoryCleanupTemplates(context.Background())
 
 	if err != nil {
 		h.Logger.Error("Error received cleaning updates table", "error", err)