将FlowTrigger触发器整合成接口的形式方便替换

Library/Utils/TaskFlowTrigger.cs

@@ -0,0 +1,206 @@
+using Newtonsoft.Json.Linq;
+using Serein.Library.Api;
+using Serein.Library.Utils;
+using System;
+using System.Collections.Concurrent;
+using System.Reactive.Linq;
+using System.Reactive.Subjects;
+using System.Threading;
+using System.Threading.Channels;
+using System.Threading.Tasks;
+using System.Transactions;
+
+namespace Serein.Library
+{
+
+
+    /// <summary>
+    /// 触发类型
+    /// </summary>
+    public enum TriggerDescription
+    {
+        /// <summary>
+        /// 外部触发
+        /// </summary>
+        External,
+        /// <summary>
+        /// 超时触发
+        /// </summary>
+        Overtime,
+        /// <summary>
+        /// 触发了，但类型不一致
+        /// </summary>
+        TypeInconsistency
+    }
+
+
+    public class TriggerResult<TResult>
+    {
+        public TriggerDescription Type { get; set; }
+        public TResult Value { get; set; }
+    }
+
+    /// <summary>
+    /// 信号触发器类，带有消息广播功能。
+    /// 使用枚举作为标记，创建
+    /// </summary>
+    public class TaskFlowTrigger<TSignal> : IFlowTrigger<TSignal> where TSignal : struct, Enum
+    {
+        // 使用并发字典管理每个信号对应的广播列表
+        private readonly ConcurrentDictionary<TSignal, Subject<TriggerResult<object>>> _subscribers = new ConcurrentDictionary<TSignal, Subject<TriggerResult<object>>>();
+
+        /// <summary>
+        /// 获取或创建指定信号的 Subject（消息广播者）
+        /// </summary>
+        /// <param name="signal">枚举信号标识符</param>
+        /// <returns>对应的 Subject</returns>
+        private Subject<TriggerResult<object>> GetOrCreateSubject(TSignal signal)
+        {
+            return _subscribers.GetOrAdd(signal, _ => new Subject<TriggerResult<object>>());
+        }
+
+        /// <summary>
+        /// 订阅指定信号的消息
+        /// </summary>
+        /// <param name="signal">枚举信号标识符</param>
+        /// <param name="observer">订阅者</param>
+        /// <returns>取消订阅的句柄</returns>
+        private IDisposable Subscribe<TResult>(TSignal signal, Action<TriggerResult<object>> action)
+        {
+            IObserver<TriggerResult<object>> observer = new Observer<TriggerResult<object>>(action);
+            var subject = GetOrCreateSubject(signal);
+            return subject.Subscribe(observer); // 返回取消订阅的句柄
+        }
+
+
+       
+        /// <summary>
+        /// 等待触发器并指定超时的时间
+        /// </summary>
+        /// <typeparam name="TResult">返回值类型</typeparam>
+        /// <param name="signal">等待信号</param>
+        /// <param name="outTime">超时时间</param>
+        /// <returns></returns>
+        public async Task<TriggerResult<TResult>> WaitTriggerWithTimeoutAsync<TResult>(TSignal signal, TimeSpan outTime)
+        {
+            var subject = GetOrCreateSubject(signal);
+            var cts = new CancellationTokenSource();
+
+            // 异步任务：超时后自动触发信号
+            _ = Task.Run(async () =>
+            {
+                try
+                {
+                    await Task.Delay(outTime, cts.Token);
+                    if (!cts.IsCancellationRequested) // 如果还没有被取消
+                    {
+                        var outResult = new TriggerResult<object>()
+                        {
+                            Type = TriggerDescription.Overtime
+                        };
+                        subject.OnNext(outResult); // 广播给所有订阅者
+                        subject.OnCompleted(); // 通知订阅结束
+                    }
+                }
+                catch (OperationCanceledException)
+                {
+                    // 超时任务被取消
+                }
+                finally
+                {
+                    cts?.Dispose();  // 确保 cts 被释放
+                }
+            }, cts.Token);
+            var result = await WaitTriggerAsync<TResult>(signal); // 返回一个可以超时触发的等待任务
+            return result;
+        }
+
+        /// <summary>
+        /// 等待触发
+        /// </summary>
+        /// <typeparam name="TResult"></typeparam>
+        /// <param name="signal"></param>
+        /// <returns></returns>
+        public async Task<TriggerResult<TResult>> WaitTriggerAsync<TResult>(TSignal signal)
+        {
+            var taskCompletionSource = new TaskCompletionSource<TriggerResult<object>>();
+            var subscription = Subscribe<TResult>(signal, taskCompletionSource.SetResult);
+            var result = await taskCompletionSource.Task;
+            subscription.Dispose(); // 取消订阅
+            if(result.Value is TResult data)
+            {
+                return new TriggerResult<TResult>()
+                {
+                    Value = data,
+                    Type = TriggerDescription.External,
+                };
+            }
+            else
+            {
+                return new TriggerResult<TResult>()
+                {
+                    Type = TriggerDescription.TypeInconsistency,
+                };
+            }
+        }
+
+
+        /// <summary>
+        /// 手动触发信号，并广播给所有订阅者
+        /// </summary>
+        /// <typeparam name="TResult">触发类型</typeparam>
+        /// <param name="signal">枚举信号标识符</param>
+        /// <param name="value">传递的数据</param>
+        /// <returns>是否成功触发</returns>
+        public Task<bool> InvokeTriggerAsync<TResult>(TSignal signal, TResult value)
+        {
+            if (_subscribers.TryGetValue(signal, out var subject))
+            {
+                var result = new TriggerResult<object>()
+                {
+                    Type = TriggerDescription.External,
+                    Value = value
+                };
+                subject.OnNext(result); // 广播给所有订阅者
+                subject.OnCompleted(); // 通知订阅结束
+                return Task.FromResult(true);
+            }
+            return Task.FromResult(false);
+        }
+        /// <summary>
+        /// 取消所有任务
+        /// </summary>
+
+        public void CancelAllTrigger()
+        {
+            foreach (var subject in _subscribers.Values)
+            {
+                subject.OnCompleted(); // 通知所有订阅者结束
+            }
+            _subscribers.Clear();
+        }
+    }
+
+
+
+
+    /// <summary>
+    /// 观察者类，用于包装 Action
+    /// </summary>
+    public class Observer<T> : IObserver<T>
+    {
+        private readonly Action<T> _onNext;
+
+        public Observer(Action<T> onNext)
+        {
+            _onNext = onNext;
+        }
+
+        public void OnCompleted() { }
+        public void OnError(Exception error) { }
+        public void OnNext(T value)
+        {
+            _onNext?.Invoke(value);
+        }
+    }
+}