SemianiakaVY/BotSharp
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

#838

Browse source
This commit is contained in:
Haiping Chen 2025-01-22 10:00:20 -06:00
parent 61e3bb83e0
commit 920e9d0341
2 changed files with 23 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
src/Infrastructure/BotSharp.Core/BotSharpCoreExtensions.cs
View file

@ -52,13 +52,21 @@ public static class BotSharpCoreExtensions
config.Bind("SharpCache", cacheSettings);
services.AddSingleton(x => cacheSettings);
services.AddSingleton<MemoryCacheService>();
if (cacheSettings.CacheType == CacheType.RedisCache)
{
services.AddSingleton<RedisCacheService>();
}
services.AddSingleton<RedisCacheService>();
services.AddSingleton<ICacheService>(sp =>
cacheSettings.CacheType switch
{
return cacheSettings.CacheType switch
{
CacheType.RedisCache => sp.GetRequiredService<RedisCacheService>(),
_ => sp.GetRequiredService<MemoryCacheService>(),
});
};
});
}
public static IServiceCollection UsingSqlServer(this IServiceCollection services, IConfiguration config)

23
src/Infrastructure/BotSharp.Core/Infrastructures/Cache/RedisCacheService.cs
View file

@ -1,22 +1,21 @@
using BotSharp.Abstraction.Infrastructures;
using Newtonsoft.Json;
using StackExchange.Redis;
using System.Linq;
namespace BotSharp.Core.Infrastructures;
public class RedisCacheService : ICacheService
{
private IConnectionMultiplexer _redis = null!;
private readonly IServiceProvider _services;
public RedisCacheService(IConnectionMultiplexer redis)
public RedisCacheService()
{
_redis = redis;
}
public async Task<T?> GetAsync<T>(string key)
{
var db = _redis.GetDatabase();
var redis = _services.GetService<IConnectionMultiplexer>();
var db = redis.GetDatabase();
var value = await db.StringGetAsync(key);
if (value.HasValue)
@ -29,7 +28,8 @@ public class RedisCacheService : ICacheService
public async Task<object> GetAsync(string key, Type type)
{
var db = _redis.GetDatabase();
var redis = _services.GetService<IConnectionMultiplexer>();
var db = redis.GetDatabase();
var value = await db.StringGetAsync(key);
if (value.HasValue)
@ -43,20 +43,23 @@ public class RedisCacheService : ICacheService
public async Task SetAsync<T>(string key, T value, TimeSpan? expiry)
{
var db = _redis.GetDatabase();
var redis = _services.GetService<IConnectionMultiplexer>();
var db = redis.GetDatabase();
await db.StringSetAsync(key, JsonConvert.SerializeObject(value), expiry);
}
public async Task RemoveAsync(string key)
{
var db = _redis.GetDatabase();
var redis = _services.GetService<IConnectionMultiplexer>();
var db = redis.GetDatabase();
await db.KeyDeleteAsync(key);
}
public async Task ClearCacheAsync(string prefix)
{
var db = _redis.GetDatabase();
var server = _redis.GetServer(_redis.GetEndPoints().First());
var redis = _services.GetService<IConnectionMultiplexer>();
var db = redis.GetDatabase();
var server = redis.GetServer(redis.GetEndPoints().First());
const int pageSize = 1000;
var keys = server.Keys(pattern: $"{prefix}*", pageSize: pageSize).ToList();