SKMC.API-v3/SKMC.API/Process/Model/ProcessStation.cs

using SKMC.Api.Common;
using SKMC.Api.Common.Tasks;
using SKMC.Api.Motion.Control;
using System;
using System.Collections.Concurrent;
using System.Collections.Generic;

namespace SKMC.Api.Process.Model
{
    /// <summary>
    /// <p>流程站点</p>
    /// <br>每个流程站点由多个ProcessStep组成一个循环流程</br>
    /// </summary>
    public abstract class ProcessStation
    {
        protected static readonly ILogger log = LogFactory.Get();

        /// <summary>
        /// 线性流程引擎
        /// </summary>
        private readonly ProcessFlow processFlow = ObjectFactory.Create<ProcessFlow>();

        /// <summary>
        /// 持久化接口
        /// </summary>
        protected readonly ProcessStorage processStorage = ObjectFactory.Create<ProcessStorage>();

        /// <summary>
        /// 流程站点管理器
        /// </summary>
        public ProcessStationManager ProcessStationManager { get; set; }

        /// <summary>
        /// 物料状态, 使用物料类型、状态值组合
        /// </summary>
        public ConcurrentDictionary<int, int> MatStatuses { get; set; } = new ConcurrentDictionary<int, int>();

        /// <summary>
        /// 流程站点的内置数据集
        /// </summary>
        public ConcurrentDictionary<string, object> DataMapping { get; set; } = new ConcurrentDictionary<string, object>();

        /// <summary>
        /// 流程动作控制接口
        /// </summary>
        protected readonly IMotionControl motionControl = ObjectFactory.Resolve<IMotionControl>();

        private int id;

        public int Id
        {
            get { return id; }
            set { id = value; processFlow.Id = value; }
        }

        public string Name { get; set; }

        /// <summary>
        /// station复位时执行
        /// </summary>
        public Action OnReset { get; set; }

        /// <summary>
        /// station复位结果, 0:复位成功
        /// </summary>
        public int ResetResult { get; set; } = -1;

        /// <summary>
        /// station启动时执行
        /// </summary>
        public Action OnStart { get; set; }

        /// <summary>
        /// station停止后执行
        /// </summary>
        public Action OnStop { get; set; }

        /// <summary>
        /// station暂停后执行
        /// </summary>
        public Action OnPause { get; set; }

        /// <summary>
        /// station暂停恢复后执行
        /// </summary>
        public Action OnResume { get; set; }

        /// <summary>
        /// station恢复状态后执行
        /// </summary>
        public Action OnRestore { get; set; }

        /// <summary>
        /// 每轮循环开始前执行
        /// </summary>
        public Action OnPrepare { set => processFlow.OnPrepare = value; }

        /// <summary>
        /// 触发入料事件时执行
        /// </summary>
        public Action Onload { get; set; }

        /// <summary>
        /// 完全收到物料准备开工时
        /// </summary>
        public Action OnReady { get; set; }

        /// <summary>
        /// 触发出料事件后执行
        /// </summary>
        public Action OnUnload { get; set; }

        /// <summary>
        /// 物料完全离开后执行
        /// </summary>
        public Action OnExit { get; set; }

        /// <summary>
        /// 是否空闲(站内没有物料)
        /// </summary>
        public bool IsIdle { get; set; }

        /// <summary>
        /// 安全运行的判断条件
        /// </summary>
        public Func<bool> SafeChecker { get; set; }

        /// <summary>
        /// 站点运行状态, 同设备状态, 见DeviceStatusEnum
        /// </summary>
        public int RunStatus { get; set; }

        /// <summary>
        /// 多工位序号, 同一类ProcessStation的多个实例, 使用该属性区分
        /// </summary>
        public short StationIndex { get; set; }

        /// <summary>
        /// 获取流程步骤
        /// </summary>
        /// <param name="processStepId"> 流程步骤Id</param>
        /// <returns></returns>
        public virtual ProcessStep Get(int processStepId) => processFlow.Get(processStepId);

        /// <summary>
        /// 注册流程步骤
        /// </summary>
        /// <param name="processStep">流程步骤</param>
        public virtual void Register(ProcessStep processStep) => processFlow.Register(processStep);

        /// <summary>
        /// 设置流程起始的步骤, 后续流程不变
        /// </summary>
        /// <param name="processStep">流程步骤</param>
        public virtual void SetFirst(ProcessStep processStep) => processFlow.SetFirst(processStep);

        /// <summary>
        /// 设置流程起始的步骤, 后续流程不变
        /// </summary>
        /// <param name="processStepId">流程步骤Id</param>
        public virtual void SetFirst(int processStepId) => processFlow.SetFirst(processStepId);

        /// <summary>
        /// 提交下个流程步骤
        /// </summary>
        public virtual void PostAutoNext() => processFlow.PostAutoNext();

