feat: restore source parity and harden agent runtime

This commit is contained in:
2026-06-22 11:17:26 +08:00
parent e33f08277b
commit 0793eb82d6
596 changed files with 168879 additions and 290 deletions
@@ -0,0 +1,154 @@
#!/usr/bin/env python
# -*- coding: utf-8 -*-
"""
办理节点处理器
处理办理任务的创建和完成逻辑
"""
import logging
from typing import Dict, Any, Optional, TYPE_CHECKING
from sqlalchemy import select
from online_dev.workflow.engine.handlers.base_handler import BaseNodeHandler
if TYPE_CHECKING:
from online_dev.workflow.engine.base import ExecutionContext
logger = logging.getLogger(__name__)
class HandleHandler(BaseNodeHandler):
"""
办理节点处理器
办理节点类似审批节点,但办理完成后自动推进流程
支持:
- 任一办理(any
- 全部办理(all
- 依次办理(sequential
"""
async def execute(self, context: 'ExecutionContext', node: Dict) -> None:
"""
执行办理节点:创建办理任务
"""
from online_dev.workflow.engine.assignee_resolver import assignee_resolver
node_id = self.get_node_id(node)
node_name = self.get_node_name(node)
node_config = self.get_node_config(node)
logger.info(f"创建办理任务 - 节点: {node_id}, 名称: {node_name}")
# 解析办理人
assignee_ids = await assignee_resolver.resolve(
context.db,
node_config,
context.instance,
context.form_data,
)
if not assignee_ids:
logger.warning(f"节点 {node_id} 没有办理人,自动跳过")
await self.advance_to_next(context, node)
return
# 更新实例当前节点
await self.update_instance_node(context, node)
# 创建任务
multi_handle = node_config.get('multiHandle', 'any')
if multi_handle == 'sequential':
# 依次办理:只创建第一个人的任务
await self.create_task(context, node, assignee_ids[0], 'handle')
else:
# 任一办理/全部办理:创建所有人的任务
for assignee_id in assignee_ids:
await self.create_task(context, node, assignee_id, 'handle')
async def handle_completion(self, context: 'ExecutionContext', task: Any) -> None:
"""
处理办理完成后的流程推进
Args:
context: 执行上下文
task: 已完成的任务
"""
from online_dev.workflow.model import WorkflowTask
from online_dev.workflow.engine.utils import FlowUtils
node = FlowUtils.find_node_by_id(context.flow_definition, task.node_id)
if not node:
logger.error(f"找不到节点: {task.node_id}")
return
node_config = self.get_node_config(node)
multi_handle = node_config.get('multiHandle', 'any')
# 检查多人办理逻辑
if multi_handle == 'all':
# 全部办理:检查是否所有人都已办理
stmt = select(WorkflowTask).where(
WorkflowTask.instance_id == str(context.instance.id),
WorkflowTask.node_id == task.node_id,
WorkflowTask.status == 'pending',
)
result = await context.db.execute(stmt)
pending_tasks = result.scalars().all()
if len(pending_tasks) > 0:
logger.info(f"全部办理模式,还有 {len(pending_tasks)} 人未办理")
return
elif multi_handle == 'sequential':
# 依次办理:检查是否还有下一个人
next_assignee = await self._get_next_sequential_assignee(context, task)
if next_assignee:
await self.create_task(context, node, next_assignee, 'handle')
return
# 办理完成,推进到下一节点
await self.advance_to_next(context, node)
async def _get_next_sequential_assignee(
self,
context: 'ExecutionContext',
current_task: Any,
) -> Optional[str]:
"""获取依次办理的下一个办理人"""
from online_dev.workflow.model import WorkflowTask
from online_dev.workflow.engine.assignee_resolver import assignee_resolver
from online_dev.workflow.engine.utils import FlowUtils
node = FlowUtils.find_node_by_id(context.flow_definition, current_task.node_id)
if not node:
return None
node_config = self.get_node_config(node)
all_assignees = await assignee_resolver.resolve(
context.db,
node_config,
context.instance,
context.form_data,
)
if not all_assignees:
return None
stmt = select(WorkflowTask.assignee_id).where(
WorkflowTask.instance_id == str(context.instance.id),
WorkflowTask.node_id == current_task.node_id,
WorkflowTask.status == 'handled',
).order_by(WorkflowTask.sys_create_datetime)
result = await context.db.execute(stmt)
handled_ids = [str(uid) for uid in result.scalars().all()]
handled_set = set(handled_ids)
for assignee_id in all_assignees:
if assignee_id not in handled_set:
return assignee_id
return None