Infer CREATE SCHEMA migrations

xataio · Jun 17, 2024 · 7ef79d2 · 7ef79d2
1 parent fca5abd
commit 7ef79d2
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/pkg/state/state.go b/pkg/state/state.go
@@ -286,7 +286,11 @@ BEGIN
 			RETURN;
 		END IF;
 
-		SELECT schema_name INTO schemaname FROM pg_catalog.pg_event_trigger_ddl_commands() WHERE schema_name IS NOT NULL;
+		IF tg_tag != 'CREATE SCHEMA' THEN
+			SELECT schema_name INTO schemaname FROM pg_catalog.pg_event_trigger_ddl_commands() WHERE schema_name IS NOT NULL;
+		ELSE
+			SELECT object_identity INTO schemaname FROM pg_event_trigger_ddl_commands();
+		END IF;
 	END IF;
 
 	IF schemaname IS NULL THEN