#!/usr/bin/env python # -*- coding: utf-8 -*- """ 外部任务记录服务模块 提供外部任务记录相关的服务方法 """ import json import uuid from typing import Dict, List, Any, Optional from sqlalchemy import select, and_, or_ import datetime from data.models.external_task_record import VWEDExternalTaskRecord, ExternalTaskTypeEnum, ExternalTaskStatusEnum from data.session import get_async_session from utils.logger import get_logger # 设置日志 logger = get_logger("service.external_task_record_service") class ExternalTaskRecordService: """ 外部任务记录服务类 提供与外部任务记录相关的方法 """ @staticmethod async def create_new_task_record( req_code: str, source_id: str, target_id: str, business_task_type: str, template_id: str, request_params: Dict[str, Any], client_ip: str = None, client_info: str = None ) -> VWEDExternalTaskRecord: """ 创建newTask任务记录 Args: req_code: 请求标识码 source_id: 来源ID target_id: 目标ID business_task_type: 业务任务类型 template_id: 任务模板ID request_params: 请求参数 client_ip: 客户端IP client_info: 客户端信息 Returns: VWEDExternalTaskRecord: 创建的外部任务记录 """ try: async with get_async_session() as session: # 生成唯一ID record_id = str(uuid.uuid4()) # 创建外部任务记录 external_record = VWEDExternalTaskRecord( id=record_id, req_code=req_code, task_type=ExternalTaskTypeEnum.NEW_TASK, task_status=ExternalTaskStatusEnum.PENDING, source_id=source_id, target_id=target_id, business_task_type=business_task_type, template_id=template_id, request_params=json.dumps(request_params, ensure_ascii=False), client_ip=client_ip, client_info=client_info, start_time=datetime.datetime.now() ) session.add(external_record) await session.commit() await session.refresh(external_record) logger.info(f"创建newTask任务记录成功: record_id={record_id}, req_code={req_code}") return external_record except Exception as e: logger.error(f"创建newTask任务记录失败: {str(e)}, req_code={req_code}") raise @staticmethod async def create_agv_scheduling_task_record( req_code: str, task_code: str, business_task_type: str, security_key: str, type_field: str, sub_type: str, area_position_code: str, area_position_name: str, position_code_path: List[Dict[str, Any]], template_id: str, request_params: Dict[str, Any], client_code: str = None, token_code: str = None, client_ip: str = None, client_info: str = None, related_req_code: str = None ) -> VWEDExternalTaskRecord: """ 创建GenAgvSchedulingTask任务记录 Args: req_code: 请求标识码 task_code: 任务代码 business_task_type: 业务任务类型 security_key: 安全密钥 type_field: 类型字段 sub_type: 子类型 area_position_code: 区域位置代码 area_position_name: 区域位置名称 position_code_path: 位置代码路径 template_id: 任务模板ID request_params: 请求参数 client_code: 客户端代码 token_code: 令牌代码 client_ip: 客户端IP client_info: 客户端信息 related_req_code: 关联的ReqCode Returns: VWEDExternalTaskRecord: 创建的外部任务记录 """ try: async with get_async_session() as session: # 生成唯一ID record_id = str(uuid.uuid4()) # 创建外部任务记录 external_record = VWEDExternalTaskRecord( id=record_id, req_code=req_code, task_type=ExternalTaskTypeEnum.GEN_AGV_SCHEDULING_TASK, task_status=ExternalTaskStatusEnum.PENDING, task_code=task_code, business_task_type=business_task_type, security_key=security_key, type_field=type_field, sub_type=sub_type, area_position_code=area_position_code, area_position_name=area_position_name, position_code_path=json.dumps(position_code_path, ensure_ascii=False), client_code=client_code, token_code=token_code, template_id=template_id, request_params=json.dumps(request_params, ensure_ascii=False), client_ip=client_ip, client_info=client_info, related_req_code=related_req_code, start_time=datetime.datetime.now() ) session.add(external_record) await session.commit() await session.refresh(external_record) logger.info(f"创建GenAgvSchedulingTask任务记录成功: record_id={record_id}, req_code={req_code}, task_code={task_code}") return external_record except Exception as e: logger.error(f"创建GenAgvSchedulingTask任务记录失败: {str(e)}, req_code={req_code}") raise @staticmethod async def update_task_record_status( req_code: str, task_status: ExternalTaskStatusEnum, task_record_id: str = None, response_code: int = None, response_message: str = None, response_row_count: int = None, response_data: Dict[str, Any] = None, error_message: str = None, error_stack: str = None ) -> bool: """ 更新外部任务记录状态 Args: req_code: 请求标识码 task_status: 任务状态 task_record_id: 内部任务记录ID response_code: 响应状态码 response_message: 响应消息 response_row_count: 响应行数 response_data: 响应数据 error_message: 错误信息 error_stack: 错误堆栈 Returns: bool: 更新是否成功 """ try: async with get_async_session() as session: # 查询外部任务记录 query = select(VWEDExternalTaskRecord).where( VWEDExternalTaskRecord.id == req_code ) result = await session.execute(query) external_record = result.scalar_one_or_none() if not external_record: logger.error(f"外部任务记录不存在: req_code={req_code}") return False # 更新状态和相关信息 response_data_json = None if response_data: try: response_data_json = json.dumps(response_data, ensure_ascii=False, default=str) except Exception as e: # logger.warning(f"序列化response_data失败: {str(e)}") response_data_json = str(response_data) external_record.update_status(task_status, error_message, response_data_json) if task_record_id: external_record.task_record_id = task_record_id if response_code is not None: external_record.response_code = response_code if response_message: external_record.response_message = json.dumps(response_message, ensure_ascii=False) if response_row_count is not None: external_record.response_row_count = response_row_count if error_stack: external_record.error_stack = error_stack await session.commit() logger.info(f"更新外部任务记录状态成功: req_code={req_code}, status={task_status}") return True except Exception as e: logger.error(f"更新外部任务记录状态失败: {str(e)}, req_code={req_code}") return False @staticmethod async def get_external_task_record(req_code: str) -> Optional[VWEDExternalTaskRecord]: """ 根据ReqCode获取外部任务记录 Args: req_code: 请求标识码 Returns: Optional[VWEDExternalTaskRecord]: 外部任务记录或None """ try: async with get_async_session() as session: query = select(VWEDExternalTaskRecord).where( VWEDExternalTaskRecord.req_code == req_code ).order_by(VWEDExternalTaskRecord.created_at.desc()).limit(1) result = await session.execute(query) return result.scalar_one_or_none() except Exception as e: logger.error(f"获取外部任务记录失败: {str(e)}, req_code={req_code}") return None @staticmethod async def get_related_tasks(req_code: str) -> List[VWEDExternalTaskRecord]: """ 获取关联的任务记录（通过req_code和related_req_code关联） Args: req_code: 请求标识码 Returns: List[VWEDExternalTaskRecord]: 关联的任务记录列表 """ try: async with get_async_session() as session: query = select(VWEDExternalTaskRecord).where( or_( VWEDExternalTaskRecord.req_code == req_code, VWEDExternalTaskRecord.related_req_code == req_code ) ) result = await session.execute(query) return result.scalars().all() except Exception as e: logger.error(f"获取关联任务记录失败: {str(e)}, req_code={req_code}") return [] @staticmethod async def get_task_records_by_task_code(task_code: str) -> List[VWEDExternalTaskRecord]: """ 根据TaskCode获取任务记录 Args: task_code: 任务代码 Returns: List[VWEDExternalTaskRecord]: 任务记录列表 """ try: async with get_async_session() as session: query = select(VWEDExternalTaskRecord).where( VWEDExternalTaskRecord.req_code == task_code ) result = await session.execute(query) return result.scalars().all() except Exception as e: logger.error(f"根据TaskCode获取任务记录失败: {str(e)}, task_code={task_code}") return [] @staticmethod async def check_task_completion_status(task_record_id: str) -> Optional[Dict[str, Any]]: """ 检查任务是否完成（通过内部任务记录ID查询） Args: task_record_id: 内部任务记录ID Returns: Optional[Dict[str, Any]]: 任务状态信息或None """ try: # 这里需要查询内部任务记录表来获取任务状态 # 可以根据实际需求实现具体的查询逻辑 from data.models.taskrecord import VWEDTaskRecord async with get_async_session() as session: query = select(VWEDTaskRecord).where( VWEDTaskRecord.id == task_record_id ) result = await session.execute(query) task_record = result.scalar_one_or_none() if not task_record: return None return { "task_record_id": task_record_id, "status": task_record.status, "ended_on": task_record.ended_on, "ended_reason": task_record.ended_reason, "state_description": task_record.state_description } except Exception as e: logger.error(f"检查任务完成状态失败: {str(e)}, task_record_id={task_record_id}") return None