Skip to content

D
dify
Commit 100fb0c5 authored by takatost's avatar takatost
Browse files
Options

optimize workflow db connections

parent b75cd251
Expand all Hide whitespace changes
Inline Side-by-side
Showing with 105 additions and 96 deletions
api/core/app/apps/workflow_based_generate_task_pipeline.py
View file @ 100fb0c5
...@@ -87,6 +87,7 @@ class WorkflowBasedGenerateTaskPipeline: ...@@ -87,6 +87,7 @@ class WorkflowBasedGenerateTaskPipeline:
workflow_run.finished_at = datetime.utcnow() workflow_run.finished_at = datetime.utcnow()
db.session.commit() db.session.commit()
db.session.refresh(workflow_run)
db.session.close() db.session.close()
return workflow_run return workflow_run
...@@ -115,6 +116,7 @@ class WorkflowBasedGenerateTaskPipeline: ...@@ -115,6 +116,7 @@ class WorkflowBasedGenerateTaskPipeline:
workflow_run.finished_at = datetime.utcnow() workflow_run.finished_at = datetime.utcnow()
db.session.commit() db.session.commit()
db.session.refresh(workflow_run)
db.session.close() db.session.close()
return workflow_run return workflow_run
...@@ -185,6 +187,7 @@ class WorkflowBasedGenerateTaskPipeline: ...@@ -185,6 +187,7 @@ class WorkflowBasedGenerateTaskPipeline:
workflow_node_execution.finished_at = datetime.utcnow() workflow_node_execution.finished_at = datetime.utcnow()
db.session.commit() db.session.commit()
db.session.refresh(workflow_node_execution)
db.session.close() db.session.close()
return workflow_node_execution return workflow_node_execution
...@@ -205,6 +208,7 @@ class WorkflowBasedGenerateTaskPipeline: ...@@ -205,6 +208,7 @@ class WorkflowBasedGenerateTaskPipeline:
workflow_node_execution.finished_at = datetime.utcnow() workflow_node_execution.finished_at = datetime.utcnow()
db.session.commit() db.session.commit()
db.session.refresh(workflow_node_execution)
db.session.close() db.session.close()
return workflow_node_execution return workflow_node_execution
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment