添加项目文件。
This commit is contained in:
45
HT.Cloud.Service/Event/BaseEventSource.cs
Normal file
45
HT.Cloud.Service/Event/BaseEventSource.cs
Normal file
@ -0,0 +1,45 @@
|
||||
using Jaina;
|
||||
using System.Threading;
|
||||
using System;
|
||||
using HT.Cloud.Code;
|
||||
|
||||
namespace HT.Cloud.Service.Event
|
||||
{
|
||||
public class BaseEventSource : IEventSource
|
||||
{
|
||||
public BaseEventSource()
|
||||
{
|
||||
}
|
||||
public BaseEventSource(string eventId, object payload, OperatorModel user = null)
|
||||
{
|
||||
EventId = eventId;
|
||||
Payload = payload;
|
||||
User = user;
|
||||
}
|
||||
// 自定义属性
|
||||
public string ToDoName { get; set; }
|
||||
/// <summary>
|
||||
/// 事件 Id
|
||||
/// </summary>
|
||||
public string EventId { get; set; }
|
||||
/// <summary>
|
||||
/// 事件承载(携带)数据
|
||||
/// </summary>
|
||||
public object Payload { get; set; }
|
||||
/// <summary>
|
||||
/// 事件创建时间
|
||||
/// </summary>
|
||||
public DateTime CreatedTime { get; set; } = DateTime.Now;
|
||||
/// <summary>
|
||||
/// 当前用户信息
|
||||
/// </summary>
|
||||
public OperatorModel User { get; set; }
|
||||
/// <summary>
|
||||
/// 取消任务 Token
|
||||
/// </summary>
|
||||
/// <remarks>用于取消本次消息处理</remarks>
|
||||
[Newtonsoft.Json.JsonIgnore]
|
||||
[System.Text.Json.Serialization.JsonIgnore]
|
||||
public CancellationToken CancellationToken { get; set; }
|
||||
}
|
||||
}
|
32
HT.Cloud.Service/Event/LogEventSubscriber.cs
Normal file
32
HT.Cloud.Service/Event/LogEventSubscriber.cs
Normal file
@ -0,0 +1,32 @@
|
||||
using Jaina;
|
||||
using Microsoft.Extensions.DependencyInjection;
|
||||
using System.Threading.Tasks;
|
||||
using HT.Cloud.Code;
|
||||
using HT.Cloud.Domain.SystemSecurity;
|
||||
using HT.Cloud.Service.SystemSecurity;
|
||||
|
||||
namespace HT.Cloud.Service.Event
|
||||
{
|
||||
// 实现 IEventSubscriber 接口
|
||||
public class LogEventSubscriber : IEventSubscriber
|
||||
{
|
||||
|
||||
public LogEventSubscriber()
|
||||
{
|
||||
}
|
||||
|
||||
[EventSubscribe("Log:create")]
|
||||
public async Task SendMessages(EventHandlerExecutingContext context)
|
||||
{
|
||||
var todo = (BaseEventSource)context.Source;
|
||||
var input = (LogEntity)todo.Payload;
|
||||
var user = todo.User;
|
||||
using (var serviceProvider = GlobalContext.RootServices.CreateScope())
|
||||
{
|
||||
var logService = serviceProvider.ServiceProvider.GetService<LogService>();
|
||||
await logService.WriteDbLog(input, user);
|
||||
}
|
||||
await Task.CompletedTask;
|
||||
}
|
||||
}
|
||||
}
|
58
HT.Cloud.Service/Event/MessageEventSubscriber.cs
Normal file
58
HT.Cloud.Service/Event/MessageEventSubscriber.cs
Normal file
@ -0,0 +1,58 @@
|
||||
using Jaina;
|
||||
using Microsoft.AspNetCore.SignalR;
|
||||
using System;
|
||||
using System.Collections.Generic;
|
||||
using System.Threading.Tasks;
|
||||
using HT.Cloud.Code;
|
||||
using HT.Cloud.Domain.InfoManage;
|
||||
|
||||
namespace HT.Cloud.Service.Event
|
||||
{
|
||||
// 实现 IEventSubscriber 接口
|
||||
public class MessageEventSubscriber : IEventSubscriber
|
||||
{
|
||||
private readonly IHubContext<MessageHub> _messageHub;
|
||||
private string cacheHubKey = GlobalContext.SystemConfig.ProjectPrefix + "_hubuserinfo_";
|
||||
|
||||
public MessageEventSubscriber(IHubContext<MessageHub> messageHub)
|
||||
{
|
||||
_messageHub = messageHub;
|
||||
}
|
||||
|
||||
[EventSubscribe("Message:send")] // 支持多个
|
||||
public async Task SendMessage(EventHandlerExecutingContext context)
|
||||
{
|
||||
var todo = (BaseEventSource)context.Source;
|
||||
var input = (MessageEntity)todo.Payload;
|
||||
if (!string.IsNullOrEmpty(input.companyId) && input.F_ToUserId.Length == 0)
|
||||
{
|
||||
await _messageHub.Clients.Group(input.companyId).SendAsync("ReceiveMessage", input.ToJson());
|
||||
}
|
||||
else
|
||||
{
|
||||
var users = input.F_ToUserId.Split(',');
|
||||
foreach (var item in users)
|
||||
{
|
||||
//存在就私信
|
||||
var connectionIDs = await CacheHelper.GetAsync<List<string>>(cacheHubKey + item);
|
||||
if (connectionIDs == null)
|
||||
{
|
||||
continue;
|
||||
}
|
||||
foreach (var connectionID in connectionIDs)
|
||||
{
|
||||
try
|
||||
{
|
||||
await _messageHub.Clients.Client(connectionID).SendAsync("ReceiveMessage", input.ToJson());
|
||||
}
|
||||
catch (Exception)
|
||||
{
|
||||
continue;
|
||||
}
|
||||
}
|
||||
}
|
||||
}
|
||||
await Task.CompletedTask;
|
||||
}
|
||||
}
|
||||
}
|
117
HT.Cloud.Service/Event/RabbitMQEventSourceStorer.cs
Normal file
117
HT.Cloud.Service/Event/RabbitMQEventSourceStorer.cs
Normal file
@ -0,0 +1,117 @@
|
||||
using Jaina;
|
||||
using RabbitMQ.Client;
|
||||
using RabbitMQ.Client.Events;
|
||||
using System;
|
||||
using System.Text;
|
||||
using System.Text.Json;
|
||||
using System.Threading;
|
||||
using System.Threading.Channels;
|
||||
using System.Threading.Tasks;
|
||||
|
||||
namespace HT.Cloud.Service.Event
|
||||
{
|
||||
public sealed class RabbitMQEventSourceStorer : IEventSourceStorer, IDisposable
|
||||
{
|
||||
/// <summary>
|
||||
/// 内存通道事件源存储器
|
||||
/// </summary>
|
||||
private readonly Channel<IEventSource> _channel;
|
||||
/// <summary>
|
||||
/// 通道对象
|
||||
/// </summary>
|
||||
private readonly IModel _model;
|
||||
/// <summary>
|
||||
/// 连接对象
|
||||
/// </summary>
|
||||
private readonly IConnection _connection;
|
||||
/// <summary>
|
||||
/// 路由键
|
||||
/// </summary>
|
||||
private readonly string _routeKey;
|
||||
/// <summary>
|
||||
/// 构造函数
|
||||
/// </summary>
|
||||
/// <param name="factory">连接工厂</param>
|
||||
/// <param name="routeKey">路由键</param>
|
||||
/// <param name="capacity">存储器最多能够处理多少消息,超过该容量进入等待写入</param>
|
||||
public RabbitMQEventSourceStorer(ConnectionFactory factory, string routeKey, int capacity)
|
||||
{
|
||||
// 配置通道,设置超出默认容量后进入等待
|
||||
var boundedChannelOptions = new BoundedChannelOptions(capacity)
|
||||
{
|
||||
FullMode = BoundedChannelFullMode.Wait
|
||||
};
|
||||
// 创建有限容量通道
|
||||
_channel = Channel.CreateBounded<IEventSource>(boundedChannelOptions);
|
||||
// 创建连接
|
||||
_connection = factory.CreateConnection();
|
||||
_routeKey = routeKey;
|
||||
// 创建通道
|
||||
_model = _connection.CreateModel();
|
||||
// 声明路由队列
|
||||
_model.QueueDeclare(routeKey, false, false, false, null);
|
||||
// 创建消息订阅者
|
||||
var consumer = new EventingBasicConsumer(_model);
|
||||
// 订阅消息并写入内存 Channel
|
||||
consumer.Received += (ch, ea) =>
|
||||
{
|
||||
// 读取原始消息
|
||||
var stringEventSource = Encoding.UTF8.GetString(ea.Body.ToArray());
|
||||
// 转换为 IEventSource,这里可以选择自己喜欢的序列化工具,如果自定义了 EventSource,注意属性是可读可写
|
||||
var eventSource = JsonSerializer.Deserialize<BaseEventSource>(stringEventSource);
|
||||
// 写入内存管道存储器
|
||||
_channel.Writer.WriteAsync(eventSource);
|
||||
// 确认该消息已被消费
|
||||
_model.BasicAck(ea.DeliveryTag, false);
|
||||
};
|
||||
// 启动消费者 设置为手动应答消息
|
||||
_model.BasicConsume(routeKey, false, consumer);
|
||||
}
|
||||
/// <summary>
|
||||
/// 将事件源写入存储器
|
||||
/// </summary>
|
||||
/// <param name="eventSource">事件源对象</param>
|
||||
/// <param name="cancellationToken">取消任务 Token</param>
|
||||
/// <returns><see cref="ValueTask"/></returns>
|
||||
public async ValueTask WriteAsync(IEventSource eventSource, CancellationToken cancellationToken)
|
||||
{
|
||||
// 空检查
|
||||
if (eventSource == default)
|
||||
{
|
||||
throw new ArgumentNullException(nameof(eventSource));
|
||||
}
|
||||
// 这里判断是否是 ChannelEventSource 或者 自定义的 EventSource
|
||||
if (eventSource is ChannelEventSource source)
|
||||
{
|
||||
// 序列化,这里可以选择自己喜欢的序列化工具
|
||||
var data = Encoding.UTF8.GetBytes(JsonSerializer.Serialize(source));
|
||||
// 发布
|
||||
_model.BasicPublish("", _routeKey, null, data);
|
||||
}
|
||||
else
|
||||
{
|
||||
// 这里处理动态订阅问题
|
||||
await _channel.Writer.WriteAsync(eventSource, cancellationToken);
|
||||
}
|
||||
}
|
||||
/// <summary>
|
||||
/// 从存储器中读取一条事件源
|
||||
/// </summary>
|
||||
/// <param name="cancellationToken">取消任务 Token</param>
|
||||
/// <returns>事件源对象</returns>
|
||||
public async ValueTask<IEventSource> ReadAsync(CancellationToken cancellationToken)
|
||||
{
|
||||
// 读取一条事件源
|
||||
var eventSource = await _channel.Reader.ReadAsync(cancellationToken);
|
||||
return eventSource;
|
||||
}
|
||||
/// <summary>
|
||||
/// 释放非托管资源
|
||||
/// </summary>
|
||||
public void Dispose()
|
||||
{
|
||||
_model.Dispose();
|
||||
_connection.Dispose();
|
||||
}
|
||||
}
|
||||
}
|
Reference in New Issue
Block a user