        /// <summary>
        /// 提交指定的流程步骤
        /// </summary>
        /// <param name="processStep">流程步骤</param>
        public virtual void PostManual(ProcessStep processStep) => processFlow.PostManual(processStep);

        /// <summary>
        /// 提交指定的流程步骤
        /// </summary>
        /// <param name="processStepId">流程步骤Id</param>
        public virtual void PostManual(int processStepId) => processFlow.PostManual(Get(processStepId));

        /// <summary>
        /// 直接调用一个流程步骤的Actions(自动模式的下一个)
        /// </summary>
        public virtual void CallAutoNext() => processFlow.CallAutoNext();

        /// <summary>
        /// 直接调用一个流程步骤的Actions(手动模式指定)
        /// </summary>
        /// <param name="processStep">流程步骤</param>
        public virtual void CallManual(ProcessStep processStep) => processFlow.CallManual(processStep);

        /// <summary>
        /// 直接调用一个流程步骤的Actions(手动模式指定)
        /// </summary>
        /// <param name="processStepId">流程步骤Id</param>
        public virtual void CallManual(int processStepId) => processFlow.CallManual(Get(processStepId));

        /// <summary>
        /// 获取数据对象
        /// </summary>
        /// <typeparam name="T"></typeparam>
        /// <param name="key"></param>
        /// <param name="stationId">流程站点的Id, 如果是本站则为-1</param>
        /// <param name="fromStorage">是否从持久化获取, 如果为true并且从持久化获取成功, 将覆盖掉内存中已有的数据</param>
        /// <returns></returns>
        public virtual T GetData<T>(string key, int stationId = -1, bool fromStorage = false)
        {
            //if (fromStorage) LoadStationData(new[] { key }, true);
            //GetProcessStation(stationId).DataMapping.TryGetValue(key, out object dataVal);
            //return (T) dataVal;

            ProcessStation processStation = GetProcessStation(stationId);
            if (fromStorage) processStorage.LoadStationData(this, new[] { key }, true);
            if (processStation.DataMapping.ContainsKey(key))
            {
                if (processStation.DataMapping[key] != null)
                {
                    T t = (T)processStation.DataMapping[key];
                    return t;
                }
            }
            return default;
        }

        /// <summary>
        /// 获取数据对象并从数据集中删除
        /// </summary>
        /// <typeparam name="T"></typeparam>
        /// <param name="key"></param>
        /// <param name="stationId"></param>
        /// <returns></returns>
        public virtual T TakeData<T>(string key, int stationId = -1)
        {
            ProcessStation processStation = GetProcessStation(stationId);
            if (processStation.DataMapping.ContainsKey(key))
            {
                if (processStation.DataMapping[key] != null)
                {
                    T t = (T)processStation.DataMapping[key];
                    processStation.DataMapping.TryRemove(key, out _);
                    return t;
                }
            }
            //return default;
            throw new KeyNotFoundException($"TakeData Failed, can't find key[{key}] from station[{processStation.Id}]");
        }

        /// <summary>
        /// 添加数据对象
        /// </summary>
        /// <param name="key"></param>
        /// <param name="value"></param>
        /// <param name="stationId">流程站点的Id, 如果是本站则为-1</param>
        /// <param name="toStorage">是否持久化保存, 如果为true将覆盖掉持久化中已有的数据</param>
        public virtual void SetData(string key, object value, int stationId = -1, bool toStorage = false)
        {
            GetProcessStation(stationId).DataMapping.AddOrUpdate(key, value, (k, oldValue) => value);
            if (stationId == -1) stationId = Id;
            log.Debug($"Set Station[{stationId}] Data , key: {key}, value: {value}");
            if (toStorage)
            {
                processStorage.SaveStationData(this, new[] { key }, true);
            }
        }

        /// <summary>
        /// 删除数据对象
        /// </summary>
        /// <param name="key"></param>
        /// <param name="stationId">流程站点的Id, 如果是本站则为-1</param>
        /// <param name="toStorage">是否持久化同步删除</param>
        public virtual void RemoveData(string key, int stationId = -1, bool toStorage = false)
        {
            bool result = GetProcessStation(stationId).DataMapping.TryRemove(key, out _);
            if (result && toStorage) processStorage.DeleteStationData(this, new[] { key });
        }

        /// <summary>
        /// 清除本站内的数据对象
        /// </summary>
        public void ClearData(int stationId = -1, bool toStorage = false)
        {
            GetProcessStation(stationId).DataMapping.Clear();
            if (toStorage) processStorage.DeleteStationDataAll(this);
        }

        /// <summary>
        /// 检查任务中断点是否触发
        /// </summary>
        /// <param name="onCancel">触发中断后的动作</param>
        /// <returns></returns>
        public virtual bool CheckTaskCancelled(Action onCancel = null)
        {
            TaskTokener.SetWaitPoint(Id);
            if (TaskTokener.IsCancelled(Id))
            {
                onCancel?.Invoke();
                return true;
            }
            return false;
        }

