add workflow_permanent_id unique constraint and index to workflows table (#309)

This commit is contained in:
Shuchang Zheng
2024-05-13 22:57:15 -04:00
committed by GitHub
parent b959822c6e
commit a4ed6de34c
2 changed files with 60 additions and 3 deletions

View File

@@ -1,6 +1,19 @@
import datetime
from sqlalchemy import JSON, Boolean, Column, DateTime, Enum, ForeignKey, Index, Integer, Numeric, String, UnicodeText
from sqlalchemy import (
JSON,
Boolean,
Column,
DateTime,
Enum,
ForeignKey,
Index,
Integer,
Numeric,
String,
UnicodeText,
UniqueConstraint,
)
from sqlalchemy.ext.asyncio import AsyncAttrs
from sqlalchemy.orm import DeclarativeBase
@@ -122,6 +135,10 @@ class ArtifactModel(Base):
class WorkflowModel(Base):
__tablename__ = "workflows"
__table_args__ = (
UniqueConstraint("organization_id", "workflow_permanent_id", "version", name="uc_org_permanent_id_version"),
Index("permanent_id_version_idx", "workflow_permanent_id", "version"),
)
workflow_id = Column(String, primary_key=True, index=True, default=generate_workflow_id)
organization_id = Column(String, ForeignKey("organizations.organization_id"))
@@ -133,8 +150,8 @@ class WorkflowModel(Base):
modified_at = Column(DateTime, default=datetime.datetime.utcnow, onupdate=datetime.datetime.utcnow, nullable=False)
deleted_at = Column(DateTime, nullable=True)
workflow_permanent_id = Column(String, default=generate_workflow_permanent_id)
version = Column(Integer, default=1)
workflow_permanent_id = Column(String, nullable=False, default=generate_workflow_permanent_id, index=True)
version = Column(Integer, default=1, nullable=False)
class WorkflowRunModel(Base):