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.UpdatedAt = DateTime.Now;
        await _rep.UpdateAsync(e);
        await InsertTimelineAsync(e.Id, "ESCALATE_START", "发起升级", null, "ESCALATED", instanceId);
    }

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

        if (finalStatus == FlowInstanceStatusEnum.Approved)
        {
            e.Status = "ASSIGNED";
            await _rep.UpdateAsync(e);
            await InsertTimelineAsync(e.Id, "ESCALATE_APPROVED", "升级已确认", "ESCALATED", "ASSIGNED", null);
        }
        else
        {
            e.Status = "IN_PROGRESS";
            await _rep.UpdateAsync(e);
            await InsertTimelineAsync(e.Id, "ESCALATE_REJECTED", "升级被驳回", "ESCALATED", "IN_PROGRESS", null);
        }
    }

    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,
            ["sceneCode"] = e.SceneCode,
            ["priorityLevel"] = e.PriorityLevel,
        };
    }

    private async Task InsertTimelineAsync(long exceptionId, string code, string label, string? from, string? to, long? instanceId)
    {
        await _timelineRep.InsertAsync(new AdoS8ExceptionTimeline
        {
            ExceptionId = exceptionId,
            ActionCode = code,
            ActionLabel = label,
            FromStatus = from,
            ToStatus = to,
            ActionRemark = instanceId.HasValue ? $"审批实例ID: {instanceId}" : null,
            CreatedAt = DateTime.Now
        });
    }
}