        /// <summary>
        /// 启动流程
        /// </summary>
        public virtual void Start()
        {
            processFlow.Start();
            TaskTokener.Add(Id);
            OnStart?.Invoke();
            processFlow.PostFirst();
        }

        /// <summary>
        /// 停止流程
        /// <param name="hasAction">是否有后续停止动作</param>
        /// </summary>
        public virtual void Stop(bool hasAction = false)
        {
            processFlow.Stop();
            TaskTokener.Stop(Id);
            if (hasAction) OnStop?.Invoke();
        }

        /// <summary>
        /// 复位流程
        /// </summary>
        public virtual void Reset()
        {
            processFlow.Reset();
            TaskTokener.Add(Id);
            OnReset?.Invoke();
        }

        /// <summary>
        /// 暂停流程
        /// </summary>
        public virtual void Pause()
        {
            processFlow.Pause();
            TaskTokener.Pause(Id);
            OnPause?.Invoke();
        }

        /// <summary>
        /// 恢复流程
        /// </summary>
        public virtual void Resume()
        {
            processFlow.Resume();
            OnResume?.Invoke();
            TaskTokener.Resume(Id);
        }

        public virtual void Restore()
        {
            OnRestore?.Invoke();
            TaskTokener.Add(Id);
        }

        /// <summary>
        /// 获取流程站点的物料状态
        /// </summary>
        /// <param name="matType">状态类型, 可以是物料也可以是站点</param>
        /// <param name="stationId">流程站点的Id, 如果是本站则为-1</param>
        /// <param name="fromStorage">是否从持久化获取, 如果为true并且从持久化获取成功, 将覆盖掉内存中已有的数据</param>
        /// <returns></returns>
        public int GetStationStatus(int matType, int stationId = -1, bool fromStorage = false)
        {
            if (fromStorage) processStorage.LoadStationStatus(this, new[] { matType }, true);
            GetProcessStation(stationId).MatStatuses.TryGetValue(matType, out int statusVal);
            return statusVal;
        }

        /// <summary>
        /// 判断流程站点的物料状态是否在给定的物料状态值内
        /// 只要满足其中一个状态值返回true, 全部不满足返回false
        /// </summary>
        /// <param name="matType">状态类型, 可以是物料也可以是站点</param>
        /// <param name="stationId">流程站点的Id, 如果是本站则为-1</param>
        /// <param name="statusVals">需要比较的物料状态值集合</param>
        /// <returns></returns>
        public bool CheckStationStatus(int matType, int stationId, int[] statusVals)
        {
            int statusVal = GetStationStatus(matType, stationId);
            bool checkResult = false;
            foreach (int val in statusVals)
            {
                if (val == statusVal) return true;
            }
            return checkResult;
        }

        /// <summary>
        /// 设置流程站点的物料状态
        /// </summary>
        /// <param name="matType">状态类型, 可以是物料也可以是站点</param>
        /// <param name="status">状态值</param>
        /// <param name="stationId">站点Id, -1表示本站</param>
        /// <param name="toStorage">是否持久化保存, 如果为true将覆盖掉持久化中已有的数据</param>
        public void SetStationStatus(int matType, int status, int stationId = -1, bool toStorage = false)
        {
            int newStatus = GetProcessStation(stationId).MatStatuses.AddOrUpdate(matType, status, (key, oldValue) => status);
            if (stationId == -1) stationId = Id;
            log.Debug($"Set Station[{stationId}] Status, matType: {matType}, status check: {status == newStatus}, val: (expected:{status}, actual:{newStatus})");
            if (toStorage)
            {
                processStorage.SaveStationStatus(this, new[] { matType }, true);
            }
        }

        /// <summary>
        /// 删除流程站点的物料状态
        /// </summary>
        /// <param name="matType"></param>
        /// <param name="stationId"></param>
        /// <param name="toStorage">是否同步删除持久化中的物料状态</param>
        public void RemoveStationStatus(int matType, int stationId = -1, bool toStorage = false)
        {
            bool result = GetProcessStation(stationId).MatStatuses.TryRemove(matType, out _);
            if (result && toStorage) processStorage.DeleteStationStatus(this, new[] { matType });
        }

        /// <summary>
        /// 清除流程站点的所有状态
        /// </summary>
        /// <param name="stationId"></param>
        public void ClearStationStatus(int stationId = -1, bool toStorage = false)
        {
            GetProcessStation(stationId).MatStatuses.Clear();
            if (toStorage) processStorage.DeleteStationStatusAll(this);
        }

        private ProcessStation GetProcessStation(int stationId = -1)
        {
            ProcessStation processStation = (stationId == -1) ? this : ProcessStationManager.Get(stationId);
            if (processStation == null) throw new ArgumentNullException($"Cannot Find stationId[{stationId}] Object");
            return processStation;
        }
    }
}