using System.Net.Sockets;
|
using WIDESEAWCS_Common.HttpEnum;
|
using WIDESEAWCS_Common.TaskEnum;
|
using WIDESEAWCS_DTO.TaskInfo;
|
using WIDESEAWCS_ITaskInfoService;
|
using WIDESEAWCS_Model.Models;
|
using WIDESEAWCS_Tasks.Workflow.Abstractions;
|
|
namespace WIDESEAWCS_Tasks.Workflow
|
{
|
/// <summary>
|
/// 前缀命令处理器
|
/// </summary>
|
/// <remarks>
|
/// 迁移原 RobotMessageHandler 的 pickfinished/putfinished 分支。
|
///
|
/// 前缀命令是指以特定前缀开头的命令，后面跟随逗号分隔的参数。
|
/// 格式：{前缀},{参数1},{参数2},...
|
///
|
/// 当前支持的前缀命令：
|
/// - pickfinished: 取货完成，后面跟随完成的位置编号列表
|
/// - putfinished: 放货完成，后面跟随完成的位置编号列表
|
///
|
/// 这些命令通常包含取货或放货的位置信息，需要解析并更新状态。
|
/// </remarks>
|
public class RobotPrefixCommandHandler : IRobotPrefixCommandHandler
|
{
|
/// <summary>
|
/// 机器人任务服务
|
/// </summary>
|
/// <remarks>
|
/// 用于查询和更新任务记录。
|
/// </remarks>
|
private readonly IRobotTaskService _robotTaskService;
|
|
/// <summary>
|
/// 任务处理器
|
/// </summary>
|
/// <remarks>
|
/// 用于处理取货/放货完成时的业务逻辑，如调用拆盘/组盘 API。
|
/// </remarks>
|
private readonly RobotTaskProcessor _taskProcessor;
|
|
/// <summary>
|
/// 状态管理器
|
/// </summary>
|
/// <remarks>
|
/// 用于安全更新机器人的状态。
|
/// </remarks>
|
private readonly RobotStateManager _stateManager;
|
|
/// <summary>
|
/// Socket 网关
|
/// </summary>
|
/// <remarks>
|
/// 用于向客户端发送响应消息。
|
/// </remarks>
|
private readonly ISocketClientGateway _socketClientGateway;
|
|
/// <summary>
|
/// 构造函数
|
/// </summary>
|
/// <param name="robotTaskService">任务服务</param>
|
/// <param name="taskProcessor">任务处理器</param>
|
/// <param name="stateManager">状态管理器</param>
|
/// <param name="socketClientGateway">Socket 网关</param>
|
public RobotPrefixCommandHandler(
|
IRobotTaskService robotTaskService,
|
RobotTaskProcessor taskProcessor,
|
RobotStateManager stateManager,
|
ISocketClientGateway socketClientGateway)
|
{
|
_robotTaskService = robotTaskService;
|
_taskProcessor = taskProcessor;
|
_stateManager = stateManager;
|
_socketClientGateway = socketClientGateway;
|
}
|
|
/// <summary>
|
/// 检查消息是否为前缀命令
|
/// </summary>
|
/// <remarks>
|
/// 前缀命令必须以 "pickfinished" 或 "putfinished" 开头（不区分大小写）。
|
/// </remarks>
|
/// <param name="message">消息内容（小写形式）</param>
|
/// <returns>如果是指缀命令返回 true</returns>
|
public bool IsPrefixCommand(string message)
|
{
|
// 检查消息是否以 pickfinished 或 putfinished 开头
|
return message.StartsWith("pickfinished") || message.StartsWith("putfinished");
|
}
|
|
/// <summary>
|
/// 处理前缀命令
|
/// </summary>
|
/// <remarks>
|
/// 处理流程：
|
/// 1. 解析消息，提取位置参数
|
/// 2. 查询当前任务
|
/// 3. 根据命令类型调用相应的处理方法
|
/// 4. 回写原消息到客户端
|
///
|
/// 消息格式：{命令前缀},{位置1},{位置2},...
|
/// 示例：pickfinished,1,2,3 表示取货完成，位置 1、2、3 的货物已取走
|
/// </remarks>
|
/// <param name="message">原始消息内容</param>
|
/// <param name="state">机器人当前状态</param>
|
/// <param name="client">TCP 客户端连接，用于发送响应</param>
|
public async Task HandleAsync(string message, RobotSocketState state, TcpClient client)
|
{
|
try
|
{
|
// 按逗号分隔消息，提取命令和参数
|
// 例如：pickfinished,1,2,3 -> ["pickfinished", "1", "2", "3"]
|
var parts = message.Split(',');
|
|
// 检查消息格式是否有效：至少要有命令前缀，且状态中有当前任务
|
if (parts.Length < 1 || state.CurrentTask == null)
|
{
|
return;
|
}
|
|
// 提取命令前缀并转换为小写
|
var cmd = parts[0].ToLowerInvariant();
|
|
// 解析位置参数（跳过命令前缀，处理后面的数字）
|
// 过滤掉无法解析为数字或值为 0 的位置
|
int[] positions = parts.Skip(1)
|
.Select(p => int.TryParse(p, out int value) ? value : (int?)null) // 尝试解析为整数
|
.Where(v => v.HasValue && v.Value != 0) // 过滤掉 null 和 0
|
.Select(v => v!.Value) // 提取值（已知非 null）
|
.ToArray();
|
|
// 从数据库重新查询当前任务（确保获取最新状态）
|
var task = await _robotTaskService.Repository.QueryFirstAsync(x => x.RobotTaskId == state.CurrentTask.RobotTaskId);
|
|
// 根据命令前缀分发处理
|
if (cmd.StartsWith("pickfinished"))
|
{
|
// 处理取货完成
|
await HandlePickFinishedAsync(state, positions, task);
|
}
|
else if (cmd.StartsWith("putfinished"))
|
{
|
// 处理放货完成
|
await HandlePutFinishedAsync(state, positions, task);
|
}
|
|
// 回写原消息到客户端（保持原有行为）
|
await _socketClientGateway.SendMessageAsync(client, message);
|
}
|
catch (Exception ex)
|
{
|
// 捕获并记录异常，防止异常向上传播导致消息处理中断
|
Console.WriteLine($"RobotJob MessageReceived Error: {ex.Message}");
|
}
|
}
|
|
/// <summary>
|
/// 处理取货完成（pickfinished）命令
|
/// </summary>
|
/// <remarks>
|
/// 处理逻辑：
|
/// 1. 如果是拆盘任务，构建库存 DTO 并调用拆盘 API
|
/// 2. 更新当前动作为"取货完成"
|
/// 3. 记录取货完成的位置
|
/// 4. 更新任务状态为"机器人取货完成"
|
/// 5. 安全更新状态到 Redis
|
/// </remarks>
|
/// <param name="state">机器人当前状态</param>
|
/// <param name="positions">取货完成的位置编号数组</param>
|
/// <param name="task">机器人任务记录</param>
|
private async Task HandlePickFinishedAsync(RobotSocketState state, int[] positions, Dt_RobotTask? task)
|
{
|
// 如果是拆盘任务
|
if (state.IsSplitPallet)
|
{
|
// 构建库存 DTO，包含位置信息和托盘条码
|
var stockDTO = RobotTaskProcessor.BuildStockDTO(state, positions);
|
|
// 记录取货完成的位置
|
state.LastPickPositions = positions;
|
|
// 调用拆盘 API
|
var result = _taskProcessor.PostSplitPalletAsync(stockDTO);
|
|
// 如果 API 调用成功
|
if (result.Data.Status && result.IsSuccess)
|
{
|
// 更新当前动作为"取货完成"
|
state.CurrentAction = "PickFinished";
|
}
|
}
|
else
|
{
|
// 非拆盘任务，直接更新动作
|
state.CurrentAction = "PickFinished";
|
}
|
|
// 记录取货完成的位置（无论是否拆盘都记录）
|
state.LastPickPositions = positions;
|
|
// 如果任务存在
|
if (task != null)
|
{
|
// 更新任务状态为"机器人取货完成"
|
task.RobotTaskState = TaskRobotStatusEnum.RobotPickFinish.GetHashCode();
|
|
// 安全更新状态到 Redis，确保更新成功后再更新数据库
|
if (_stateManager.TryUpdateStateSafely(state.IPAddress, state))
|
{
|
await _robotTaskService.Repository.UpdateDataAsync(task);
|
}
|
}
|
}
|
|
/// <summary>
|
/// 处理放货完成（putfinished）命令
|
/// </summary>
|
/// <remarks>
|
/// 处理逻辑：
|
/// 1. 如果是组盘任务，构建库存 DTO 并调用组盘/换盘 API
|
/// 2. 如果组盘成功，增加任务计数
|
/// 3. 更新当前动作为"放货完成"
|
/// 4. 更新任务状态为"机器人放货完成"
|
/// 5. 安全更新状态到 Redis
|
/// </remarks>
|
/// <param name="state">机器人当前状态</param>
|
/// <param name="positions">放货完成的位置编号数组</param>
|
/// <param name="task">机器人任务记录</param>
|
private async Task HandlePutFinishedAsync(RobotSocketState state, int[] positions, Dt_RobotTask? task)
|
{
|
// 假设放货成功（如果后续 API 调用失败也不回退计数）
|
bool putSuccess = true;
|
|
// 如果是组盘任务（包含换盘）
|
if (state.IsGroupPallet)
|
{
|
// 记录放货完成的位置
|
state.LastPutPositions = positions;
|
|
// 构建库存 DTO
|
var stockDTO = RobotTaskProcessor.BuildStockDTO(state, positions);
|
|
// 根据任务类型决定调用哪个 API
|
// 换盘任务调用 ChangePalletAsync，组盘任务调用 GroupPalletAsync
|
var configKey = state.CurrentTask?.RobotTaskType == RobotTaskTypeEnum.ChangePallet.GetHashCode()
|
? nameof(ConfigKey.ChangePalletAsync)
|
: nameof(ConfigKey.GroupPalletAsync);
|
|
// 调用组盘/换盘 API
|
var result = _taskProcessor.PostGroupPalletAsync(configKey, stockDTO);
|
|
// 检查 API 返回状态
|
putSuccess = result.Data.Status && result.IsSuccess;
|
}
|
|
// 如果放货成功
|
if (putSuccess)
|
{
|
// 更新当前动作为"放货完成"
|
state.CurrentAction = "PutFinished";
|
|
// 增加任务计数（累加本次完成的数量）
|
state.RobotTaskTotalNum += positions.Length;
|
|
// 如果任务存在，同步更新任务的计数
|
if (task != null)
|
{
|
task.RobotTaskTotalNum -= positions.Length;
|
}
|
}
|
|
// 如果任务存在
|
if (task != null)
|
{
|
// 更新任务状态为"机器人放货完成"
|
task.RobotTaskState = TaskRobotStatusEnum.RobotPutFinish.GetHashCode();
|
|
// 安全更新状态到 Redis
|
if (_stateManager.TryUpdateStateSafely(state.IPAddress, state))
|
{
|
await _robotTaskService.Repository.UpdateDataAsync(task);
|
}
|
}
|
}
|
}
|
}
|
