serein-flow/NodeFlow/Tool/TcsSignal.cs

using System.Collections.Concurrent;
using Serein.NodeFlow;
using Serein.NodeFlow.Model;

namespace Serein.NodeFlow.Tool
{
    public class TcsSignalException : Exception
    {
        public FlowStateType FsState { get; set; }
        public TcsSignalException(string? message) : base(message)
        {
            FsState = FlowStateType.Error;
        }
    }

    public class TcsSignal<TSignal> where TSignal : struct, Enum
    {
        //public ConcurrentDictionary<TSignal, Queue<TaskCompletionSource<object>>> TcsEvent { get; } = new();
        public ConcurrentDictionary<TSignal, TaskCompletionSource<object>> TcsEvent { get; } = new();
        public ConcurrentDictionary<TSignal, object> TcsValue { get; } = new();

        /// <summary>
        /// 触发信号
        /// </summary>
        /// <typeparam name="T"></typeparam>
        /// <param name="signal">信号</param>
        /// <param name="value">传递的参数</param>
        /// <returns>是否成功触发</returns>
        public bool TriggerSignal<T>(TSignal signal, T value)
        {
            if (TcsEvent.TryRemove(signal, out var waitTcs))
            {
                waitTcs.SetResult(value);
                return true;
            }
            return false;
        }

        public TaskCompletionSource<object> CreateTcs(TSignal signal)
        {
            var tcs = TcsEvent.GetOrAdd(signal,_ = new TaskCompletionSource<object>());
            return tcs;
        }

        public void CancelTask()
        {
            lock (TcsEvent)
            {
                foreach (var tcs in TcsEvent.Values)
                {
                    tcs.SetException(new TcsSignalException("任务取消"));
                }
                TcsEvent.Clear();
            }
        }
    }
}