624 lines
22 KiB
Python
624 lines
22 KiB
Python
|
import logging
|
|||
|
|
from typing import Dict, Any
|
|||
|
|
from services.execution.task_context import TaskContext
|
|||
|
|
from .base import BlockHandler, register_handler
|
|||
|
|
from utils.logger import get_logger
|
|||
|
|
from .model.block_name import FoundationBlockName
|
|||
|
|
# 获取日志记录器
|
|||
|
|
logger = get_logger("services.execution.handlers.foundation")
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 检查任务实例ID是否存在处理器
|
|||
|
|
@register_handler(FoundationBlockName.CHECK_TASK_RECORD_ID_IS_EXIST)
|
|||
|
|
class CheckTaskRecordIdIsExistBlockHandler(BlockHandler):
|
|||
|
|
"""检查任务实例ID是否存在处理器"""
|
|||
|
|
|
|||
|
|
async def execute(
|
|||
|
|
self,
|
|||
|
|
block: Dict[str, Any],
|
|||
|
|
input_params: Dict[str, Any],
|
|||
|
|
context: TaskContext
|
|||
|
|
) -> Dict[str, Any]:
|
|||
|
|
"""检查任务记录ID是否存在"""
|
|||
|
|
from sqlalchemy import text, select
|
|||
|
|
from data.models import VWEDTaskRecord
|
|||
|
|
from data.session import get_async_session
|
|||
|
|
import json
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
# 获取任务记录ID
|
|||
|
|
task_record_id = input_params.get("taskRecordId")
|
|||
|
|
|
|||
|
|
if not task_record_id:
|
|||
|
|
result = {
|
|||
|
|
"success": False,
|
|||
|
|
"message": "缺少任务记录ID"
|
|||
|
|
}
|
|||
|
|
# 记录执行结果
|
|||
|
|
await self._record_task_log(block, result, context)
|
|||
|
|
return result
|
|||
|
|
|
|||
|
|
logger.info(f"检查任务记录ID: {task_record_id}")
|
|||
|
|
|
|||
|
|
record = None
|
|||
|
|
exists = False
|
|||
|
|
record_info = None
|
|||
|
|
|
|||
|
|
# 查询任务记录
|
|||
|
|
async with get_async_session() as session:
|
|||
|
|
try:
|
|||
|
|
# 使用ORM查询
|
|||
|
|
query = select(VWEDTaskRecord).where(VWEDTaskRecord.id == task_record_id)
|
|||
|
|
result = await session.execute(query)
|
|||
|
|
record = result.scalar_one_or_none()
|
|||
|
|
|
|||
|
|
if record:
|
|||
|
|
logger.info(f"找到任务记录: id={record.id}, def_id={record.def_id}, status={record.status}")
|
|||
|
|
exists = True
|
|||
|
|
|
|||
|
|
# 记录详细信息
|
|||
|
|
record_info = {
|
|||
|
|
"id": record.id,
|
|||
|
|
"def_id": record.def_id,
|
|||
|
|
"status": record.status,
|
|||
|
|
"def_label": record.def_label,
|
|||
|
|
"created_on": record.created_on.isoformat() if record.created_on else None,
|
|||
|
|
"ended_on": record.ended_on.isoformat() if record.ended_on else None,
|
|||
|
|
}
|
|||
|
|
else:
|
|||
|
|
logger.info(f"未找到任务记录ID: {task_record_id}")
|
|||
|
|
|
|||
|
|
except Exception as db_error:
|
|||
|
|
logger.error(f"数据库查询错误: {str(db_error)}")
|
|||
|
|
# 使用原生SQL作为备选
|
|||
|
|
# 设置上下文变量
|
|||
|
|
context.set_block_output(block.get("name"), {"taskRecordIdIsExist": exists})
|
|||
|
|
|
|||
|
|
# 记录任务信息（如果找到）
|
|||
|
|
if record_info:
|
|||
|
|
context.set_variable("taskRecordInfo", record_info)
|
|||
|
|
context.set_block_output(block.get("name"), {"taskRecordInfo": record_info})
|
|||
|
|
|
|||
|
|
exec_result = {
|
|||
|
|
"success": True,
|
|||
|
|
"message": f"任务记录ID {'存在' if exists else '不存在'}",
|
|||
|
|
"output": {
|
|||
|
|
"exists": exists,
|
|||
|
|
"taskRecordId": task_record_id,
|
|||
|
|
"recordInfo": record_info
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
# 记录执行结果
|
|||
|
|
await self._record_task_log(block, exec_result, context)
|
|||
|
|
return exec_result
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f"检查任务记录ID是否存在失败: {str(e)}")
|
|||
|
|
result = {
|
|||
|
|
"success": False,
|
|||
|
|
"message": f"检查任务记录ID是否存在失败: {str(e)}"
|
|||
|
|
}
|
|||
|
|
# 记录异常
|
|||
|
|
await self._record_task_log(block, result, context)
|
|||
|
|
return result
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 释放资源处理器
|
|||
|
|
@register_handler(FoundationBlockName.RELEASE_RESOURCE)
|
|||
|
|
class ReleaseResourceBlockHandler(BlockHandler):
|
|||
|
|
"""释放资源处理器"""
|
|||
|
|
|
|||
|
|
async def execute(
|
|||
|
|
self,
|
|||
|
|
block: Dict[str, Any],
|
|||
|
|
input_params: Dict[str, Any],
|
|||
|
|
context: TaskContext
|
|||
|
|
) -> Dict[str, Any]:
|
|||
|
|
"""释放资源"""
|
|||
|
|
from sqlalchemy import select
|
|||
|
|
from data.models import VWEDTaskRecord
|
|||
|
|
from data.session import get_async_session
|
|||
|
|
|
|||
|
|
task_record_id = context.task_record_id
|
|||
|
|
query = select(VWEDTaskRecord).where(VWEDTaskRecord.id == task_record_id)
|
|||
|
|
async with get_async_session() as session:
|
|||
|
|
try:
|
|||
|
|
result = await session.execute(query)
|
|||
|
|
record = result.scalar_one()
|
|||
|
|
if record:
|
|||
|
|
record.allow_restart_same_location = True
|
|||
|
|
await session.commit()
|
|||
|
|
logger.info(f"任务记录ID: {task_record_id} 释放资源成功")
|
|||
|
|
else:
|
|||
|
|
logger.error(f"未找到任务记录ID: {task_record_id}")
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f"释放资源失败: {str(e)}")
|
|||
|
|
result = {
|
|||
|
|
"success": False,
|
|||
|
|
"message": f"释放资源失败: {str(e)}"
|
|||
|
|
}
|
|||
|
|
await self._record_task_log(block, result, context)
|
|||
|
|
return result
|
|||
|
|
result = {
|
|||
|
|
"success": True,
|
|||
|
|
"message": "释放资源成功",
|
|||
|
|
"output": {}
|
|||
|
|
}
|
|||
|
|
await self._record_task_log(block, result, context)
|
|||
|
|
return result
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 创建唯一ID处理器
|
|||
|
|
@register_handler(FoundationBlockName.CREATE_UUID)
|
|||
|
|
class CreateUuidBlockHandler(BlockHandler):
|
|||
|
|
"""创建唯一ID处理器"""
|
|||
|
|
|
|||
|
|
async def execute(
|
|||
|
|
self,
|
|||
|
|
block: Dict[str, Any],
|
|||
|
|
input_params: Dict[str, Any],
|
|||
|
|
context: TaskContext
|
|||
|
|
) -> Dict[str, Any]:
|
|||
|
|
"""创建唯一ID"""
|
|||
|
|
import uuid
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
# 生成UUID
|
|||
|
|
generated_uuid = str(uuid.uuid4())
|
|||
|
|
|
|||
|
|
# 设置上下文变量
|
|||
|
|
result = {
|
|||
|
|
"success": True,
|
|||
|
|
"message": "创建唯一ID成功",
|
|||
|
|
"output": {
|
|||
|
|
"createUuid": generated_uuid,
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
context.set_block_output(block.get("name"), {"createUuid": generated_uuid})
|
|||
|
|
# 记录执行结果
|
|||
|
|
await self._record_task_log(block, result, context)
|
|||
|
|
return result
|
|||
|
|
except Exception as e:
|
|||
|
|
result = {
|
|||
|
|
"success": False,
|
|||
|
|
"message": f"创建唯一ID失败: {str(e)}"
|
|||
|
|
}
|
|||
|
|
# 记录异常
|
|||
|
|
await self._record_task_log(block, result, context)
|
|||
|
|
return result
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 打印块处理器
|
|||
|
|
@register_handler(FoundationBlockName.PRINT)
|
|||
|
|
class PrintBlockHandler(BlockHandler):
|
|||
|
|
"""打印块处理器"""
|
|||
|
|
|
|||
|
|
async def execute(
|
|||
|
|
self,
|
|||
|
|
block: Dict[str, Any],
|
|||
|
|
input_params: Dict[str, Any],
|
|||
|
|
context: TaskContext
|
|||
|
|
) -> Dict[str, Any]:
|
|||
|
|
"""执行打印块"""
|
|||
|
|
try:
|
|||
|
|
# 获取消息参数
|
|||
|
|
message = input_params.get("message", "")
|
|||
|
|
|
|||
|
|
# 记录日志
|
|||
|
|
logger.info(f"任务 {context.task_record_id} 打印: {message}")
|
|||
|
|
|
|||
|
|
# 返回结果
|
|||
|
|
result = {
|
|||
|
|
"success": True,
|
|||
|
|
"message": "打印成功",
|
|||
|
|
"output": {
|
|||
|
|
"message": message
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
# 记录执行结果
|
|||
|
|
await self._record_task_log(block, result, context)
|
|||
|
|
|
|||
|
|
return result
|
|||
|
|
except Exception as e:
|
|||
|
|
result = {
|
|||
|
|
"success": False,
|
|||
|
|
"message": f"打印失败: {str(e)}"
|
|||
|
|
}
|
|||
|
|
# 记录异常
|
|||
|
|
await self._record_task_log(block, result, context)
|
|||
|
|
return result
|
|||
|
|
|
|||
|
|
# 当前时间戳处理器
|
|||
|
|
@register_handler(FoundationBlockName.CURRENT_TIME_STAMP)
|
|||
|
|
class CurrentTimeStampBlockHandler(BlockHandler):
|
|||
|
|
"""当前时间戳处理器"""
|
|||
|
|
|
|||
|
|
async def execute(
|
|||
|
|
self,
|
|||
|
|
block: Dict[str, Any],
|
|||
|
|
input_params: Dict[str, Any],
|
|||
|
|
context: TaskContext
|
|||
|
|
) -> Dict[str, Any]:
|
|||
|
|
"""获取当前时间戳"""
|
|||
|
|
import time
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
# 获取当前时间戳(毫秒)
|
|||
|
|
current_timestamp = int(time.time() * 1000)
|
|||
|
|
|
|||
|
|
# 设置上下文变量
|
|||
|
|
context.set_block_output(block.get("name"), {"currentTimeStamp": current_timestamp})
|
|||
|
|
|
|||
|
|
result = {
|
|||
|
|
"success": True,
|
|||
|
|
"message": "获取当前时间戳成功",
|
|||
|
|
"output": {
|
|||
|
|
"timestamp": current_timestamp
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
# 记录执行结果
|
|||
|
|
await self._record_task_log(block, result, context)
|
|||
|
|
return result
|
|||
|
|
except Exception as e:
|
|||
|
|
result = {
|
|||
|
|
"success": False,
|
|||
|
|
"message": f"获取当前时间戳失败: {str(e)}"
|
|||
|
|
}
|
|||
|
|
# 记录异常
|
|||
|
|
await self._record_task_log(block, result, context)
|
|||
|
|
return result
|
|||
|
|
|
|||
|
|
|
|||
|
|
@register_handler(FoundationBlockName.JDBC_EXECUTE)
|
|||
|
|
class JdbcExecuteBlockHandler(BlockHandler):
|
|||
|
|
"""执行SQL处理器"""
|
|||
|
|
|
|||
|
|
async def execute(
|
|||
|
|
self,
|
|||
|
|
block: Dict[str, Any],
|
|||
|
|
input_params: Dict[str, Any],
|
|||
|
|
context: TaskContext
|
|||
|
|
) -> Dict[str, Any]:
|
|||
|
|
"""执行SQL语句"""
|
|||
|
|
from sqlalchemy import text
|
|||
|
|
from data.session import get_async_session
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
# 获取SQL语句
|
|||
|
|
sql = input_params.get("sql")
|
|||
|
|
|
|||
|
|
if not sql:
|
|||
|
|
result = {
|
|||
|
|
"success": False,
|
|||
|
|
"message": "缺少SQL语句"
|
|||
|
|
}
|
|||
|
|
# 记录执行结果
|
|||
|
|
await self._record_task_log(block, result, context)
|
|||
|
|
return result
|
|||
|
|
|
|||
|
|
logger.info(f"执行SQL语句: {sql}")
|
|||
|
|
|
|||
|
|
# 执行SQL
|
|||
|
|
async with get_async_session() as session:
|
|||
|
|
try:
|
|||
|
|
result = await session.execute(text(sql))
|
|||
|
|
await session.commit()
|
|||
|
|
|
|||
|
|
# 获取受影响的行数
|
|||
|
|
row_count = result.rowcount if hasattr(result, 'rowcount') else 0
|
|||
|
|
|
|||
|
|
logger.info(f"SQL执行成功，影响了 {row_count} 行")
|
|||
|
|
|
|||
|
|
exec_result = {
|
|||
|
|
"success": True,
|
|||
|
|
"message": f"SQL执行成功，影响了 {row_count} 行",
|
|||
|
|
"output": {
|
|||
|
|
"rowCount": row_count
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
except Exception as db_error:
|
|||
|
|
logger.error(f"SQL执行错误: {str(db_error)}")
|
|||
|
|
exec_result = {
|
|||
|
|
"success": False,
|
|||
|
|
"message": f"SQL执行失败: {str(db_error)}"
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
# 记录执行结果
|
|||
|
|
await self._record_task_log(block, exec_result, context)
|
|||
|
|
return exec_result
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f"SQL执行失败: {str(e)}")
|
|||
|
|
result = {
|
|||
|
|
"success": False,
|
|||
|
|
"message": f"SQL执行失败: {str(e)}"
|
|||
|
|
}
|
|||
|
|
# 记录异常
|
|||
|
|
await self._record_task_log(block, result, context)
|
|||
|
|
return result
|
|||
|
|
|
|||
|
|
|
|||
|
|
|
|||
|
|
# 查询SQL处理器
|
|||
|
|
@register_handler(FoundationBlockName.JDBC_QUERY)
|
|||
|
|
class JdbcQueryBlockHandler(BlockHandler):
|
|||
|
|
"""查询SQL处理器"""
|
|||
|
|
|
|||
|
|
async def execute(
|
|||
|
|
self,
|
|||
|
|
block: Dict[str, Any],
|
|||
|
|
input_params: Dict[str, Any],
|
|||
|
|
context: TaskContext
|
|||
|
|
) -> Dict[str, Any]:
|
|||
|
|
"""执行SQL查询"""
|
|||
|
|
from sqlalchemy import text
|
|||
|
|
from data.session import get_async_session
|
|||
|
|
import json
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
# 获取SQL语句
|
|||
|
|
sql = input_params.get("sql")
|
|||
|
|
|
|||
|
|
if not sql:
|
|||
|
|
result = {
|
|||
|
|
"success": False,
|
|||
|
|
"message": "缺少SQL语句"
|
|||
|
|
}
|
|||
|
|
# 记录执行结果
|
|||
|
|
await self._record_task_log(block, result, context)
|
|||
|
|
return result
|
|||
|
|
|
|||
|
|
logger.info(f"执行SQL查询: {sql}")
|
|||
|
|
|
|||
|
|
# 执行SQL查询
|
|||
|
|
async with get_async_session() as session:
|
|||
|
|
try:
|
|||
|
|
result = await session.execute(text(sql))
|
|||
|
|
|
|||
|
|
# 处理结果
|
|||
|
|
if result.returns_rows:
|
|||
|
|
# 获取列名
|
|||
|
|
columns = result.keys()
|
|||
|
|
|
|||
|
|
# 转换为字典列表
|
|||
|
|
rows = [dict(zip(columns, row)) for row in result.fetchall()]
|
|||
|
|
|
|||
|
|
# 转换为JSON字符串
|
|||
|
|
result_json = json.dumps(rows, ensure_ascii=False, default=str)
|
|||
|
|
|
|||
|
|
# 设置上下文变量
|
|||
|
|
context.set_variable("resultSet", result_json)
|
|||
|
|
context.set_block_output(block.get("name"), {"resultSet": rows})
|
|||
|
|
|
|||
|
|
logger.info(f"SQL查询成功，返回 {len(rows)} 条记录")
|
|||
|
|
|
|||
|
|
exec_result = {
|
|||
|
|
"success": True,
|
|||
|
|
"message": f"SQL查询成功，返回 {len(rows)} 条记录",
|
|||
|
|
"output": {
|
|||
|
|
"rowCount": len(rows),
|
|||
|
|
"resultSet": rows,
|
|||
|
|
"resultJson": result_json
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
else:
|
|||
|
|
# 非查询语句，没有返回行
|
|||
|
|
context.set_variable("resultSet", "[]")
|
|||
|
|
context.set_block_output(block.get("name"), {"resultSet": []})
|
|||
|
|
|
|||
|
|
logger.info("SQL执行成功，没有返回数据")
|
|||
|
|
|
|||
|
|
exec_result = {
|
|||
|
|
"success": True,
|
|||
|
|
"message": "SQL执行成功，没有返回数据",
|
|||
|
|
"output": {
|
|||
|
|
"rowCount": 0,
|
|||
|
|
"resultSet": [],
|
|||
|
|
"resultJson": "[]"
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
except Exception as db_error:
|
|||
|
|
logger.error(f"SQL查询错误: {str(db_error)}")
|
|||
|
|
exec_result = {
|
|||
|
|
"success": False,
|
|||
|
|
"message": f"SQL查询失败: {str(db_error)}"
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
# 记录执行结果
|
|||
|
|
await self._record_task_log(block, exec_result, context)
|
|||
|
|
return exec_result
|
|||
|
|
|
|||
|
|
except Exception as e:
|
|||
|
|
logger.error(f"SQL查询失败: {str(e)}")
|
|||
|
|
result = {
|
|||
|
|
"success": False,
|
|||
|
|
"message": f"SQL查询失败: {str(e)}"
|
|||
|
|
}
|
|||
|
|
# 记录异常
|
|||
|
|
await self._record_task_log(block, result, context)
|
|||
|
|
return result
|
|||
|
|
|
|||
|
|
# 当前时间处理器
|
|||
|
|
@register_handler(FoundationBlockName.TIMESTAMP)
|
|||
|
|
class TimestampBlockHandler(BlockHandler):
|
|||
|
|
"""当前时间处理器"""
|
|||
|
|
|
|||
|
|
async def execute(
|
|||
|
|
self,
|
|||
|
|
block: Dict[str, Any],
|
|||
|
|
input_params: Dict[str, Any],
|
|||
|
|
context: TaskContext
|
|||
|
|
) -> Dict[str, Any]:
|
|||
|
|
"""获取当前时间"""
|
|||
|
|
from datetime import datetime
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
# 获取当前时间
|
|||
|
|
current_time = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
|
|||
|
|
|
|||
|
|
# 设置上下文变量
|
|||
|
|
context.set_block_output(block.get("name"), {"timestamp": current_time})
|
|||
|
|
result = {
|
|||
|
|
"success": True,
|
|||
|
|
"message": "获取当前时间成功",
|
|||
|
|
"output": {
|
|||
|
|
"timestamp": current_time
|
|||
|
|
}
|
|||
|
|
}
|
|||
|
|
# 记录执行结果
|
|||
|
|
await self._record_task_log(block, result, context)
|
|||
|
|
return result
|
|||
|
|
except Exception as e:
|
|||
|
|
result = {
|
|||
|
|
"success": False,
|
|||
|
|
"message": f"获取当前时间失败: {str(e)}"
|
|||
|
|
}
|
|||
|
|
# 记录异常
|
|||
|
|
await self._record_task_log(block, result, context)
|
|||
|
|
return result
|
|||
|
|
|
|||
|
|
# 字符串转JSON对象处理器
|
|||
|
|
@register_handler(FoundationBlockName.STRING_TO_JSON_OBJECT)
|
|||
|
|
class StringToJsonObjectBlockHandler(BlockHandler):
|
|||
|
|
"""字符串转JSON对象处理器"""
|
|||
|
|
|
|||
|
|
async def execute(
|
|||
|
|
self,
|
|||
|
|
block: Dict[str, Any],
|
|||
|
|
input_params: Dict[str, Any],
|
|||
|
|
context: TaskContext
|
|||
|
|
) -> Dict[str, Any]:
|
|||
|
|
"""将字符串转换为JSON对象"""
|
|||
|
|
import json
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
# 获取输入参数
|
|||
|
|
convert_string = input_params.get("convertString", "")
|
|||
|
|
|
|||
|
|
if not convert_string:
|
|||
|
|
result = {
|
|||
|
|
"success": False,
|
|||
|
|
"message": "转换字符串为空"
|
|||
|
|
}
|
|||
|
|
# 记录执行结果
|
|||
|
|
await self._record_task_log(block, result, context)
|
|||
|
|
return result
|
|||
|
|
|
|||
|
|
# 转换为JSON对象
|
|||
|
|
json_object = json.loads(convert_string)
|
|||
|
|
|
|||
|
|
# 设置输出参数
|
|||
|
|
output = {
|
|||
|
|
"convertObject": json_object
|
|||
|
|
}
|
|||
|
|
context.set_block_output(block.get("name"), {"convertObject": json_object})
|
|||
|
|
|
|||
|
|
result = {
|
|||
|
|
"success": True,
|
|||
|
|
"message": "字符串转换为JSON对象成功",
|
|||
|
|
"output": output
|
|||
|
|
}
|
|||
|
|
# 记录执行结果
|
|||
|
|
await self._record_task_log(block, result, context)
|
|||
|
|
return result
|
|||
|
|
except json.JSONDecodeError as e:
|
|||
|
|
result = {
|
|||
|
|
"success": False,
|
|||
|
|
"message": f"字符串解析为JSON对象失败: {str(e)}"
|
|||
|
|
}
|
|||
|
|
# 记录异常
|
|||
|
|
await self._record_task_log(block, result, context)
|
|||
|
|
return result
|
|||
|
|
except Exception as e:
|
|||
|
|
result = {
|
|||
|
|
"success": False,
|
|||
|
|
"message": f"字符串转换为JSON对象失败: {str(e)}"
|
|||
|
|
}
|
|||
|
|
# 记录异常
|
|||
|
|
await self._record_task_log(block, result, context)
|
|||
|
|
return result
|
|||
|
|
|
|||
|
|
# 字符串转JSON数组处理器
|
|||
|
|
@register_handler(FoundationBlockName.STRING_TO_JSON_ARRAY)
|
|||
|
|
class StringToJsonArrayBlockHandler(BlockHandler):
|
|||
|
|
"""字符串转JSON数组处理器"""
|
|||
|
|
|
|||
|
|
async def execute(
|
|||
|
|
self,
|
|||
|
|
block: Dict[str, Any],
|
|||
|
|
input_params: Dict[str, Any],
|
|||
|
|
context: TaskContext
|
|||
|
|
) -> Dict[str, Any]:
|
|||
|
|
"""将字符串转换为JSON数组"""
|
|||
|
|
import json
|
|||
|
|
|
|||
|
|
try:
|
|||
|
|
# 获取输入参数
|
|||
|
|
convert_string = input_params.get("convertString", "")
|
|||
|
|
|
|||
|
|
if not convert_string:
|
|||
|
|
result = {
|
|||
|
|
"success": False,
|
|||
|
|
"message": "转换字符串为空"
|
|||
|
|
}
|
|||
|
|
# 记录执行结果
|
|||
|
|
await self._record_task_log(block, result, context)
|
|||
|
|
return result
|
|||
|
|
|
|||
|
|
# 处理可能被额外引号包裹的情况
|
|||
|
|
try:
|
|||
|
|
# 首先尝试直接解析
|
|||
|
|
json_array = json.loads(convert_string)
|
|||
|
|
except json.JSONDecodeError:
|
|||
|
|
# 如果失败，尝试去除可能的外层引号
|
|||
|
|
if convert_string.startswith("'") and convert_string.endswith("'"):
|
|||
|
|
convert_string = convert_string[1:-1]
|
|||
|
|
elif convert_string.startswith('"') and convert_string.endswith('"'):
|
|||
|
|
convert_string = convert_string[1:-1]
|
|||
|
|
# 再次尝试解析
|
|||
|
|
json_array = json.loads(convert_string)
|
|||
|
|
|
|||
|
|
# 设置上下文输出
|
|||
|
|
context.set_block_output(block.get("name"), {"convertArray": json_array})
|
|||
|
|
|
|||
|
|
# 验证是否为数组
|
|||
|
|
if not isinstance(json_array, list):
|
|||
|
|
result = {
|
|||
|
|
"success": False,
|
|||
|
|
"message": "转换结果不是JSON数组"
|
|||
|
|
}
|
|||
|
|
# 记录执行结果
|
|||
|
|
await self._record_task_log(block, result, context)
|
|||
|
|
return result
|
|||
|
|
|
|||
|
|
# 设置输出参数
|
|||
|
|
output = {
|
|||
|
|
"convertArray": json_array
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
result = {
|
|||
|
|
"success": True,
|
|||
|
|
"message": "字符串转换为JSON数组成功",
|
|||
|
|
"output": output
|
|||
|
|
}
|
|||
|
|
# 记录执行结果
|
|||
|
|
await self._record_task_log(block, result, context)
|
|||
|
|
return result
|
|||
|
|
except json.JSONDecodeError as e:
|
|||
|
|
result = {
|
|||
|
|
"success": False,
|
|||
|
|
"message": f"字符串解析为JSON数组失败: {str(e)}"
|
|||
|
|
}
|
|||
|
|
# 记录异常
|
|||
|
|
await self._record_task_log(block, result, context)
|
|||
|
|
return result
|
|||
|
|
except Exception as e:
|
|||
|
|
result = {
|
|||
|
|
"success": False,
|
|||
|
|
"message": f"字符串转换为JSON数组失败: {str(e)}"
|
|||
|
|
}
|
|||
|
|
# 记录异常
|
|||
|
|
await self._record_task_log(block, result, context)
|
|||
|
|
return result
|
|||
|
|
|