chore: integration keys sqlc (#3293)

* integration key sqlc queries * remove statement preparation * tidy up queries * require db/tx
target · Sep 27, 2023 · a4c2aac · a4c2aac
1 parent 62fb4b4
commit a4c2aac
Show file tree

Hide file tree

Showing 12 changed files with 335 additions and 88 deletions.
diff --git a/app/initstores.go b/app/initstores.go
@@ -220,10 +220,7 @@ func (app *App) initStores(ctx context.Context) error {
 	}
 
 	if app.IntegrationKeyStore == nil {
-		app.IntegrationKeyStore, err = integrationkey.NewStore(ctx, app.db)
-	}
-	if err != nil {
-		return errors.Wrap(err, "init integration key store")
+		app.IntegrationKeyStore = integrationkey.NewStore(ctx, app.db)
 	}
 
 	if app.ScheduleRuleStore == nil {

diff --git a/devtools/pgdump-lite/pgd/db.go b/devtools/pgdump-lite/pgd/db.go
diff --git a/devtools/pgdump-lite/pgd/models.go b/devtools/pgdump-lite/pgd/models.go
diff --git a/devtools/pgdump-lite/pgd/queries.sql.go b/devtools/pgdump-lite/pgd/queries.sql.go
diff --git a/gadb/queries.sql.go b/gadb/queries.sql.go
diff --git a/graphql2/graphqlapp/integrationkey.go b/graphql2/graphqlapp/integrationkey.go
@@ -29,7 +29,7 @@ func (m *Mutation) CreateIntegrationKey(ctx context.Context, input graphql2.Crea
 			Name:      input.Name,
 			Type:      integrationkey.Type(input.Type),
 		}
-		key, err = m.IntKeyStore.CreateKeyTx(ctx, tx, key)
+		key, err = m.IntKeyStore.Create(ctx, tx, key)
 		return err
 	})
 	return key, err

diff --git a/graphql2/graphqlapp/mutation.go b/graphql2/graphqlapp/mutation.go
@@ -261,7 +261,7 @@ func (a *Mutation) tryDeleteAll(ctx context.Context, input []assignment.RawTarge
 		case assignment.TargetTypeEscalationPolicy:
 			err = errors.Wrap(a.PolicyStore.DeleteManyPoliciesTx(ctx, tx, ids), "delete escalation policies")
 		case assignment.TargetTypeIntegrationKey:
-			err = errors.Wrap(a.IntKeyStore.DeleteManyTx(ctx, tx, ids), "delete integration keys")
+			err = errors.Wrap(a.IntKeyStore.DeleteMany(ctx, tx, ids), "delete integration keys")
 		case assignment.TargetTypeSchedule:
 			err = errors.Wrap(a.ScheduleStore.DeleteManyTx(ctx, tx, ids), "delete schedules")
 		case assignment.TargetTypeCalendarSubscription:

diff --git a/integrationkey/queries.sql b/integrationkey/queries.sql
@@ -0,0 +1,39 @@
+-- name: IntKeyGetServiceID :one
+SELECT
+    service_id
+FROM
+    integration_keys
+WHERE
+    id = $1
+    AND type = $2;
+
+-- name: IntKeyCreate :exec
+INSERT INTO integration_keys(id, name, type, service_id)
+    VALUES ($1, $2, $3, $4);
+
+-- name: IntKeyFindOne :one
+SELECT
+    id,
+    name,
+    type,
+    service_id
+FROM
+    integration_keys
+WHERE
+    id = $1;
+
+-- name: IntKeyFindByService :many
+SELECT
+    id,
+    name,
+    type,
+    service_id
+FROM
+    integration_keys
+WHERE
+    service_id = $1;
+
+-- name: IntKeyDelete :exec
+DELETE FROM integration_keys
+WHERE id = ANY (@ids::uuid[]);
+