kanshan/dify
1
0
Fork 0
mirror of https://github.com/langgenius/dify.git synced 2026-01-14 06:07:33 +08:00
Code Issues Actions 8 Packages Projects Releases Wiki Activity

refactor: Use Repository Pattern for Model Layer (#27663)

Browse Source
This commit is contained in:
heyszt 2025-11-04 09:53:22 +08:00 committed by GitHub
parent 56047f638f
commit 3c23375607
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 16 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
api/models/model.py
View File

@ -1219,9 +1219,13 @@ class Message(Base):
@property @property
def workflow_run(self): def workflow_run(self):
if self.workflow_run_id: if self.workflow_run_id:
from .workflow import WorkflowRun from sqlalchemy.orm import sessionmaker
return db.session.query(WorkflowRun).where(WorkflowRun.id == self.workflow_run_id).first() from repositories.factory import DifyAPIRepositoryFactory
session_maker = sessionmaker(bind=db.engine, expire_on_commit=False)
repo = DifyAPIRepositoryFactory.create_api_workflow_run_repository(session_maker)
return repo.get_workflow_run_by_id_without_tenant(run_id=self.workflow_run_id)
return None return None

11
api/models/workflow.py
View File

@ -1058,7 +1058,16 @@ class WorkflowAppLog(Base):
@property @property
def workflow_run(self): def workflow_run(self):
return db.session.get(WorkflowRun, self.workflow_run_id) if self.workflow_run_id:
from sqlalchemy.orm import sessionmaker
from repositories.factory import DifyAPIRepositoryFactory
session_maker = sessionmaker(bind=db.engine, expire_on_commit=False)
repo = DifyAPIRepositoryFactory.create_api_workflow_run_repository(session_maker)
return repo.get_workflow_run_by_id_without_tenant(run_id=self.workflow_run_id)
return None
@property @property
def created_by_account(self): def created_by_account(self):