feat: 新增VIP过期自动卸载功能
- 新增`AiRechargeManager`类,实现VIP过期用户的自动卸载逻辑。 - 新增`AiHubConst`常量类,统一管理角色名称。 - 在`IRoleService`中添加`RemoveUserRoleByRoleCodeAsync`方法,用于移除指定用户的角色。 - 在`RoleManager`中实现`RemoveUserRoleByRoleCodeAsync`方法。 - 优化`CurrentExtensions`中VIP角色判断逻辑,使用常量替代硬编码。 - 调整`YiAbpWebModule`中部分代码格式,提升可读性。
This commit is contained in:
@@ -0,0 +1,6 @@
|
||||
namespace Yi.Framework.AiHub.Domain.Shared.Consts;
|
||||
|
||||
public class AiHubConst
|
||||
{
|
||||
public const string VipRole = "YiXinAi-Vip";
|
||||
}
|
||||
@@ -6,7 +6,6 @@
|
||||
</ItemGroup>
|
||||
|
||||
<ItemGroup>
|
||||
<Folder Include="Consts\" />
|
||||
<Folder Include="Etos\" />
|
||||
</ItemGroup>
|
||||
|
||||
|
||||
@@ -1,5 +1,6 @@
|
||||
using Microsoft.AspNetCore.Http;
|
||||
using Volo.Abp.Users;
|
||||
using Yi.Framework.AiHub.Domain.Shared.Consts;
|
||||
|
||||
namespace Yi.Framework.AiHub.Domain.Extensions;
|
||||
|
||||
@@ -7,6 +8,6 @@ public static class CurrentExtensions
|
||||
{
|
||||
public static bool IsAiVip(this ICurrentUser currentUser)
|
||||
{
|
||||
return currentUser.Roles.Contains("YiXinAi-Vip") || currentUser.UserName == "cc";
|
||||
return currentUser.Roles.Contains(AiHubConst.VipRole) || currentUser.UserName == "cc";
|
||||
}
|
||||
}
|
||||
@@ -0,0 +1,59 @@
|
||||
using Microsoft.Extensions.Logging;
|
||||
using Volo.Abp.Domain.Services;
|
||||
using Yi.Framework.AiHub.Domain.Entities;
|
||||
using Yi.Framework.AiHub.Domain.Entities.OpenApi;
|
||||
using Yi.Framework.AiHub.Domain.Shared.Consts;
|
||||
using Yi.Framework.Rbac.Application.Contracts.IServices;
|
||||
using Yi.Framework.SqlSugarCore.Abstractions;
|
||||
|
||||
namespace Yi.Framework.AiHub.Domain.Managers;
|
||||
|
||||
public class AiRechargeManager : DomainService
|
||||
{
|
||||
private readonly ISqlSugarRepository<AiRechargeAggregateRoot> _rechargeRepository;
|
||||
private readonly IRoleService _roleService;
|
||||
private readonly ISqlSugarRepository<TokenAggregateRoot> _tokenRepository;
|
||||
private readonly ILogger<AiRechargeManager> _logger;
|
||||
|
||||
public AiRechargeManager(ISqlSugarRepository<AiRechargeAggregateRoot> rechargeRepository,
|
||||
ISqlSugarRepository<TokenAggregateRoot> tokenRepository, ILogger<AiRechargeManager> logger,
|
||||
IRoleService roleService)
|
||||
{
|
||||
_rechargeRepository = rechargeRepository;
|
||||
_tokenRepository = tokenRepository;
|
||||
_logger = logger;
|
||||
_roleService = roleService;
|
||||
}
|
||||
|
||||
public async Task RemoveVipRoleByExpireAsync()
|
||||
{
|
||||
_logger.LogInformation("开始执行VIP过期自动卸载任务");
|
||||
|
||||
// 获取当前时间
|
||||
var currentTime = DateTime.Now;
|
||||
|
||||
// 查找过期的充值记录
|
||||
var expiredRecharges = await _rechargeRepository._DbQueryable
|
||||
.Where(x => x.ExpireDateTime.HasValue && x.ExpireDateTime.Value < currentTime)
|
||||
.ToListAsync();
|
||||
|
||||
if (!expiredRecharges.Any())
|
||||
{
|
||||
_logger.LogInformation("没有找到过期的VIP用户");
|
||||
return;
|
||||
}
|
||||
|
||||
// 获取过期用户的ID列表
|
||||
var expiredUserIds = expiredRecharges.Select(x => x.UserId).Distinct().ToList();
|
||||
_logger.LogInformation($"找到 {expiredUserIds.Count} 个过期的VIP用户");
|
||||
|
||||
// 获取YiXinAi-Vip角色ID
|
||||
await _roleService.RemoveUserRoleByRoleCodeAsync(expiredUserIds, AiHubConst.VipRole);
|
||||
|
||||
// 删除过期用户的Token密钥
|
||||
var removedTokenCount = await _tokenRepository.DeleteAsync(x => expiredUserIds.Contains(x.UserId));
|
||||
|
||||
_logger.LogInformation($"成功删除 {removedTokenCount} 个用户的Token密钥");
|
||||
_logger.LogInformation($"VIP过期自动卸载任务执行完成,共处理 {expiredUserIds.Count} 个过期用户");
|
||||
}
|
||||
}
|
||||
Reference in New Issue
Block a user