GIT练习

Library/DynamicFlow/Tool/TcsSignal.cs

@@ -0,0 +1,94 @@
+using System;
+using System.Collections.Concurrent;
+using System.Collections.Generic;
+using System.Linq;
+using System.Text;
+using System.Threading.Tasks;
+
+namespace Serein.DynamicFlow.Tool
+{
+    public class TcsSignalException : Exception
+    {
+        public FfState FfState { get; set; }
+        public TcsSignalException(string? message) : base(message)
+        {
+            FfState = FfState.Cancel;
+        }
+    }
+
+    public class TcsSignal<TSignal> where TSignal : struct, Enum
+    {
+
+        public ConcurrentDictionary<TSignal, Stack<TaskCompletionSource<object>>> TcsEvent { get; } = new();
+
+        // public object tcsObj = new object();
+
+        public bool TriggerSignal<T>(TSignal signal, T state)
+        {
+            if (TcsEvent.TryRemove(signal, out var waitTcss))
+            {
+                while (waitTcss.Count > 0)
+                {
+                    waitTcss.Pop().SetResult(state);
+                }
+                return true;
+            }
+            return false;
+            lock (TcsEvent)
+            {
+                
+            }
+        }
+        
+        public TaskCompletionSource<object> CreateTcs(TSignal signal)
+        {
+            
+            var tcs = new TaskCompletionSource<object>();
+            TcsEvent.GetOrAdd(signal, _ => new Stack<TaskCompletionSource<object>>()).Push(tcs);
+            return tcs;
+            lock (TcsEvent)
+            {
+                /*if(TcsEvent.TryRemove(signal, out var tcss))
+                {
+                    //tcs.TrySetException(new TcsSignalException("试图获取已存在的任务"));
+                    throw new TcsSignalException("试图获取已存在的任务");
+                }*/
+                
+               
+                /*TcsEvent.TryAdd(signal, tcs);
+                return tcs;*/
+            }
+        }
+        //public TaskCompletionSource<object> GetOrCreateTcs(TSignal signal)
+        //{
+        //    lock (tcsObj)
+        //    {
+        //        var tcs = TcsEvent.GetOrAdd(signal, _ => new TaskCompletionSource<object>());
+        //        if (tcs.Task.IsCompleted)
+        //        {
+        //            TcsEvent.TryRemove(signal, out _);
+        //            tcs = new TaskCompletionSource<object>();
+        //            TcsEvent[signal] = tcs;
+        //        }
+        //        return tcs;
+        //    }
+        //}
+
+        public void CancelTask()
+        {
+            lock(TcsEvent)
+            {
+
+                foreach (var tcss in TcsEvent.Values)
+                {
+                    while (tcss.Count > 0)
+                    {
+                        tcss.Pop().SetException(new TcsSignalException("Task Cancel"));
+                    }
+                }
+                TcsEvent.Clear();
+            }
+        }
+
+    }
+}