fix: Add guards to migrations

julep-ai · Jan 9, 2025 · c923f78 · c923f78
1 parent 69ce438
commit c923f78
Show file tree

Hide file tree

Showing 5 changed files with 112 additions and 31 deletions.
diff --git a/memory-store/migrations/000012_transitions.down.sql b/memory-store/migrations/000012_transitions.down.sql
@@ -14,6 +14,16 @@ DROP INDEX IF EXISTS idx_transitions_next;
 DROP INDEX IF EXISTS idx_transitions_current;
 
 -- Drop the transitions table (this will also remove it from hypertables)
+DO $$
+BEGIN
+    BEGIN
+        DELETE FROM transitions;
+    EXCEPTION
+        WHEN others THEN
+            RAISE NOTICE 'An error occurred during deleting all from transitions: %, %', SQLSTATE, SQLERRM;
+    END;
+END $$;
+
 DROP TABLE IF EXISTS transitions;
 
 -- Drop custom types if they exist

diff --git a/memory-store/migrations/000012_transitions.up.sql b/memory-store/migrations/000012_transitions.up.sql
@@ -158,7 +158,7 @@ END;
 $$ LANGUAGE plpgsql;
 
 -- Create a trigger on the transitions table
-CREATE TRIGGER validate_transition BEFORE INSERT ON transitions FOR EACH ROW
+CREATE OR REPLACE TRIGGER validate_transition BEFORE INSERT ON transitions FOR EACH ROW
 EXECUTE FUNCTION check_valid_transition ();
 
 COMMIT;
diff --git a/memory-store/migrations/000013_executions_continuous_view.down.sql b/memory-store/migrations/000013_executions_continuous_view.down.sql
@@ -1,8 +1,16 @@
 BEGIN;
 
 -- Drop the continuous aggregate policy
-SELECT
-    remove_continuous_aggregate_policy ('latest_transitions');
+DO $$
+BEGIN
+    BEGIN
+        SELECT
+            remove_continuous_aggregate_policy ('latest_transitions');
+    EXCEPTION
+        WHEN others THEN
+            RAISE NOTICE 'An error occurred during remove_continuous_aggregate_policy(latest_transitions): %, %', SQLSTATE, SQLERRM;
+    END;
+END $$;
 
 -- Drop the views
 DROP VIEW IF EXISTS latest_executions;

diff --git a/memory-store/migrations/000017_compression.down.sql b/memory-store/migrations/000017_compression.down.sql
@@ -1,17 +1,49 @@
 BEGIN;
 
-SELECT
-    remove_compression_policy ('entries');
+DO $$
+BEGIN
+    BEGIN
+        SELECT
+            remove_compression_policy ('entries');
+    EXCEPTION
+        WHEN others THEN
+            RAISE NOTICE 'An error occurred during remove_compression_policy(entries): %, %', SQLSTATE, SQLERRM;
+    END;
+END $$;
 
-SELECT
-    remove_compression_policy ('transitions');
+DO $$
+BEGIN
+    BEGIN
+        SELECT
+            remove_compression_policy ('transitions');
+    EXCEPTION
+        WHEN others THEN
+            RAISE NOTICE 'An error occurred during remove_compression_policy(transitions): %, %', SQLSTATE, SQLERRM;
+    END;
+END $$;
 
-ALTER TABLE entries
-SET
-    (timescaledb.compress = FALSE);
+DO $$
+BEGIN
+    BEGIN
+        ALTER TABLE entries
+        SET
+            (timescaledb.compress = FALSE);
+    EXCEPTION
+        WHEN others THEN
+            RAISE NOTICE 'An error occurred during unsetting entries.compress: %, %', SQLSTATE, SQLERRM;
+    END;
+END $$;
 
-ALTER TABLE transitions
-SET
-    (timescaledb.compress = FALSE);
+DO $$
+BEGIN
+    BEGIN
+        ALTER TABLE transitions
+        SET
+            (timescaledb.compress = FALSE);
+    EXCEPTION
+        WHEN others THEN
+            RAISE NOTICE 'An error occurred during unsetting transitions.compress: %, %', SQLSTATE, SQLERRM;
+    END;
+END $$;
 
 COMMIT;
diff --git a/memory-store/migrations/000017_compression.up.sql b/memory-store/migrations/000017_compression.up.sql
@@ -7,26 +7,57 @@
 
 BEGIN;
 
-ALTER TABLE entries
-SET
-    (
-        timescaledb.compress = TRUE,
-        timescaledb.compress_segmentby = 'session_id',
-        timescaledb.compress_orderby = 'created_at DESC, entry_id DESC'
-    );
+DO $$
+BEGIN
+    BEGIN
+        ALTER TABLE entries
+            SET (
+                timescaledb.compress = TRUE,
+                timescaledb.compress_segmentby = 'session_id',
+                timescaledb.compress_orderby = 'created_at DESC, entry_id DESC'
+            );
+    EXCEPTION
+        WHEN others THEN
+            RAISE NOTICE 'An error occurred during entries.compress: %, %', SQLSTATE, SQLERRM;
+    END;
+END $$;
 
-SELECT
-    add_compression_policy ('entries', INTERVAL '7 days');
+DO $$
+BEGIN
+    BEGIN
+        SELECT
+            add_compression_policy ('entries', INTERVAL '7 days');
+    EXCEPTION
+        WHEN others THEN
+            RAISE NOTICE 'An error occurred during add_compression_policy(entries): %, %', SQLSTATE, SQLERRM;
+    END;
+END $$;
 
-ALTER TABLE transitions
-SET
-    (
-        timescaledb.compress = TRUE,
-        timescaledb.compress_segmentby = 'execution_id',
-        timescaledb.compress_orderby = 'created_at DESC, transition_id DESC'
-    );
+DO $$
+BEGIN
+    BEGIN
+        ALTER TABLE transitions
+        SET
+            (
+                timescaledb.compress = TRUE,
+                timescaledb.compress_segmentby = 'execution_id',
+                timescaledb.compress_orderby = 'created_at DESC, transition_id DESC'
+            );
+    EXCEPTION
+        WHEN others THEN
+            RAISE NOTICE 'An error occurred during transitions.compress: %, %', SQLSTATE, SQLERRM;
+    END;
+END $$;
 
-SELECT
-    add_compression_policy ('transitions', INTERVAL '7 days');
+DO $$
+BEGIN
+    BEGIN
+        SELECT
+            add_compression_policy ('transitions', INTERVAL '7 days');
+    EXCEPTION
+        WHEN others THEN
+            RAISE NOTICE 'An error occurred during add_compression_policy(transitions): %, %', SQLSTATE, SQLERRM;
+    END;
+END $$;
 
 COMMIT;