From b29431de5bcf917baa0fb3b0ff1c2aae0ca45a42 Mon Sep 17 00:00:00 2001 From: ziin Date: Tue, 16 Sep 2025 18:02:45 +0800 Subject: [PATCH] =?UTF-8?q?1.=E4=BF=AE=E6=94=B9=E5=AE=9A=E6=97=B6=E4=BB=BB?= =?UTF-8?q?=E5=8A=A1=E9=80=BB=E8=BE=91?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../system/job/DisableAIExpiredAccount.java | 93 ++++++++++++++ .../job/DisableBrotherExpiredAccount.java | 91 ++++++++++++++ .../system/job/DisableExpiredAccount.java | 119 ------------------ 3 files changed, 184 insertions(+), 119 deletions(-) create mode 100644 yudao-module-system/src/main/java/cn/iocoder/yudao/module/system/job/DisableAIExpiredAccount.java create mode 100644 yudao-module-system/src/main/java/cn/iocoder/yudao/module/system/job/DisableBrotherExpiredAccount.java delete mode 100644 yudao-module-system/src/main/java/cn/iocoder/yudao/module/system/job/DisableExpiredAccount.java diff --git a/yudao-module-system/src/main/java/cn/iocoder/yudao/module/system/job/DisableAIExpiredAccount.java b/yudao-module-system/src/main/java/cn/iocoder/yudao/module/system/job/DisableAIExpiredAccount.java new file mode 100644 index 0000000..1a986f7 --- /dev/null +++ b/yudao-module-system/src/main/java/cn/iocoder/yudao/module/system/job/DisableAIExpiredAccount.java @@ -0,0 +1,93 @@ +package cn.iocoder.yudao.module.system.job; + +import cn.hutool.core.date.DateUnit; +import cn.hutool.core.date.DateUtil; +import cn.hutool.core.date.LocalDateTimeUtil; +import cn.iocoder.yudao.framework.common.enums.CommonStatusEnum; +import cn.iocoder.yudao.framework.quartz.core.handler.JobHandler; +import cn.iocoder.yudao.framework.tenant.core.context.TenantContextHolder; +import cn.iocoder.yudao.framework.tenant.core.job.TenantJob; +import cn.iocoder.yudao.module.system.dal.dataobject.tenant.TenantDO; +import cn.iocoder.yudao.module.system.dal.dataobject.user.AdminUserDO; +import cn.iocoder.yudao.module.system.dal.mysql.tenant.TenantMapper; +import cn.iocoder.yudao.module.system.dal.mysql.user.AdminUserMapper; +import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; +import lombok.extern.slf4j.Slf4j; +import org.springframework.stereotype.Component; + +import javax.annotation.Resource; +import java.time.Duration; +import java.time.LocalDate; +import java.time.LocalDateTime; +import java.time.LocalTime; +import java.util.List; + +/** + * 禁用过期账号定时任务处理器 + *

+ * 该类用于定时检查并禁用已过期的AI账号和大哥账号。 + * 通过查询租户表中已过期的账号，然后将对应用户的权限状态更新为禁用。 + *

+ * + * @author ziin + * @date 2025/9/16 14:35 + */ +@Slf4j +@Component +public class DisableAIExpiredAccount implements JobHandler{ + + /** + * 租户数据访问对象 + * 用于查询租户信息和过期时间 + */ + @Resource + TenantMapper tenantMapper; + + /** + * 管理员用户数据访问对象 + * 用于查询和更新用户权限状态 + */ + @Resource + private AdminUserMapper userMapper; + + /** + * 执行禁用过期账号的定时任务 + *

+ * 该方法会执行以下操作： + * 1. 查询所有AI账号已过期的租户 + * 2. 查询所有大哥账号已过期的租户 + * 3. 分别禁用这些租户下的用户对应权限 + * 4. 返回操作结果统计信息 + *

+ * + * @param param 任务参数（当前未使用） + * @return 操作结果字符串，包含禁用的AI账号和大哥账号数量 + * @throws Exception 执行过程中可能抛出的异常 + */ + @Override + @TenantJob + public String execute(String param) throws Exception { + Long tenantId = TenantContextHolder.getTenantId(); + TenantDO tenant = tenantMapper.selectById(tenantId); + if (tenant.getAiExpireTime()==null) { + return "租户未配置AI过期时间"; + } + Duration brotherDuration = LocalDateTimeUtil.between(tenant.getAiExpireTime(), LocalDateTime.now()); + long minutes = brotherDuration.toMinutes(); + LambdaQueryWrapper aiUserQueryWrapper = new LambdaQueryWrapper<>(); + aiUserQueryWrapper.eq(AdminUserDO::getTenantId, tenantId); + List aiUserList = userMapper.selectList(aiUserQueryWrapper); + int aiAccountNum = 0 ; + if (minutes >= 0) { + for (AdminUserDO adminUserDO : aiUserList) { + adminUserDO.setAiChat((byte) 0); + userMapper.updateById(adminUserDO); + aiAccountNum++; + log.info("禁用过期AI账号，账号ID：{}", adminUserDO.getId()); + } + } + + // 返回操作结果：包含禁用的AI账号和大哥账号数量统计 + return "禁用过期账号成功，禁用了 " + aiAccountNum + " 个 AI 账号。"; + } +} \ No newline at end of file diff --git a/yudao-module-system/src/main/java/cn/iocoder/yudao/module/system/job/DisableBrotherExpiredAccount.java b/yudao-module-system/src/main/java/cn/iocoder/yudao/module/system/job/DisableBrotherExpiredAccount.java new file mode 100644 index 0000000..21e9842 --- /dev/null +++ b/yudao-module-system/src/main/java/cn/iocoder/yudao/module/system/job/DisableBrotherExpiredAccount.java @@ -0,0 +1,91 @@ +package cn.iocoder.yudao.module.system.job; + +import cn.hutool.core.date.LocalDateTimeUtil; +import cn.iocoder.yudao.framework.common.enums.CommonStatusEnum; +import cn.iocoder.yudao.framework.quartz.core.handler.JobHandler; +import cn.iocoder.yudao.framework.tenant.core.context.TenantContextHolder; +import cn.iocoder.yudao.framework.tenant.core.job.TenantJob; +import cn.iocoder.yudao.module.system.dal.dataobject.tenant.TenantDO; +import cn.iocoder.yudao.module.system.dal.dataobject.user.AdminUserDO; +import cn.iocoder.yudao.module.system.dal.mysql.tenant.TenantMapper; +import cn.iocoder.yudao.module.system.dal.mysql.user.AdminUserMapper; +import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; +import lombok.extern.slf4j.Slf4j; +import org.springframework.stereotype.Component; + +import javax.annotation.Resource; +import java.time.Duration; +import java.time.LocalDate; +import java.time.LocalDateTime; +import java.time.LocalTime; +import java.util.List; + +/** + * 禁用过期账号定时任务处理器 + *

+ * 该类用于定时检查并禁用已过期的AI账号和大哥账号。 + * 通过查询租户表中已过期的账号，然后将对应用户的权限状态更新为禁用。 + *

+ * + * @author ziin + * @date 2025/9/16 14:35 + */ +@Slf4j +@Component +public class DisableBrotherExpiredAccount implements JobHandler{ + + /** + * 租户数据访问对象 + * 用于查询租户信息和过期时间 + */ + @Resource + TenantMapper tenantMapper; + + /** + * 管理员用户数据访问对象 + * 用于查询和更新用户权限状态 + */ + @Resource + private AdminUserMapper userMapper; + + /** + * 执行禁用过期账号的定时任务 + *

+ * 该方法会执行以下操作： + * 1. 查询所有AI账号已过期的租户 + * 2. 查询所有大哥账号已过期的租户 + * 3. 分别禁用这些租户下的用户对应权限 + * 4. 返回操作结果统计信息 + *

+ * + * @param param 任务参数（当前未使用） + * @return 操作结果字符串，包含禁用的AI账号和大哥账号数量 + * @throws Exception 执行过程中可能抛出的异常 + */ + @Override + @TenantJob + public String execute(String param) throws Exception { + Long tenantId = TenantContextHolder.getTenantId(); + TenantDO tenant = tenantMapper.selectById(tenantId); + if (tenant.getBrotherExpireTime()==null) { + return "租户未配置大哥过期时间"; + } + Duration brotherDuration = LocalDateTimeUtil.between(tenant.getBrotherExpireTime(), LocalDateTime.now()); + long minutes = brotherDuration.toMinutes(); + LambdaQueryWrapper aiUserQueryWrapper = new LambdaQueryWrapper<>(); + aiUserQueryWrapper.eq(AdminUserDO::getTenantId, tenantId); + List aiUserList = userMapper.selectList(aiUserQueryWrapper); + int brotherAccountNum = 0 ; + if (minutes >= 0) { + for (AdminUserDO adminUserDO : aiUserList) { + adminUserDO.setBigBrother((byte) 0); + userMapper.updateById(adminUserDO); + brotherAccountNum++; + log.info("禁用过期大哥账号，账号ID：{}", adminUserDO.getId()); + } + } + + // 返回操作结果：包含禁用的AI账号和大哥账号数量统计 + return "禁用过期账号成功，禁用了 " + brotherAccountNum + " 个 大哥账号。"; + } +} \ No newline at end of file diff --git a/yudao-module-system/src/main/java/cn/iocoder/yudao/module/system/job/DisableExpiredAccount.java b/yudao-module-system/src/main/java/cn/iocoder/yudao/module/system/job/DisableExpiredAccount.java deleted file mode 100644 index 9a74168..0000000 --- a/yudao-module-system/src/main/java/cn/iocoder/yudao/module/system/job/DisableExpiredAccount.java +++ /dev/null @@ -1,119 +0,0 @@ -package cn.iocoder.yudao.module.system.job; - -import cn.iocoder.yudao.framework.common.enums.CommonStatusEnum; -import cn.iocoder.yudao.framework.quartz.core.handler.JobHandler; -import cn.iocoder.yudao.framework.tenant.core.aop.TenantIgnore; -import cn.iocoder.yudao.framework.tenant.core.context.TenantContextHolder; -import cn.iocoder.yudao.framework.tenant.core.job.TenantJob; -import cn.iocoder.yudao.framework.tenant.core.util.TenantUtils; -import cn.iocoder.yudao.module.system.dal.dataobject.tenant.TenantDO; -import cn.iocoder.yudao.module.system.dal.dataobject.user.AdminUserDO; -import cn.iocoder.yudao.module.system.dal.mysql.tenant.TenantMapper; -import cn.iocoder.yudao.module.system.dal.mysql.user.AdminUserMapper; -import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper; -import org.springframework.stereotype.Component; - -import javax.annotation.Resource; -import java.time.LocalDate; -import java.time.LocalDateTime; -import java.time.LocalTime; -import java.util.List; - -/** - * 禁用过期账号定时任务处理器 - *

- * 该类用于定时检查并禁用已过期的AI账号和大哥账号。 - * 通过查询租户表中已过期的账号，然后将对应用户的权限状态更新为禁用。 - *

- * - * @author ziin - * @date 2025/9/16 14:35 - */ -@Component -public class DisableExpiredAccount implements JobHandler{ - - /** - * 租户数据访问对象 - * 用于查询租户信息和过期时间 - */ - @Resource - TenantMapper tenantMapper; - - /** - * 管理员用户数据访问对象 - * 用于查询和更新用户权限状态 - */ - @Resource - private AdminUserMapper userMapper; - - /** - * 执行禁用过期账号的定时任务 - *

- * 该方法会执行以下操作： - * 1. 查询所有AI账号已过期的租户 - * 2. 查询所有大哥账号已过期的租户 - * 3. 分别禁用这些租户下的用户对应权限 - * 4. 返回操作结果统计信息 - *

- * - * @param param 任务参数（当前未使用） - * @return 操作结果字符串，包含禁用的AI账号和大哥账号数量 - * @throws Exception 执行过程中可能抛出的异常 - */ - @Override - @TenantJob - public String execute(String param) throws Exception { - // 获取今天的开始时间（00:00:00）和结束时间（23:59:59.999999999） - LocalDateTime todayStart = LocalDate.now().atStartOfDay(); - LocalDateTime todayEnd = LocalDate.now().atTime(LocalTime.MAX); - - Long tenantId = TenantContextHolder.getTenantId(); - - // 构建AI账号过期查询条件：查询AI过期时间在今天的租户 - LambdaQueryWrapper aiQueryWrapper = new LambdaQueryWrapper<>(); - aiQueryWrapper.between(TenantDO::getAiExpireTime, todayStart, todayEnd) - .eq(TenantDO::getId, tenantId) - .eq(TenantDO::getStatus, CommonStatusEnum.ENABLE.getStatus()); - - // 构建大哥账号过期查询条件：查询大哥过期时间在今天的租户 - LambdaQueryWrapper brotherQueryWrapper = new LambdaQueryWrapper<>(); - brotherQueryWrapper.between(TenantDO::getBrotherExpireTime, todayStart, todayEnd) - .eq(TenantDO::getId, tenantId) - .eq(TenantDO::getStatus, CommonStatusEnum.ENABLE.getStatus()); - - - // 初始化计数器：记录禁用的AI账号和大哥账号数量 - int aiAccountNum = 0 ; - int brotherAccountNum = 0 ; - - // 处理AI账号过期：查询所有AI账号已过期的租户，禁用其下所有用户的AI聊天权限 - TenantDO aiTenantDOS = tenantMapper.selectOne(aiQueryWrapper); - if (aiTenantDOS != null) { - LambdaQueryWrapper aiUserQueryWrapper = new LambdaQueryWrapper<>(); - aiUserQueryWrapper.eq(AdminUserDO::getTenantId, aiTenantDOS.getId()); - List aiUserList = userMapper.selectList(aiUserQueryWrapper); - // 禁用每个用户的AI聊天权限 - for (AdminUserDO adminUserDO : aiUserList) { - adminUserDO.setAiChat((byte) 0); - userMapper.updateById(adminUserDO); - aiAccountNum++; - } - } - // 处理大哥账号过期：查询所有大哥账号已过期的租户，禁用其下所有用户的大哥权限 - TenantDO brotherTenantDOS = tenantMapper.selectOne(brotherQueryWrapper); - if (brotherTenantDOS != null) { - LambdaQueryWrapper brotherUserQueryWrapper = new LambdaQueryWrapper<>(); - brotherUserQueryWrapper.eq(AdminUserDO::getTenantId, brotherTenantDOS.getId()); - List brotherUserList = userMapper.selectList(brotherUserQueryWrapper); - // 禁用每个用户的大哥权限 - for (AdminUserDO adminUserDO : brotherUserList) { - adminUserDO.setBigBrother((byte) 0); - userMapper.updateById(adminUserDO); - brotherAccountNum++; - } - } - - // 返回操作结果：包含禁用的AI账号和大哥账号数量统计 - return "禁用过期账号成功，禁用了 " + aiAccountNum + " 个 AI 账号，禁用了 " + brotherAccountNum + " 个 大哥账号。"; - } -} \ No newline at end of file