add temp cleanup function

db/queries/query.sql.go

@@ -348,6 +348,55 @@ func (q *Queries) GetInventoryEventId(ctx context.Context, cloudid sql.NullStrin
 	return i, err
 }
 
+const getInventoryVcUrl = `-- name: GetInventoryVcUrl :many
+SELECT Iid, Name, Vcenter, VmId, EventKey, CloudId, CreationTime, DeletionTime, ResourcePool, VmType, Datacenter, Cluster, Folder, ProvisionedDisk, InitialVcpus, InitialRam, IsTemplate, PoweredOn, SrmPlaceholder, VmUuid FROM "Inventory"
+WHERE "Vcenter" = ?1
+`
+
+func (q *Queries) GetInventoryVcUrl(ctx context.Context, vc string) ([]Inventory, error) {
+	rows, err := q.db.QueryContext(ctx, getInventoryVcUrl, vc)
+	if err != nil {
+		return nil, err
+	}
+	defer rows.Close()
+	var items []Inventory
+	for rows.Next() {
+		var i Inventory
+		if err := rows.Scan(
+			&i.Iid,
+			&i.Name,
+			&i.Vcenter,
+			&i.VmId,
+			&i.EventKey,
+			&i.CloudId,
+			&i.CreationTime,
+			&i.DeletionTime,
+			&i.ResourcePool,
+			&i.VmType,
+			&i.Datacenter,
+			&i.Cluster,
+			&i.Folder,
+			&i.ProvisionedDisk,
+			&i.InitialVcpus,
+			&i.InitialRam,
+			&i.IsTemplate,
+			&i.PoweredOn,
+			&i.SrmPlaceholder,
+			&i.VmUuid,
+		); err != nil {
+			return nil, err
+		}
+		items = append(items, i)
+	}
+	if err := rows.Close(); err != nil {
+		return nil, err
+	}
+	if err := rows.Err(); err != nil {
+		return nil, err
+	}
+	return items, nil
+}
+
 const getInventoryVmId = `-- name: GetInventoryVmId :one
 SELECT Iid, Name, Vcenter, VmId, EventKey, CloudId, CreationTime, DeletionTime, ResourcePool, VmType, Datacenter, Cluster, Folder, ProvisionedDisk, InitialVcpus, InitialRam, IsTemplate, PoweredOn, SrmPlaceholder, VmUuid FROM "Inventory"
 WHERE "VmId" = ?1  AND "Datacenter" = ?2
@@ -529,6 +578,17 @@ func (q *Queries) InventoryCleanup(ctx context.Context, arg InventoryCleanupPara
 	return err
 }
 
+const inventoryCleanupVcenter = `-- name: InventoryCleanupVcenter :exec
+DELETE FROM "Inventory"
+WHERE "Vcenter" = ?1
+RETURNING Iid, Name, Vcenter, VmId, EventKey, CloudId, CreationTime, DeletionTime, ResourcePool, VmType, Datacenter, Cluster, Folder, ProvisionedDisk, InitialVcpus, InitialRam, IsTemplate, PoweredOn, SrmPlaceholder, VmUuid
+`
+
+func (q *Queries) InventoryCleanupVcenter(ctx context.Context, vc string) error {
+	_, err := q.db.ExecContext(ctx, inventoryCleanupVcenter, vc)
+	return err
+}
+
 const inventoryMarkDeleted = `-- name: InventoryMarkDeleted :exec
 UPDATE "Inventory"
 SET "DeletionTime" = ?1