ZT-CP-WCS/成品库/WCS.WorkEngineering/Extensions/TaskExtension.cs

using Newtonsoft.Json;
using ServiceCenter.Extensions;
using ServiceCenter.Logs;
using ServiceCenter.Redis;
using ServiceCenter.SqlSugars;
using SqlSugar;
using WCS.Entity;
using TaskStatus = WCS.Entity.TaskStatus;

namespace WCS.WorkEngineering.Extensions
{
    /// <summary>
    /// 任务扩展
    /// </summary>
    public static class TaskExtension
    {
        /// <summary>
        /// 更新任务执行记录
        /// </summary>
        /// <param name="task">任务信息</param>
        /// <param name="db">数据库上下文</param>
        /// <param name="curPoint">当前地址</param>
        /// <param name="desc">描述</param>
        public static void AddWCS_TASK_DTL(this WCS_TaskInfo task, SqlSugarHelper db, string curPoint, string desc) => task.AddWCS_TASK_DTL(db, curPoint, "", desc);

        /// <summary>
        /// 更新任务执行记录
        /// </summary>
        /// <param name="task">任务信息</param>
        /// <param name="db">数据库上下文</param>
        /// <param name="curPoint">当前地址</param>
        /// <param name="NextPoint">下一个地址</param>
        /// <param name="desc">描述</param>
        public static void AddWCS_TASK_DTL(this WCS_TaskInfo task, SqlSugarHelper db, string curPoint, string NextPoint, string desc)
        {
            db.Default.Insertable(new WCS_TaskDtl
            {
                ID = Guid.NewGuid(),
                ParentTaskCode = task.ID,
                CurPoint = curPoint,
                NextPoint = NextPoint,
                Desc = desc,
                AddWho = "WCS"
            }).SplitTable().ExecuteCommand();
        }

        public static void AddWCS_TASK_DTL1(this WCS_TaskInfo task, SqlSugarHelper db, string curPoint, string NextPoint, string desc)
        {
            db.Default.Insertable(new WCS_TaskDtl
            {
                ID = Guid.NewGuid(),
                ParentTaskCode = task.ID,
                CurPoint = curPoint,
                NextPoint = NextPoint,
                Desc = desc,
                AddWho = "WCS"
            }).SplitTable().ExecuteCommand();
            task.UpdateableOldTask(db);
        }


        /// <summary>
        ///  更新历史表数据
        /// </summary>
        /// <param name="taskInfo"></param>
        /// <param name="db"></param>
        public static void UpdateableOldTask(this WCS_TaskInfo taskInfo, SqlSugarHelper db)
        {
            if (taskInfo.Status == TaskStatus.NewBuild) return;

            // 同步任务信息
            var taskOld = db.Default.Queryable<WCS_TaskOld>().Where(v => v.ID == taskInfo.ID).SplitTable(tabs => tabs.Take(2)).ToList().OrderByDescending(v => v.AddTime).First();
            if (taskOld is not null)
            {
                if (taskInfo.Status >= TaskStatus.Finish) taskInfo.CompleteOrCancelTasks(db);
                else
                {
                    taskOld = taskInfo.Mapper<WCS_TaskOld, WCS_TaskInfo>();
                    taskOld.ID = taskInfo.ID;
                    db.Default.Updateable(taskOld).Where(x => x.ID == taskOld.ID).SplitTable(tabs => tabs.Take(2)).ExecuteCommand();
                }
            }
            else
            {
                throw new KnownException($"WCS_TaskOld表中不存在任务:{taskInfo.ID}，无法执行WCS_TaskInfo与WCS_TaskOld同步动作", LogLevelEnum.Mid);
            }
        }

