Add Migration From TS (#538)

Author: kraftp

dbos/_migration.py

@@ -243,6 +243,10 @@ def get_dbos_migration_six(schema: str) -> str:
 """
 
 
+def get_dbos_migration_seven(schema: str) -> str:
+    return f"""ALTER TABLE "{schema}"."workflow_status" ADD COLUMN "owner_xid" TEXT DEFAULT NULL;"""
+
+
 def get_dbos_migrations(schema: str) -> list[str]:
     return [
         get_dbos_migration_one(schema),
@@ -251,6 +255,7 @@ def get_dbos_migrations(schema: str) -> list[str]:
         get_dbos_migration_four(schema),
         get_dbos_migration_five(schema),
         get_dbos_migration_six(schema),
+        get_dbos_migration_seven(schema),
     ]
 
 
@@ -372,6 +377,10 @@ def get_sqlite_timestamp_expr() -> str:
 ALTER TABLE streams ADD COLUMN function_id INTEGER NOT NULL DEFAULT 0;
 """
 
+sqlite_migration_seven = (
+    """ALTER TABLE workflow_status ADD COLUMN "owner_xid" TEXT DEFAULT NULL;"""
+)
+
 
 sqlite_migrations = [
     sqlite_migration_one,

dbos/_schemas/system_database.py

@@ -80,6 +80,7 @@ def set_schema(cls, schema_name: Optional[str]) -> None:
         Column("priority", Integer(), nullable=False, server_default=text("'0'::int")),
         Column("queue_partition_key", Text()),
         Column("forked_from", Text()),
+        Column("owner_xid", Text()),
         Index("workflow_status_created_at_index", "created_at"),
         Index("workflow_status_executor_id_index", "executor_id"),
         Index("workflow_status_status_index", "status"),