kxs-sp-net7/Common/RabbitMQClient.cs

using System;
using System.Collections.Generic;
using System.Text;
using RabbitMQ.Client;
using RabbitMQ.Client.Events;
using System.Threading;
using Infrastructure;
using Infrastructure.Model;

namespace Common
{
    public class RabbitMQClient
    {
        public readonly static RabbitMQClient Instance = new RabbitMQClient();
        string UserName,Password,HostName,VirtualHostName;
        private RabbitMQClient()
        {
            var options = App.OptionsSetting;
            RabbitMqConfigs RabbitMqConfigs = options.RabbitMqConfigs;
            UserName = RabbitMqConfigs.UserName;
            Password = RabbitMqConfigs.Password;
            HostName = RabbitMqConfigs.HostName;
            VirtualHostName = RabbitMqConfigs.VirtualHostName;
        }

        public static IConnection _connection;
        public void CreateConn()
        { 
            var factory = new ConnectionFactory()
            {
                HostName = HostName,
                UserName = UserName,
                Password = Password,
                VirtualHost = VirtualHostName
            };
            _connection = factory.CreateConnection();
        }

        #region 单对单接收

        public void StartReceive(string QueueName, Action<string> CallBack)
        {
            if (_connection == null)
            {
                CreateConn();
            }
            else if (!_connection.IsOpen)
            {
                CreateConn();
            }

            var consumer = new EventingBasicConsumer(_channel[QueueName]);
            consumer.Received += (model, ea) =>
            {
                var body = ea.Body.ToArray();
                //获取接收的数据
                var message = Encoding.UTF8.GetString(body);

                // 模拟消息处理逻辑
                try
                {
                    CallBack(message);
                    // 手动确认消息
                    _channel[QueueName].BasicAck(deliveryTag: ea.DeliveryTag, multiple: false);
                }
                catch (Exception ex)
                {
                    // 如果处理失败，可以选择拒绝消息或重新入队
                    _channel[QueueName].BasicNack(deliveryTag: ea.DeliveryTag, multiple: false, requeue: true);
                    Utils.WriteLog(ex.ToString(), "MQ异常");
                }
            };

            // 设置 autoAck = false
            _channel[QueueName].BasicConsume(queue: QueueName, autoAck: false, consumer: consumer);
        }
        #endregion

        #region 单对单发送
        public Dictionary<string, IModel> _channel = new Dictionary<string, IModel>();
        public void Conn(string QueueName)
        {
            if (_connection == null)
            {
                CreateConn();
            }
            else if (!_connection.IsOpen)
            {
                CreateConn();
            }
            var channel = _connection.CreateModel();
            channel.ExchangeDeclare("kxs_direct_ranch", "direct", true);
            channel.QueueDeclare(QueueName, true, false, false);
            channel.QueueBind(QueueName, "kxs_direct_ranch", QueueName);
            if(!_channel.ContainsKey(QueueName)) _channel.Add(QueueName, channel);
        }
        public void Push(string QueueName, string Content)
        {
            _channel[QueueName].BasicPublish("", QueueName, null, Encoding.Default.GetBytes(Content));
        }
        #endregion
    }
}