        /// <summary>
        /// 完成或取消任务
        /// </summary>
        /// <param name="taskInfo"></param>
        /// <param name="db"></param>
        /// <param name="world"></param>
        public static void CompleteOrCancelTasks(this WCS_TaskInfo taskInfo, SqlSugarHelper db)
        {
            if (taskInfo.Status is not Entity.TaskStatus.Finish and not Entity.TaskStatus.Cancel) throw new KnownException("任务未完成或取消，无法执行WCS_TaskInfo与WCS_TaskOld同步动作", LogLevelEnum.Mid);
            else // 任务完成或取消,进行相关同步动作
            {
                WCS_TaskOld taskOld = db.Default.Queryable<WCS_TaskOld>().SplitTable(tabs => tabs.Take(2)).Where(v => v.ID == taskInfo.ID).OrderByDescending(v => v.AddTime).First();
                if (taskOld is null) throw new KnownException($"WCS_TaskOld表中不存在任务:{taskInfo.ID}，无法执行WCS_TaskInfo与WCS_TaskOld同步动作", LogLevelEnum.Mid);
                else
                {
                    //状态
                    taskOld.Status = taskInfo.Status;
                    taskOld.Uploaded = taskInfo.Uploaded;
                    //任务号
                    taskOld.WmsTask = taskInfo.WmsTask;
                    taskOld.AgvTaskID = taskInfo.AgvTaskID;
                    taskOld.TaskGroupKey = taskInfo.TaskGroupKey;
                    taskOld.Priority = taskInfo.Priority;
                    //时间与用户
                    taskOld.AddTime = taskInfo.AddTime;
                    taskOld.AddWho = taskInfo.AddWho;
                    taskOld.StartTime = taskInfo.StartTime;
                    taskOld.EndTime = taskInfo.EedTime;
                    taskOld.EditWho = taskInfo.EditWho;
                    taskOld.EditTime = taskInfo.EditTime;
                    //条码与地址
                    taskOld.BarCode = taskInfo.BarCode;
                    taskOld.MaterialCode = taskInfo.MaterialCode;
                    taskOld.AddrFrom = taskInfo.AddrFrom;
                    taskOld.AddrTo = taskInfo.AddrTo;
                    taskOld.LastInteractionPoint = taskInfo.LastInteractionPoint;
                    taskOld.AddrNext = taskInfo.AddrNext;
                    taskOld.Tunnel = taskInfo.Tunnel;
                    taskOld.Floor = taskInfo.Floor;
                    taskOld.Device = taskInfo.Device;
                    taskOld.SrmStation = taskInfo.SrmStation;
                    //码垛信息
                    taskOld.DocID = taskInfo.DocID;
                    taskOld.PalletType = taskInfo.PalletType;
                    taskOld.ProdLine = taskInfo.ProdLine;
                    taskOld.FullQty = taskInfo.FullQty;
                    //外形信息
                    taskOld.Height = taskInfo.Height;
                    taskOld.Length = taskInfo.Length;
                    taskOld.Weight = taskInfo.Weight;
                    //手动处理备注
                    taskOld.ManualRemarks = taskInfo.ManualRemarks;

                    //更新任务历史表,删除任务当前表
                    db.Default.Updateable(taskOld).SplitTable(tas=>tas.Take(3)).ExecuteCommand();
                    db.Default.Deleteable(taskInfo).ExecuteCommand();
                }
            }
        }
        public static void CompleteTasks(this WCS_TaskInfo taskInfo, SqlSugarHelper db)
        {
            if (taskInfo.Status is not Entity.TaskStatus.Finish and not Entity.TaskStatus.Cancel) throw new KnownException("任务未完成或取消，无法执行WCS_TaskInfo与WCS_TaskOld同步动作", LogLevelEnum.Mid);
            else // 任务完成或取消,进行相关同步动作
            {
                WCS_TaskOld taskOld = db.Default.Queryable<WCS_TaskOld>().SplitTable(tabs => tabs.Take(2)).Where(v => v.ID == taskInfo.ID).OrderByDescending(v => v.AddTime).First();
                if (taskOld is null) throw new KnownException($"WCS_TaskOld表中不存在任务:{taskInfo.ID}，无法执行WCS_TaskInfo与WCS_TaskOld同步动作", LogLevelEnum.Mid);
                else
                {
                    //状态
                    taskOld.Status = taskInfo.Status;
                    taskOld.Uploaded = taskInfo.Uploaded;
                    //任务号
                    taskOld.WmsTask = taskInfo.WmsTask;
                    taskOld.AgvTaskID = taskInfo.AgvTaskID;
                    taskOld.TaskGroupKey = taskInfo.TaskGroupKey;
                    taskOld.Priority = taskInfo.Priority;
                    //时间与用户
                    taskOld.AddTime = taskInfo.AddTime;
                    taskOld.AddWho = taskInfo.AddWho;
                    taskOld.StartTime = taskInfo.StartTime;
                    taskOld.EndTime = taskInfo.EedTime;
                    taskOld.EditWho = taskInfo.EditWho;
                    taskOld.EditTime = taskInfo.EditTime;
                    //条码与地址
                    taskOld.BarCode = taskInfo.BarCode;
                    taskOld.MaterialCode = taskInfo.MaterialCode;
                    taskOld.AddrFrom = taskInfo.AddrFrom;
                    taskOld.AddrTo = taskInfo.AddrTo;
                    taskOld.LastInteractionPoint = taskInfo.LastInteractionPoint;
                    taskOld.AddrNext = taskInfo.AddrNext;
                    taskOld.Tunnel = taskInfo.Tunnel;
                    taskOld.Floor = taskInfo.Floor;
                    taskOld.Device = taskInfo.Device;
                    taskOld.SrmStation = taskInfo.SrmStation;
                    //码垛信息
                    taskOld.DocID = taskInfo.DocID;
                    taskOld.PalletType = taskInfo.PalletType;
                    taskOld.ProdLine = taskInfo.ProdLine;
                    taskOld.FullQty = taskInfo.FullQty;
                    //外形信息
                    taskOld.Height = taskInfo.Height;
                    taskOld.Length = taskInfo.Length;
                    taskOld.Weight = taskInfo.Weight;
                    //手动处理备注
                    taskOld.ManualRemarks = taskInfo.ManualRemarks;

                    //更新任务历史表,删除任务当前表
                    db.Default.Updateable(taskOld).SplitTable(tas => tas.Take(3)).ExecuteCommand();
                }
            }
        }

