Add ListRuleTypesReferencesByDataSource db query (#5057)

Signed-off-by: Radoslav Dimitrov <[email protected]>
mindersec · Nov 26, 2024 · ef132da · ef132da
1 parent e51bae0
commit ef132da
Show file tree

Hide file tree

Showing 4 changed files with 60 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
@@ -71,3 +71,9 @@ WHERE project_id = ANY(sqlc.arg(projects)::uuid[]);
 SELECT * FROM data_sources_functions
 WHERE data_source_id = $1 AND project_id = $2;
 
+-- ListRuleTypesReferencesByDataSource retrieves all rule types
+-- referencing a given data source in a given project.
+--
+-- name: ListRuleTypesReferencesByDataSource :many
+SELECT * FROM rule_type_data_sources
+WHERE data_sources_id = $1 AND project_id = $2;
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