JYHB.SlnMesnac/SlnMesnac.Redis/RedisHandler.cs

using Microsoft.Extensions.Logging;
using ServiceStack.Messaging;
using ServiceStack.Redis;
using SlnMesnac.Config;
using StackExchange.Redis;
using System;
using System.Linq;
using System.Security.Cryptography;

#region << 版 本 注 释 >>
/*--------------------------------------------------------------------
* 版权所有 (c) 2024 WenJY 保留所有权利。
* CLR版本：4.0.30319.42000
* 机器名称：LAPTOP-E0N2L34V
* 命名空间：SlnMesnac.Redis
* 唯一标识：98350fd3-5b25-4395-a2f0-eb949cc4c6f6
*
* 创建者：WenJY
* 电子邮箱：wenjy@mesnac.com
* 创建时间：2024-04-11 16:44:24
* 版本：V1.0.0
* 描述：
*
*--------------------------------------------------------------------
* 修改人：
* 时间：
* 修改说明：
*
* 版本：V1.0.0
*--------------------------------------------------------------------*/
#endregion << 版 本 注 释 >>
namespace SlnMesnac.Redis
{
    /// <summary>
    /// Redis
    /// </summary>
    public class RedisHandler
    {
        private ILogger<RedisHandler> _logger;

        private readonly AppConfig _appConfig;

        private readonly ISubscriber _subscriber;

        public readonly ConnectionMultiplexer redis;

        public RedisHandler(AppConfig appConfig, ILogger<RedisHandler> logger)
        {
            _appConfig = appConfig;

            redis = ConnectionMultiplexer.Connect(_appConfig.redisConfig);
            _subscriber = redis.GetSubscriber();
            _logger = logger;
        }



        /// <summary>
        /// 推送消息
        /// </summary>
        /// <param name="channel"></param>
        /// <param name="message"></param>
        public void PublishMessage(string channel, string message)
        {

            // 生成唯一的消息ID
          // var messageId = $"{DateTime.UtcNow.Ticks}_{Guid.NewGuid().ToString("N")}";

            // 存储消息内容
          //  var redisDb = redis.GetDatabase();
          //  redisDb.StringSet(messageId, message, TimeSpan.FromMinutes(10)); // 设置消息的有效期

            // 发布消息ID
            long res = _subscriber.Publish(channel, message);
            _logger.LogInformation($"向主题:{channel};推送消息:{message};结果:{res}");
        }

      

        /// <summary>
        /// 推送消息到指定工位的队列
        /// </summary>
        /// <param name="workerId">工位ID</param>
        /// <param name="message">消息内容</param>
        public void PublishMessageToWorker(int workerId, string message)
        {

            var channel = $"channel_worker_{workerId}";

            // 生成唯一的消息ID
            var messageId = $"{DateTime.UtcNow.Ticks}_{Guid.NewGuid().ToString("N")}";

            // 存储消息内容，使用列表结构
            var redisDb = redis.GetDatabase();
            redisDb.ListRightPush(channel, message); // 将消息推入对应工位的队列

            _logger.LogInformation($"向工位:{workerId} 的队列推送消息ID:{messageId},消息内容:{message}");
        }


        /// <summary>
        /// 指定工位，从队列中获取一条消息（客户端消费时使用）
        /// </summary>
        /// <param name="workerId">工位ID</param>
        /// <returns>消息内容，如果没有消息则返回null</returns>
        public string ConsumeMessageFromWorker(string workId,int dataBaseId = 0)
        {
            var channel = workId;
            var redisDb = redis.GetDatabase(dataBaseId);

            // 从队列左侧弹出消息
            var message = redisDb.ListLeftPop(channel);
            if (message.IsNull)
            {
                return null;
            }

            return (string)message;
        }



        /// <summary>
        /// 确认消息
        /// </summary>
        /// <param name="messageId"></param>
        public void AcknowledgeMessage(string messageId)
        {
            var redisDb = redis.GetDatabase();
            redisDb.KeyDelete(messageId); // 删除确认的消息
            _logger.LogInformation($"消息ID:{messageId} 已被确认处理");
        }


        /// <summary>
        /// 订阅消息
        /// </summary>
        /// <param name="channel"></param>
        /// <param name="onMessageReceived"></param>
        public void SubscribeToChannel(string channel, Action<string, string> onMessageReceived)
        {
            _subscriber.Subscribe(channel, (ch, message) =>
            {
                onMessageReceived(ch, message);
                _logger.LogInformation($"订阅主题:{channel};收到主题:{ch};推送的消息:{message}");
            });
        }

    }
}