增加DistributedCache,实现IDistributedCache,实现验证码分布式缓存

This commit is contained in:
shiningrise 2024-12-24 09:07:05 +08:00
parent ec1baf2ecd
commit 44405f8879
2 changed files with 82 additions and 0 deletions

View File

@ -0,0 +1,77 @@
// Admin.NET 项目的版权、商标、专利和其他相关权利均受相应法律法规的保护。使用本项目应遵守相关法律法规和许可证的要求。
//
// 本项目主要遵循 MIT 许可证和 Apache 许可证(版本 2.0)进行分发和使用。许可证位于源代码树根目录中的 LICENSE-MIT 和 LICENSE-APACHE 文件。
//
// 不得利用本项目从事危害国家安全、扰乱社会秩序、侵犯他人合法权益等法律法规禁止的活动!任何基于本项目二次开发而产生的一切法律纠纷和责任,我们不承担任何责任!
using Lazy.Captcha.Core;
using Microsoft.Extensions.Caching.Distributed;
using Microsoft.Identity.Client;
using System;
using System.Collections.Generic;
using System.Linq;
using System.Text;
using System.Threading.Tasks;
namespace Admin.NET.Core;
public class DistributedCache : IDistributedCache
{
private static ICacheProvider _cacheProvider;
public DistributedCache(ICacheProvider cacheProvider)
{
_cacheProvider = cacheProvider;
}
public string GetKey(string key)
{
return "Captcha:" + key;
}
public byte[] Get(string key)
{
return _cacheProvider.Cache.Get<byte[]>(GetKey(key));
}
public Task<byte[]> GetAsync(string key, CancellationToken token = default)
{
return Task.FromResult<byte[]>(_cacheProvider.Cache.Get<byte[]>(GetKey(key)));
}
public void Refresh(string key)
{
//_cacheProvider.Cache.Re<byte[]>(key);
}
public Task RefreshAsync(string key, CancellationToken token = default)
{
//throw new NotImplementedException();
return Task.CompletedTask;
}
public void Remove(string key)
{
_cacheProvider.Cache.Remove(GetKey(key));
}
public Task RemoveAsync(string key, CancellationToken token = default)
{
return Task.FromResult(_cacheProvider.Cache.Remove(GetKey(key)));
}
public void Set(string key, byte[] value, DistributedCacheEntryOptions options)
{
if(options.AbsoluteExpirationRelativeToNow != null)
_cacheProvider.Cache.Set(GetKey(key), value, options.AbsoluteExpirationRelativeToNow.Value);
else
_cacheProvider.Cache.Set(GetKey(key), value, TimeSpan.FromMinutes(1));
}
public Task SetAsync(string key, byte[] value, DistributedCacheEntryOptions options, CancellationToken token = default)
{
if (options.AbsoluteExpirationRelativeToNow != null)
return Task.FromResult(_cacheProvider.Cache.Set(GetKey(key), value, options.AbsoluteExpirationRelativeToNow.Value));
else
return Task.FromResult(_cacheProvider.Cache.Set(GetKey(key), value, TimeSpan.FromMinutes(1)));
}
}

View File

@ -19,7 +19,9 @@ using Microsoft.AspNetCore.Hosting;
using Microsoft.AspNetCore.Http;
using Microsoft.AspNetCore.HttpOverrides;
using Microsoft.AspNetCore.ResponseCompression;
using Microsoft.Extensions.Caching.Distributed;
using Microsoft.Extensions.DependencyInjection;
using Microsoft.Extensions.DependencyInjection.Extensions;
using Microsoft.Extensions.FileProviders;
using Microsoft.Extensions.Hosting;
using Newtonsoft.Json;
@ -208,8 +210,11 @@ public class Startup : AppStartup
// 系统日志
services.AddLoggingSetup();
services.AddSingleton<IDistributedCache, DistributedCache>();
// 验证码
services.AddCaptcha();
// 控制台logo
services.AddConsoleLogo();