Commit 100fb0c5 authored by takatost's avatar takatost

optimize workflow db connections

parent b75cd251
...@@ -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