Add cascade delete to kv store table fk

smartcontractkit · Mar 28, 2024 · 5c724be · 5c724be
1 parent ffd4922
commit 5c724be
Show file tree

Hide file tree

Showing 2 changed files with 25 additions and 0 deletions.
diff --git a/core/services/job/kv_orm_test.go b/core/services/job/kv_orm_test.go
@@ -82,4 +82,6 @@ func TestJobKVStore(t *testing.T) {
 	require.NoError(t, kvStore.Store(key, td2))
 	require.NoError(t, kvStore.Get(key, &retData))
 	require.Equal(t, td2, retData)
+
+	require.NoError(t, jobORM.DeleteJob(jobID))
 }
diff --git a/core/store/migrate/migrations/0229_add_kv_store_job_fk_cascade_delete.sql b/core/store/migrate/migrations/0229_add_kv_store_job_fk_cascade_delete.sql
@@ -0,0 +1,23 @@
+-- +goose Up
+
+BEGIN;
+
+ALTER TABLE job_kv_store DROP CONSTRAINT job_kv_store_job_id_fkey;
+ALTER TABLE job_kv_store
+    ADD CONSTRAINT job_kv_store_job_id_fkey
+        FOREIGN KEY (job_id)
+            REFERENCES jobs(id)
+            ON DELETE CASCADE;
+
+COMMIT;
+
+-- +goose Down
+BEGIN;
+
+ALTER TABLE job_kv_store DROP CONSTRAINT job_kv_store_job_id_fkey;
+ALTER TABLE job_kv_store
+    ADD CONSTRAINT job_kv_store_job_id_fkey
+        FOREIGN KEY (job_id)
+            REFERENCES jobs(id);
+
+COMMIT;