yumoqing/dagflow
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

bugfix

Browse Source
This commit is contained in:
yumoqing 2026-02-13 15:46:17 +08:00
parent 975f14ed30
commit de796b7d7e
2 changed files with 9 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
dagflow/dagflow.py
View File

@ -108,7 +108,6 @@ class FlowEngine:
env = ServerEnv() env = ServerEnv()
self.aio = [] self.aio = []
async with (env, 'workflow') as sor: async with (env, 'workflow') as sor:
fid = env.uuid()
self.aio = await sor.R('flow_instance', {'status': 'running'}) self.aio = await sor.R('flow_instance', {'status': 'running'})
while True: while True:
for r in self.aio: for r in self.aio:

11
dagflow/init.py
View File

@ -1,10 +1,17 @@
from functools import partial
from ahserver.serverenv import ServerEnv from ahserver.serverenv import ServerEnv
from ahserver.configuredServer import add_startup from ahserver.configuredServer import add_cleanupctx
from .dagflow import FlowEngine from .dagflow import FlowEngine
async def dagbacktask(engine, app):
task = asyncio.create_task(engine.bgtask())
yield
task.cancel()
def load_dagflow(): def load_dagflow():
engine = FlowEngine() engine = FlowEngine()
add_startup(engine.bgtask) f = partial(dagbacktask, engine)
add_cleanupctx(f)
env = ServerEnv() env = ServerEnv()
env.workflow_engine = engine env.workflow_engine = engine