Add DB methods for selector management

Adds the needed SQL methods to manage profile selectors Fixes: mindersec#3720
jhrozek · Jun 27, 2024 · aa40ff6 · aa40ff6
1 parent 2ddd501
commit aa40ff6
Show file tree

Hide file tree

Showing 6 changed files with 519 additions and 0 deletions.
diff --git a/database/mock/store.go b/database/mock/store.go
diff --git a/database/query/selectors.sql b/database/query/selectors.sql
@@ -0,0 +1,24 @@
+-- name: CreateSelector :one
+INSERT INTO profile_selectors (profile_id, entity, selector, comment)
+VALUES ($1, $2, $3, $4)
+RETURNING id, profile_id, entity, selector, comment;
+
+-- name: GetSelectorsByProfileID :many
+SELECT id, profile_id, entity, selector, comment
+FROM profile_selectors
+WHERE profile_id = $1;
+
+-- name: UpdateSelector :one
+UPDATE profile_selectors
+SET entity = $2, selector = $3, comment = $4
+WHERE id = $1
+RETURNING id, profile_id, entity, selector, comment;
+
+-- name: DeleteSelector :exec
+DELETE FROM profile_selectors
+WHERE id = $1;
+
+-- name: GetSelectorByID :one
+SELECT id, profile_id, entity, selector, comment
+FROM profile_selectors
+WHERE id = $1;
diff --git a/internal/db/models.go b/internal/db/models.go
diff --git a/internal/db/querier.go b/internal/db/querier.go
diff --git a/internal/db/selectors.sql.go b/internal/db/selectors.sql.go