BabyVideoService/Core/Task/NTask.cs

using System;
using System.Collections;
using System.Collections.Generic;
using System.Threading.Tasks;

namespace NBC
{
    public abstract class NTask : ITask
    {
        protected readonly Dictionary<string, object> _argsDic = new();
        protected string _errorMsg;
        protected float _progress;
        protected string TaskInfo;

        public virtual string Info
        {
            get => TaskInfo;
            set => TaskInfo = value;
        }

        public NTaskStatus Status { get; protected set; } = NTaskStatus.None;
        public virtual string ErrorMsg => _errorMsg;
        public virtual float Progress => _progress;

        public virtual bool IsRunning => Status == NTaskStatus.Running;
        public virtual bool IsDone => Status == NTaskStatus.Success || Status == NTaskStatus.Fail;

        public object this[string argsName]
        {
            get
            {
                if (_argsDic.TryGetValue(argsName, out var args)) return args;

                return null;
            }
            set => _argsDic[argsName] = value;
        }


        public virtual void Stop()
        {
            Status = NTaskStatus.None;
        }


        public virtual void Run(IRunner runner)
        {
            Reset();
            runner?.Run(this);
        }

        public NTaskStatus Process()
        {
            if (Status == NTaskStatus.None) Start();

            Status = OnProcess();
            CallUpdateListener();
            if (Status == NTaskStatus.Success)
            {
                _progress = 1;
                CallCompleteListener(Status);
            }
            else if (Status == NTaskStatus.Fail)
            {
                _progress = 1;
                CallCompleteListener(Status);
            }

            return Status;
        }

        protected virtual void OnStart()
        {
        }

        protected virtual NTaskStatus OnProcess()
        {
            return Status;
        }

        protected void Finish()
        {
            _progress = 1;
            Status = NTaskStatus.Success;
        }

        protected void Fail(string message)
        {
            _progress = 1;
            Status = NTaskStatus.Fail;
            _errorMsg = message;
        }

        private void Start()
        {
            Reset();
            Status = NTaskStatus.Running;
            _progress = 0;
            OnStart();
            CallStartListener();
        }

        #region 事件

        protected event Action<ITask> OnStartListener;
        protected event Action<ITask> OnCompleteListener;
        protected event Action<ITask> OnUpdateListener;

        public ITask OnStarted(Action<ITask> callback, bool cover = false)
        {
            if (cover)
                OnStartListener = callback;
            else
                OnStartListener += callback;

            return this;
        }

        public ITask OnUpdated(Action<ITask> callback, bool cover = false)
        {
            if (cover)
                OnUpdateListener = callback;
            else
                OnUpdateListener += callback;

            return this;
        }

        public ITask OnCompleted(Action<ITask> callback, bool cover = false)
        {
            if (cover)
                OnCompleteListener = callback;
            else
                OnCompleteListener += callback;

            return this;
        }

        protected void CallStartListener()
        {
            OnStartListener?.Invoke(this);
        }

        protected void CallCompleteListener(NTaskStatus taskStatus)
        {
            OnCompleteListener?.Invoke(this);

            _taskCompletionSource?.TrySetResult(null);
        }

        protected void CallUpdateListener()
        {
            OnUpdateListener?.Invoke(this);
        }

        #endregion

        #region 异步编程相关

        private TaskCompletionSource<object> _taskCompletionSource;

        /// <summary>
        ///     异步操作任务
        /// </summary>
        public Task Task
        {
            get
            {
                if (_taskCompletionSource == null)
                {
                    _taskCompletionSource = new TaskCompletionSource<object>();
                    if (IsDone)
                        _taskCompletionSource.SetResult(null);
                }

                return _taskCompletionSource.Task;
            }
        }

        #endregion

        #region IEnumerator

        bool IEnumerator.MoveNext()
        {
            return !IsDone;
        }

        public virtual void Reset()
        {
            Status = NTaskStatus.None;
        }

        object IEnumerator.Current => null;

        #endregion
    }
}