Skip to content

Commit

Permalink
chore: 完善演示环境任务重置演示环境数据定时任务
Browse files Browse the repository at this point in the history
  • Loading branch information
@Charles7c
Charles7c committed Dec 30, 2024
1 parent 8c53700 commit 00e5d92
Showing 1 changed file with 26 additions and 15 deletions.
41 changes: 26 additions & 15 deletions continew-webapi/src/main/java/top/continew/admin/controller/schedule/DemoEnvironmentJob.java
View file
Original file line number Diff line number Diff line change
Expand Up @@ -25,6 +25,8 @@
import org.springframework.stereotype.Component;
import org.springframework.transaction.annotation.Transactional;
import top.continew.admin.common.constant.CacheConstants;
import top.continew.admin.open.mapper.AppMapper;
import top.continew.admin.open.model.entity.AppDO;
import top.continew.admin.system.mapper.*;
import top.continew.admin.system.model.entity.*;
import top.continew.starter.cache.redisson.util.RedisUtils;
Expand Down Expand Up @@ -56,14 +58,13 @@ public class DemoEnvironmentJob {
private final MenuMapper menuMapper;
private final DeptMapper deptMapper;

private static final Long DICT_ITEM_FLAG = 4L;
private static final Long DICT_FLAG = 2L;
private static final Long STORAGE_FLAG = 2L;
private static final Long NOTICE_FLAG = 7L;
private final AppMapper appMapper;
private final ClientMapper clientsMapper;

private static final Long DELETE_FLAG = 1000L;
private static final Long MESSAGE_FLAG = 0L;
private static final Long[] USER_FLAG = {1L, 547889293968801831L};
private static final Long ROLE_FLAG = 547888897925840928L;
private static final Long MENU_FLAG = 10000L;
private static final Long DEPT_FLAG = 547888580614160409L;

/**
Expand All @@ -76,36 +77,42 @@ public void resetEnvironmentData() {
SnailJobLog.REMOTE.info("定时任务 [重置演示环境数据] 开始执行。");
// 检测待清理数据
SnailJobLog.REMOTE.info("开始检测演示环境待清理数据项，请稍候...");
Long dictItemCount = dictItemMapper.lambdaQuery().gt(DictItemDO::getId, DICT_ITEM_FLAG).count();
Long dictItemCount = dictItemMapper.lambdaQuery().gt(DictItemDO::getId, DELETE_FLAG).count();
this.log(dictItemCount, "字典项");
Long dictCount = dictMapper.lambdaQuery().gt(DictDO::getId, DICT_FLAG).count();
Long dictCount = dictMapper.lambdaQuery().gt(DictDO::getId, DELETE_FLAG).count();
this.log(dictCount, "字典");
Long storageCount = storageMapper.lambdaQuery().gt(StorageDO::getId, STORAGE_FLAG).count();
Long storageCount = storageMapper.lambdaQuery().gt(StorageDO::getId, DELETE_FLAG).count();
this.log(storageCount, "存储");
Long noticeCount = noticeMapper.lambdaQuery().gt(NoticeDO::getId, NOTICE_FLAG).count();
Long noticeCount = noticeMapper.lambdaQuery().gt(NoticeDO::getId, DELETE_FLAG).count();
this.log(noticeCount, "公告");
Long messageCount = messageMapper.lambdaQuery().count();
this.log(messageCount, "通知");
Long userCount = userMapper.lambdaQuery().notIn(UserDO::getId, USER_FLAG).count();
this.log(userCount, "用户");
Long roleCount = roleMapper.lambdaQuery().gt(RoleDO::getId, ROLE_FLAG).count();
this.log(roleCount, "角色");
Long menuCount = menuMapper.lambdaQuery().gt(MenuDO::getId, MENU_FLAG).count();
Long menuCount = menuMapper.lambdaQuery().gt(MenuDO::getId, DELETE_FLAG).count();
this.log(menuCount, "菜单");
Long deptCount = deptMapper.lambdaQuery().gt(DeptDO::getId, DEPT_FLAG).count();
this.log(deptCount, "部门");
Long appCount = appMapper.lambdaQuery().gt(AppDO::getId, DELETE_FLAG).count();
this.log(appCount, "应用");
Long clientCount = clientsMapper.lambdaQuery().gt(ClientDO::getId, DELETE_FLAG).count();
this.log(clientCount, "客户端");
// 清理数据
InterceptorIgnoreHelper.handle(IgnoreStrategy.builder().blockAttack(true).build());
SnailJobLog.REMOTE.info("演示环境待清理数据项检测完成，开始执行清理。");
this.clean(dictItemCount, "字典项", null, () -> dictItemMapper.lambdaUpdate()
.gt(DictItemDO::getId, DICT_ITEM_FLAG)
.gt(DictItemDO::getId, DELETE_FLAG)
.remove());
this.clean(dictCount, "字典", CacheConstants.DICT_KEY_PREFIX, () -> dictMapper.lambdaUpdate()
.gt(DictDO::getId, DELETE_FLAG)
.remove());
this.clean(dictCount, "字典", null, () -> dictMapper.lambdaUpdate().gt(DictDO::getId, DICT_FLAG).remove());
this.clean(storageCount, "存储", null, () -> storageMapper.lambdaUpdate()
.gt(StorageDO::getId, STORAGE_FLAG)
.gt(StorageDO::getId, DELETE_FLAG)
.remove());
this.clean(noticeCount, "公告", null, () -> noticeMapper.lambdaUpdate()
.gt(NoticeDO::getId, NOTICE_FLAG)
.gt(NoticeDO::getId, DELETE_FLAG)
.remove());
this.clean(messageCount, "通知", null, () -> {
messageUserMapper.lambdaUpdate().gt(MessageUserDO::getMessageId, MESSAGE_FLAG).remove();
Expand All @@ -122,9 +129,13 @@ public void resetEnvironmentData() {
return roleMapper.lambdaUpdate().gt(RoleDO::getId, ROLE_FLAG).remove();
});
this.clean(menuCount, "菜单", CacheConstants.MENU_KEY_PREFIX, () -> menuMapper.lambdaUpdate()
.gt(MenuDO::getId, MENU_FLAG)
.gt(MenuDO::getId, DELETE_FLAG)
.remove());
this.clean(deptCount, "部门", null, () -> deptMapper.lambdaUpdate().gt(DeptDO::getId, DEPT_FLAG).remove());
this.clean(appCount, "应用", null, () -> appMapper.lambdaUpdate().gt(AppDO::getId, DEPT_FLAG).remove());
this.clean(clientCount, "客户端", null, () -> clientsMapper.lambdaUpdate()
.gt(ClientDO::getId, DEPT_FLAG)
.remove());
SnailJobLog.REMOTE.info("演示环境数据已清理完成。");
SnailJobLog.REMOTE.info("定时任务 [重置演示环境数据] 执行结束。");
} finally {
Expand Down

0 comments on commit 00e5d92

Please sign in to comment.