Files
parr_api/PARR.BLL/Services/Implementations/MqService.cs

167 lines
6.8 KiB
C#
Raw Normal View History

2023-09-21 09:37:36 +10:00
using Microsoft.Extensions.Logging;
using PARR.BLL.Contracts.Interfaces;
using PARR.BLL.Domain.Mq;
using PARR.BLL.Services.Interfaces;
using RabbitMQ.Client;
2023-09-21 15:21:33 +10:00
using RabbitMQ.Client.Events;
using System.Text;
2023-09-21 09:37:36 +10:00
namespace PARR.BLL.Services.Implementations
{
2023-09-26 11:01:20 +10:00
2023-09-21 09:37:36 +10:00
internal class MqService : IMqService
{
2023-09-21 09:37:36 +10:00
private readonly ILogger<MqService> logger;
2023-09-21 16:38:38 +10:00
private IConnection? connection;
private IModel? channel;
2023-09-21 09:37:36 +10:00
public MqService(ILogger<MqService> logger)
{
this.logger = logger;
}
2023-09-21 09:37:36 +10:00
public MqSendResult Send(IMqSettings mqSettings, string[] msgList)
{
var factory = new ConnectionFactory
{
2023-09-21 09:37:36 +10:00
HostName = mqSettings.HostName,
UserName = mqSettings.User,
Password = mqSettings.Password
};
2023-09-21 09:37:36 +10:00
// индекс текущей отправки в очередь из массива msgList
// нужен для формирования списка неотправленных сообщений
var sendIdx = 0;
try
{
using (var connection = factory.CreateConnection())
using (var channel = connection.CreateModel())
{
//https://www.rabbitmq.com/lazy-queues.html
//Рекомендовано при работе порциями использовать ленивые очереди. сообщения не используют память
//Формируем соответствующий аргумент
var args = new Dictionary<string, object> { { "x-queue-mode", "lazy" } };
//Объявляем очередь с которой будем работать.
//Если такой очереди ещё нет, то создатся.
//Если нет, нужно параметры типа durable должны совпадать иначе будет ошибка.
//В целом эти параметры можно посмотреть в админке RabbitMQ
channel.QueueDeclare(
2023-09-21 09:37:36 +10:00
queue: mqSettings.QueueName,
durable: true,
exclusive: false,
autoDelete: false,
arguments: args
);
var props = channel.CreateBasicProperties();
//храним на диске
props.DeliveryMode = 2;
//время жизни, мс
//props.Expiration = "60000";
2023-09-21 09:37:36 +10:00
foreach (var msg in msgList)
{
var body = Encoding.UTF8.GetBytes(msg);
channel.BasicPublish(exchange: "", routingKey: mqSettings.QueueName, basicProperties: props, body: body);
2023-09-21 09:37:36 +10:00
sendIdx++;
2023-09-21 15:37:38 +10:00
logger.LogDebug($"Отправлено сообщение в очередь: {mqSettings.HostName} {mqSettings.QueueName}, {msg}");
2023-09-21 09:37:36 +10:00
}
}
2023-09-21 09:37:36 +10:00
return new MqSendResult { IsSuccess = true };
}
catch (Exception ex)
{
2023-09-21 09:37:36 +10:00
var notSendMsg = new List<string>();
// формируем список неотправленных элементов
for (int i = sendIdx; i < msgList.Length; i++)
notSendMsg.Add(msgList[i]);
2023-09-21 15:37:38 +10:00
logger.LogError(ex, $"Ошибка при отправке сообщения в очередь {mqSettings.HostName} {mqSettings.QueueName}, {string.Join(',', notSendMsg)}");
2023-09-21 09:37:36 +10:00
return new MqSendResult { IsSuccess = false, NotSendMessages = notSendMsg.ToArray() };
}
}
2023-09-21 15:21:33 +10:00
2023-09-26 11:01:20 +10:00
public bool InitConsumer(IMqSettings mqSettings, MqMessageHandlerDelegate messageHandler)
2023-09-21 15:21:33 +10:00
{
logger.LogInformation($"Устанавливаю соединение с RabbitMQ: {mqSettings.HostName}, {mqSettings.QueueName}");
var facory = new ConnectionFactory
{
HostName = mqSettings.HostName,
UserName = mqSettings.User,
Password = mqSettings.Password,
AutomaticRecoveryEnabled = true,
DispatchConsumersAsync = true
};
try
{
2023-09-21 16:38:38 +10:00
//using var connection = facory.CreateConnection();
//using var channel = connection.CreateModel();
connection = facory.CreateConnection();
channel = connection.CreateModel();
2023-09-21 15:21:33 +10:00
var args = new Dictionary<string, object> { { "x-queue-mode", "lazy" } };
channel.QueueDeclare(
queue: mqSettings.QueueName,
durable: true,
exclusive: false,
autoDelete: false,
arguments: args
);
logger.LogInformation($"Соединение с RabbitMQ установлено: {mqSettings.HostName}, {mqSettings.QueueName}");
channel.CallbackException += (ch, ea) =>
{
var _channel = (IModel?)ch;
// может тут если канал упал переоткрывать его
logger.LogError(ea.Exception, $"Ошибка при обработке очереди MqService сообщений из очереди. Канал открыт: {_channel?.IsOpen}");
};
2023-09-21 15:21:33 +10:00
var consumer = new AsyncEventingBasicConsumer(channel);
consumer.Received += async (ch, ea) =>
{
var content = Encoding.UTF8.GetString(ea.Body.ToArray());
logger.LogInformation($"Получено сообщение: {content}");
2023-09-21 15:21:33 +10:00
2023-09-26 11:01:20 +10:00
await messageHandler.Invoke(content);
2023-09-21 15:21:33 +10:00
channel.BasicAck(ea.DeliveryTag, false);
await Task.Yield();
};
2023-09-26 11:01:20 +10:00
string consumerTag = channel.BasicConsume(mqSettings.QueueName, false, consumer);
2023-09-21 15:21:33 +10:00
return true;
2023-09-21 15:21:33 +10:00
}
catch (Exception ex)
{
logger.LogError(ex, $"Ошибка при получении сообщений из очереди: {mqSettings.HostName}, {mqSettings.QueueName}");
return false;
2023-09-21 15:21:33 +10:00
}
}
2023-09-21 16:38:38 +10:00
public void Dispose()
{
channel?.Close();
connection?.Close();
channel?.Dispose();
connection?.Dispose();
}
}
}