mirror of
https://gitee.com/langsisi_admin/serein-flow
synced 2026-03-31 21:46:34 +08:00
在ChannelFlowTrigger的基础上添加了观察者模式,使其具备通知多个消费者的能力。
This commit is contained in:
@@ -1,11 +1,10 @@
|
||||
|
||||
using System;
|
||||
using System;
|
||||
using System.Collections.Concurrent;
|
||||
using System.Reactive.Subjects;
|
||||
using System.Threading;
|
||||
using System.Threading.Channels;
|
||||
using System.Threading.Tasks;
|
||||
|
||||
|
||||
namespace Serein.Library.NodeFlow.Tool
|
||||
{
|
||||
/// <summary>
|
||||
@@ -22,50 +21,69 @@ namespace Serein.Library.NodeFlow.Tool
|
||||
/// </summary>
|
||||
Overtime
|
||||
}
|
||||
|
||||
public class TriggerData
|
||||
{
|
||||
public TriggerType Type { get; set; }
|
||||
public object Value { get; set; }
|
||||
}
|
||||
|
||||
|
||||
/// <summary>
|
||||
/// 信号触发器类,带有消息广播功能
|
||||
/// </summary>
|
||||
public class ChannelFlowTrigger<TSignal> where TSignal : struct, Enum
|
||||
{
|
||||
// 使用并发字典管理每个枚举信号对应的 Channel
|
||||
private readonly ConcurrentDictionary<TSignal, Channel<TriggerData>> _channels = new ConcurrentDictionary<TSignal, Channel<TriggerData>>();
|
||||
// 使用并发字典管理每个信号对应的广播列表
|
||||
private readonly ConcurrentDictionary<TSignal, Subject<TriggerData>> _subscribers = new ConcurrentDictionary<TSignal, Subject<TriggerData>>();
|
||||
|
||||
/// <summary>
|
||||
/// 获取或创建指定信号的 Channel
|
||||
/// 获取或创建指定信号的 Subject(消息广播者)
|
||||
/// </summary>
|
||||
/// <param name="signal">枚举信号标识符</param>
|
||||
/// <returns>对应的 Channel</returns>
|
||||
private Channel<TriggerData> GetOrCreateChannel(TSignal signal)
|
||||
/// <returns>对应的 Subject</returns>
|
||||
private Subject<TriggerData> GetOrCreateSubject(TSignal signal)
|
||||
{
|
||||
return _channels.GetOrAdd(signal, _ => Channel.CreateUnbounded<TriggerData>());
|
||||
return _subscribers.GetOrAdd(signal, _ => new Subject<TriggerData>());
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// 创建信号并指定超时时间的Channel.
|
||||
/// 订阅指定信号的消息
|
||||
/// </summary>
|
||||
/// <param name="signal">枚举信号标识符</param>
|
||||
/// <param name="observer">订阅者</param>
|
||||
/// <returns>取消订阅的句柄</returns>
|
||||
public IDisposable Subscribe(TSignal signal, IObserver<TriggerData> observer)
|
||||
{
|
||||
var subject = GetOrCreateSubject(signal);
|
||||
return subject.Subscribe(observer); // 返回取消订阅的句柄
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// 创建信号并指定超时时间,触发时通知所有订阅者
|
||||
/// </summary>
|
||||
/// <param name="signal">枚举信号标识符</param>
|
||||
/// <param name="outTime">超时时间</param>
|
||||
/// <returns>等待任务</returns>
|
||||
public async Task<TriggerData> CreateChannelWithTimeoutAsync<TResult>(TSignal signal, TimeSpan outTime, TResult outValue)
|
||||
{
|
||||
var channel = GetOrCreateChannel(signal);
|
||||
var subject = GetOrCreateSubject(signal);
|
||||
var cts = new CancellationTokenSource();
|
||||
|
||||
// 异步任务:超时后自动触发信号
|
||||
_ = Task.Run(async () =>
|
||||
{
|
||||
try
|
||||
{
|
||||
await Task.Delay(outTime, cts.Token);
|
||||
if(!cts.IsCancellationRequested) // 如果还没有被取消
|
||||
if (!cts.IsCancellationRequested) // 如果还没有被取消
|
||||
{
|
||||
TriggerData triggerData = new TriggerData()
|
||||
{
|
||||
Value = outValue,
|
||||
Type = TriggerType.Overtime,
|
||||
};
|
||||
await channel.Writer.WriteAsync(triggerData);
|
||||
subject.OnNext(triggerData); // 广播给所有订阅者
|
||||
subject.OnCompleted(); // 通知订阅结束
|
||||
}
|
||||
}
|
||||
catch (OperationCanceledException)
|
||||
@@ -74,35 +92,44 @@ namespace Serein.Library.NodeFlow.Tool
|
||||
}
|
||||
finally
|
||||
{
|
||||
cts?.Cancel();
|
||||
cts?.Dispose(); // 确保 cts 被释放
|
||||
}
|
||||
}, cts.Token);
|
||||
|
||||
|
||||
// 等待信号传入(超时或手动触发)
|
||||
var result = await channel.Reader.ReadAsync();
|
||||
cts?.Cancel();
|
||||
// 返回一个等待的任务
|
||||
return await WaitForSignalAsync(signal);
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// 等待指定信号的触发
|
||||
/// </summary>
|
||||
/// <param name="signal">枚举信号标识符</param>
|
||||
/// <returns>等待任务</returns>
|
||||
public async Task<TriggerData> WaitForSignalAsync(TSignal signal)
|
||||
{
|
||||
var taskCompletionSource = new TaskCompletionSource<TriggerData>();
|
||||
var subscription = Subscribe(signal, new Observer<TriggerData>(taskCompletionSource.SetResult));
|
||||
var result = await taskCompletionSource.Task;
|
||||
subscription.Dispose(); // 取消订阅
|
||||
return result;
|
||||
}
|
||||
|
||||
/// <summary>
|
||||
/// 触发信号
|
||||
/// 手动触发信号,并广播给所有订阅者
|
||||
/// </summary>
|
||||
/// <param name="signal">枚举信号标识符</param>
|
||||
/// <returns>是否成功触发</returns>
|
||||
public bool TriggerSignal<TResult>(TSignal signal, TResult value)
|
||||
{
|
||||
if (_channels.TryGetValue(signal, out var channel))
|
||||
if (_subscribers.TryGetValue(signal, out var subject))
|
||||
{
|
||||
TriggerData triggerData = new TriggerData()
|
||||
{
|
||||
Value = value,
|
||||
Type = TriggerType.External,
|
||||
};
|
||||
// 手动触发信号
|
||||
|
||||
channel.Writer.TryWrite(triggerData);
|
||||
subject.OnNext(triggerData); // 广播给所有订阅者
|
||||
//subject.OnCompleted(); // 通知订阅结束
|
||||
return true;
|
||||
}
|
||||
return false;
|
||||
@@ -113,13 +140,32 @@ namespace Serein.Library.NodeFlow.Tool
|
||||
/// </summary>
|
||||
public void CancelAllTasks()
|
||||
{
|
||||
foreach (var channel in _channels.Values)
|
||||
foreach (var subject in _subscribers.Values)
|
||||
{
|
||||
channel.Writer.Complete();
|
||||
subject.OnCompleted(); // 通知所有订阅者结束
|
||||
}
|
||||
_channels.Clear();
|
||||
_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);
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user