ds8-solution-pro/ds-wms-service/DS.WMS.Core/Op/Method/TaskInteraction/ActionManagerService.cs

using DS.Module.Core;
using DS.Module.Core.Data;
using DS.WMS.Core.Op.Dtos.TaskInteraction;
using DS.WMS.Core.Op.Entity.TaskInteraction;
using DS.WMS.Core.Op.Interface.TaskInteraction;
using DS.WMS.Core.TaskPlat;

namespace DS.WMS.Core.Op.Method.TaskInteraction
{
    /// <summary>
    /// 动作执行管理
    /// </summary>
    public class ActionManagerService : ServiceBase, IActionManagerService
    {
        Dictionary<TaskActionType, IActionExecutor> ExecutorMappings;

        /// <summary>
        /// 初始化
        /// </summary>
        public ActionManagerService(IServiceProvider serviceProvider) : base(serviceProvider)
        {
            ExecutorMappings = new Dictionary<TaskActionType, IActionExecutor>();
            ExecutorMappings[TaskActionType.Mail] = new MailActionExecutor();
        }

        /// <summary>
        /// 触发任务执行动作
        /// </summary>
        /// <param name="businessTask">任务信息</param>
        /// <returns></returns>
        public async Task TriggerAction(BusinessTask businessTask)
        {
            ArgumentNullException.ThrowIfNull(businessTask, nameof(businessTask));

            //目前只限制任务完成才触发
            if (businessTask.TaskStatus != TaskStatusEnum.Complete)
                return;

            // 执行自动化操作
            TaskFlowDataContext dataContext = new(
                (TaskFlowDataNameConst.BusinessTask, businessTask)
            );

            TaskFlowRuner taskFlow = new(TenantDb, ServiceProvider);
            await taskFlow.RunWithBsno(businessTask.TaskType, businessTask.BusinessId, dataContext);
        }

        public async Task ExecuteAsync(TaskFlowDataContext dataContext)
        {
            ArgumentNullException.ThrowIfNull(dataContext, nameof(dataContext));

            TaskActionType actionType = (TaskActionType)dataContext.Get<int>("ActionType");
            if (ExecutorMappings.TryGetValue(actionType, out IActionExecutor executor))
            {
                var context = new ActionExecutionContext
                {
                    TaskInfo = dataContext.Get<BusinessTask>(TaskFlowDataNameConst.BusinessTask),
                    ServiceProvider = ServiceProvider,
                    TenantDb = TenantDb
                };
                foreach (var key in dataContext.Keys)
                {
                    context.AdditionalData[key] = dataContext[key];
                }

                await executor.ExecuteAsync(context);
            }
        }
    }

    /// <summary>
    /// 任务执行类型
    /// </summary>
    public enum TaskActionType
    {
        /// <summary>
        /// 后台任务
        /// </summary>
        BackgroundWorker,

        /// <summary>
        /// 邮件
        /// </summary>
        Mail,
    }
}