        public static void CancelTasks(this WCS_TaskInfo taskInfo, SqlSugarHelper db)
        {
             db.Default.Deleteable(taskInfo).ExecuteCommand();
        }
        ///// <summary>
        /////  获取出库任务
        ///// </summary>
        ///// <param name="taskInfo">任务</param>
        ///// <param name="db">db</param>
        ///// <param name="allOutCode">可用出库站台</param>
        ///// <param name="floor">楼层</param>
        ///// <param name="obj">堆垛机</param>
        ///// <param name="index">递归次数</param>
        ///// <returns></returns>
        ///// <exception cref="KnownException"></exception>
        //public static WCS_TaskInfo GetOutTask(this WCS_TaskInfo taskInfo, SqlSugarHelper db, List<string> allOutCode, int floor, SRM obj, int index = 1)
        //{
        //    return task;
        //}

        /// <summary>
        /// 获取AGV任务ID
        /// </summary>
        /// <param name="db">db</param>
        /// <returns></returns>
        public static int GetAGVTaskId(this SqlSugarHelper db)
        {
            var id = db.Default.Queryable<WCS_AgvTaskInfo>().SplitTable(v => v.Take(1)).Max(v => v.ID);
            return id + 1;
        }

        /// <summary>
        /// 更新任务执行记录
        /// </summary>
        /// <param name="task">任务信息</param>
        public static void updateRedisHash(this WCS_TaskInfo task)
        {
            RedisHub.WMS.HMSet($"Hash:{task.ID}", task.ToDic());
        }

        
    }
}