47 lines
1.8 KiB
C#
47 lines
1.8 KiB
C#
using IM_API.Interface.Services;
|
|
using IM_API.Models;
|
|
using IM_API.Tools;
|
|
using Microsoft.EntityFrameworkCore;
|
|
using RedLockNet;
|
|
using StackExchange.Redis;
|
|
|
|
namespace IM_API.Services
|
|
{
|
|
public class SequenceIdService : ISequenceIdService
|
|
{
|
|
private IDatabase _database;
|
|
private IDistributedLockFactory _lockFactory;
|
|
private ImContext _context;
|
|
public SequenceIdService(IConnectionMultiplexer connectionMultiplexer,
|
|
IDistributedLockFactory distributedLockFactory, ImContext imContext)
|
|
{
|
|
_database = connectionMultiplexer.GetDatabase();
|
|
_lockFactory = distributedLockFactory;
|
|
_context = imContext;
|
|
}
|
|
public async Task<long> GetNextSquenceIdAsync(string streamKey)
|
|
{
|
|
string key = RedisKeys.GetSequenceIdKey(streamKey);
|
|
string lockKey = RedisKeys.GetSequenceIdLockKey(streamKey);
|
|
var exists = await _database.KeyExistsAsync(key);
|
|
if (!exists)
|
|
{
|
|
using (var _lock = await _lockFactory.CreateLockAsync(lockKey, TimeSpan.FromSeconds(5)))
|
|
{
|
|
if (_lock.IsAcquired)
|
|
{
|
|
if(!await _database.KeyExistsAsync(key))
|
|
{
|
|
var max = await _context.Messages
|
|
.Where(x => x.StreamKey == streamKey)
|
|
.MaxAsync(m => (long?)m.SequenceId) ?? 0;
|
|
await _database.StringSetAsync(key, max, TimeSpan.FromDays(7));
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return await _database.StringIncrementAsync(key);
|
|
}
|
|
}
|
|
}
|