implement message subscribing and worker

This commit is contained in:
2026-02-14 19:17:05 +01:00
parent 4cdfa1e096
commit 0dadaf1280
19 changed files with 376 additions and 11 deletions

View File

@@ -20,7 +20,15 @@ public static class AipsRegistrationExtensions
services.AddSingleton<IRabbitMqConnection, RabbitMqConnection>();
services.AddSingleton<IMessagePublisher, RabbitMqPublisher>();
services.AddSingleton<IMessageSubscriber, RabbitMqSubscriber>();
return services;
}
public static IServiceCollection AddAipsMessageHandlers(this IServiceCollection services)
{
services.AddMessageHandlersFromAssembly(typeof(IMessage).Assembly);
return services;
}
}

View File

@@ -0,0 +1,36 @@
using System.Reflection;
using AipsCore.Application.Abstract.MessageBroking;
using Microsoft.Extensions.DependencyInjection;
namespace AipsCore.Infrastructure.DI;
public static class MessageHandlerRegistrationExtensions
{
public static IServiceCollection AddMessageHandlersFromAssembly(this IServiceCollection services, Assembly assembly)
{
var handlerInterface = typeof(IMessageHandler<>);
var types = assembly.GetTypes()
.Where(t => t is { IsAbstract: false, IsInterface: false });
foreach (var type in types)
{
var interfaces = type.GetInterfaces();
foreach (var @interface in interfaces)
{
if (!@interface.IsGenericType)
continue;
var genericDef = @interface.GetGenericTypeDefinition();
if (handlerInterface != genericDef)
continue;
services.AddTransient(@interface, type);
}
}
return services;
}
}

View File

@@ -0,0 +1,86 @@
using System.Text;
using System.Text.Json;
using AipsCore.Application.Abstract.MessageBroking;
using AipsCore.Infrastructure.DI.Configuration;
using Microsoft.Extensions.Configuration;
using RabbitMQ.Client;
using RabbitMQ.Client.Events;
namespace AipsCore.Infrastructure.MessageBroking.RabbitMQ;
public class RabbitMqSubscriber : IMessageSubscriber
{
private readonly IRabbitMqConnection _connection;
private readonly IConfiguration _configuration;
public RabbitMqSubscriber(IRabbitMqConnection connection, IConfiguration configuration)
{
_connection = connection;
_configuration = configuration;
}
public async Task SubscribeAsync<T>(Func<T, CancellationToken, Task> handler)
{
var channel = await _connection.CreateChannelAsync();
await channel.ExchangeDeclareAsync(
exchange: GetExchange(),
type: ExchangeType.Topic,
durable: true);
await channel.QueueDeclareAsync(
queue: GetQueueName<T>(),
durable: true);
await channel.QueueBindAsync(
queue: GetQueueName<T>(),
exchange: GetExchange(),
routingKey: GetQueueName<T>());
var consumer = new AsyncEventingBasicConsumer(channel);
consumer.ReceivedAsync += async (sender, args) =>
{
var message = Deserialize<T>(args.Body.ToArray());
try
{
await handler(message, CancellationToken.None);
await channel.BasicAckAsync(args.DeliveryTag, multiple: false);
}
catch (Exception exception)
{
Console.Error.WriteLine(exception);
await channel.BasicNackAsync(args.DeliveryTag, multiple: false, requeue: false);
}
};
await channel.BasicConsumeAsync(
queue: GetQueueName<T>(),
autoAck: false,
consumer: consumer);
}
private string GetQueueName<T>()
{
return typeof(T).Name;
}
private string GetExchange()
{
return _configuration.GetEnvRabbitMqExchange();
}
private T Deserialize<T>(byte[] bytes)
{
var message = JsonSerializer.Deserialize<T>(bytes);
if (message is null)
{
throw new Exception();
}
return message;
}
}