添加Redis服务与缓存增强

wanshenmean

2026-03-02 bfd4fd8e4a05a681ec10a47992294cf752a764c4

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
using Microsoft.Extensions.Logging;
using Microsoft.Extensions.Options;
using StackExchange.Redis;
using WIDESEAWCS_RedisService.Connection;
using WIDESEAWCS_RedisService.Options;
 
namespace WIDESEAWCS_RedisService.DelayQueue
{
    public class RedisDelayQueueService : IDelayQueueService
    {
        private readonly IRedisConnectionManager _connectionManager;
        private readonly RedisOptions _options;
        private readonly ILogger<RedisDelayQueueService> _logger;
 
        public RedisDelayQueueService(
            IRedisConnectionManager connectionManager,
            IOptions<RedisOptions> options,
            ILogger<RedisDelayQueueService> logger)
        {
            _connectionManager = connectionManager;
            _options = options.Value;
            _logger = logger;
        }
 
        private string BuildKey(string key) => $"{_options.KeyPrefix}delay:{key}";
 
        public bool Enqueue(string queueName, string message, TimeSpan delay)
        {
            var score = DateTimeOffset.UtcNow.Add(delay).ToUnixTimeMilliseconds();
            return _connectionManager.GetDatabase().SortedSetAdd(BuildKey(queueName), message, score);
        }
 
        public List<string> DequeueDue(string queueName, int count = 10)
        {
            var db = _connectionManager.GetDatabase();
            var fullKey = BuildKey(queueName);
            var now = DateTimeOffset.UtcNow.ToUnixTimeMilliseconds();
 
            var entries = db.SortedSetRangeByScore(fullKey, 0, now, take: count);
            var result = new List<string>();
 
            foreach (var entry in entries)
            {
                if (db.SortedSetRemove(fullKey, entry))
                    result.Add(entry.ToString());
            }
            return result;
        }
 
        public bool Remove(string queueName, string message)
        {
            return _connectionManager.GetDatabase().SortedSetRemove(BuildKey(queueName), message);
        }
 
        public long GetQueueLength(string queueName)
        {
            return _connectionManager.GetDatabase().SortedSetLength(BuildKey(queueName));
        }
    }
}