AiDOPWarehouse/server/Plugins/Admin.NET.Plugin.AiDOP/Service/S8/ExceptionEscalationBizHandler.cs

using Admin.NET.Plugin.ApprovalFlow;
using Admin.NET.Plugin.ApprovalFlow.Service;
using Admin.NET.Plugin.AiDOP.Entity.S8;

namespace Admin.NET.Plugin.AiDOP.Service.S8;

/// <summary>
/// 异常升级审批业务回调
/// BizType = "EXCEPTION_ESCALATION"
/// </summary>
public class ExceptionEscalationBizHandler : IFlowBizHandler, ITransient
{
    public string BizType => "EXCEPTION_ESCALATION";

    private readonly SqlSugarRepository<AdoS8Exception> _rep;
    private readonly SqlSugarRepository<AdoS8ExceptionTimeline> _timelineRep;

    public ExceptionEscalationBizHandler(
        SqlSugarRepository<AdoS8Exception> rep,
        SqlSugarRepository<AdoS8ExceptionTimeline> timelineRep)
    {
        _rep = rep;
        _timelineRep = timelineRep;
    }

    public async Task OnFlowStarted(long bizId, long instanceId)
    {
        var e = await _rep.GetByIdAsync(bizId) ?? throw new S8BizException("异常不存在");
        e.Status = "ESCALATED";
        e.ActiveFlowInstanceId = instanceId;
        e.ActiveFlowBizType = BizType;
        e.UpdatedAt = DateTime.Now;
        await _rep.UpdateAsync(e);
        await InsertTimelineAsync(e.Id, "ESCALATE_START", "发起升级", null, "ESCALATED", instanceId, null);
    }

    public async Task OnFlowCompleted(long bizId, long instanceId, FlowInstanceStatusEnum finalStatus, long? lastApproverId)
    {
        var e = await _rep.GetByIdAsync(bizId) ?? throw new S8BizException("异常不存在");
        e.ActiveFlowInstanceId = null;
        e.ActiveFlowBizType = null;
        e.UpdatedAt = DateTime.Now;

        if (finalStatus == FlowInstanceStatusEnum.Approved)
        {
            e.Status = "ASSIGNED";
            if (lastApproverId.HasValue)
                e.AssigneeId = lastApproverId.Value;
            await _rep.UpdateAsync(e);
            await InsertTimelineAsync(e.Id, "ESCALATE_APPROVED", "升级已确认", "ESCALATED", "ASSIGNED",
                instanceId, lastApproverId);
        }
        else if (finalStatus == FlowInstanceStatusEnum.Cancelled)
        {
            e.Status = "IN_PROGRESS";
            await _rep.UpdateAsync(e);
            await InsertTimelineAsync(e.Id, "ESCALATE_CANCELLED", "发起人撤回升级申请", "ESCALATED", "IN_PROGRESS",
                instanceId, lastApproverId);
        }
        else
        {
            e.Status = "IN_PROGRESS";
            await _rep.UpdateAsync(e);
            await InsertTimelineAsync(e.Id, "ESCALATE_REJECTED", "升级被驳回", "ESCALATED", "IN_PROGRESS",
                instanceId, lastApproverId);
        }
    }

    public async Task<Dictionary<string, object>> GetBizData(long bizId)
    {
        var e = await _rep.GetByIdAsync(bizId);
        if (e == null) return new Dictionary<string, object>();
        return new Dictionary<string, object>
        {
            ["severity"] = e.Severity ?? string.Empty,
            ["sceneCode"] = e.SceneCode ?? string.Empty,
            ["priorityLevel"] = e.PriorityLevel ?? string.Empty,
            ["factoryCode"] = e.FactoryId.ToString(),
        };
    }

    private async Task InsertTimelineAsync(long exceptionId, string code, string label,
        string? from, string? to, long? instanceId, long? approverId)
    {
        string? remark = null;
        if (instanceId.HasValue && approverId.HasValue)
            remark = $"审批实例ID: {instanceId}，审批人: {approverId}";
        else if (instanceId.HasValue)
            remark = $"审批实例ID: {instanceId}";
        else if (approverId.HasValue)
            remark = $"审批人: {approverId}";

        await _timelineRep.InsertAsync(new AdoS8ExceptionTimeline
        {
            ExceptionId = exceptionId,
            ActionCode = code,
            ActionLabel = label,
            FromStatus = from,
            ToStatus = to,
            OperatorId = approverId,
            ActionRemark = remark,
            CreatedAt = DateTime.Now
        });
    }
}