Add delete statement for rule_type_data_sources. (#5097)

This statement is necessary to allow updates on Rule Types. Since an update does not modify the ID, we can't expect cascading deletes to happen, and thus have to perform the cleanup manually.
mindersec · Dec 2, 2024 · 1bb8571 · 1bb8571
1 parent 532e3be
commit 1bb8571
Show file tree

Hide file tree

Showing 4 changed files with 36 additions and 0 deletions.
diff --git a/database/mock/store.go b/database/mock/store.go
diff --git a/database/query/datasources.sql b/database/query/datasources.sql
@@ -93,3 +93,8 @@ WHERE data_sources_id = $1 AND project_id = $2;
 INSERT INTO rule_type_data_sources (rule_type_id, data_sources_id, project_id)
 VALUES (sqlc.arg(ruleTypeID)::uuid, sqlc.arg(dataSourceID)::uuid, sqlc.arg(projectID)::uuid)
 RETURNING rule_type_id, data_sources_id, project_id;
+
+-- name: DeleteRuleTypeDataSource :exec
+DELETE FROM rule_type_data_sources
+ WHERE rule_type_id = sqlc.arg(ruleid)
+   AND project_id = sqlc.arg(projectid);
diff --git a/internal/db/datasources.sql.go b/internal/db/datasources.sql.go
diff --git a/internal/db/querier.go b/internal/db/querier.go