From 7ef4892f9f24f941aca37e6b3991b808a0aca619 Mon Sep 17 00:00:00 2001 From: zhanghua <314079846@qq.com> Date: 星期五, 08 九月 2023 11:16:35 +0800 Subject: [PATCH] 优化 --- ycl-common/src/main/java/com/ycl/service/user/impl/UmsAdminCacheServiceImpl.java | 133 ++++++++++++++++++++++++++++++++++++++++++++ 1 files changed, 133 insertions(+), 0 deletions(-) diff --git a/ycl-common/src/main/java/com/ycl/service/user/impl/UmsAdminCacheServiceImpl.java b/ycl-common/src/main/java/com/ycl/service/user/impl/UmsAdminCacheServiceImpl.java new file mode 100644 index 0000000..9e1a4cb --- /dev/null +++ b/ycl-common/src/main/java/com/ycl/service/user/impl/UmsAdminCacheServiceImpl.java @@ -0,0 +1,133 @@ +package com.ycl.service.user.impl; + +import cn.hutool.core.collection.CollUtil; +import com.alibaba.fastjson.JSON; +import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper; +import com.ycl.entity.user.UmsAdmin; +import com.ycl.entity.user.UmsAdminRoleRelation; +import com.ycl.entity.user.UmsMenu; +import com.ycl.entity.user.UmsResource; +import com.ycl.mapper.user.UmsAdminMapper; +import com.ycl.service.redis.RedisService; +import com.ycl.service.user.UmsAdminCacheService; +import com.ycl.service.user.UmsAdminRoleRelationService; +import com.ycl.service.user.UmsAdminService; +import org.springframework.beans.factory.annotation.Value; +import org.springframework.stereotype.Service; + +import javax.annotation.Resource; +import java.util.List; +import java.util.stream.Collectors; + +/** + * 鍚庡彴鐢ㄦ埛缂撳瓨绠$悊Service瀹炵幇绫� + * Created by macro on 2020/3/13. + */ +@Service +public class UmsAdminCacheServiceImpl implements UmsAdminCacheService { + @Resource + private UmsAdminService umsAdminService; + @Resource + private RedisService redisService; + @Resource + private UmsAdminMapper umsAdminMapper; + @Resource + private UmsAdminRoleRelationService adminRoleRelationService; + @Value("${redis.database}") + private String REDIS_DATABASE; + @Value("${redis.expire.common}") + private Long REDIS_EXPIRE; + @Value("${redis.key.admin}") + private String REDIS_KEY_ADMIN; + @Value("${redis.key.resourceList}") + private String REDIS_KEY_RESOURCE_LIST; + + @Override + public void delAdmin(Long adminId) { + UmsAdmin admin = umsAdminService.getById(adminId); + if (admin != null) { + String key = REDIS_DATABASE + ":" + REDIS_KEY_ADMIN + ":" + admin.getUsername(); + redisService.del(key); + } + } + + @Override + public void delResourceList(Long adminId) { + String key = REDIS_DATABASE + ":" + REDIS_KEY_RESOURCE_LIST + ":" + adminId; + redisService.del(key); + } + + @Override + public void delResourceListByRole(Long roleId) { + QueryWrapper<UmsAdminRoleRelation> wrapper = new QueryWrapper<>(); + wrapper.lambda().eq(UmsAdminRoleRelation::getRoleId, roleId); + List<UmsAdminRoleRelation> relationList = adminRoleRelationService.list(wrapper); + if (CollUtil.isNotEmpty(relationList)) { + String keyPrefix = REDIS_DATABASE + ":" + REDIS_KEY_RESOURCE_LIST + ":"; + List<String> keys = relationList.stream().map(relation -> keyPrefix + relation.getAdminId()).collect(Collectors.toList()); + redisService.del(keys); + } + } + + @Override + public void delResourceListByRoleIds(List<Long> roleIds) { + QueryWrapper<UmsAdminRoleRelation> wrapper = new QueryWrapper<>(); + wrapper.lambda().in(UmsAdminRoleRelation::getRoleId, roleIds); + List<UmsAdminRoleRelation> relationList = adminRoleRelationService.list(wrapper); + if (CollUtil.isNotEmpty(relationList)) { + String keyPrefix = REDIS_DATABASE + ":" + REDIS_KEY_RESOURCE_LIST + ":"; + List<String> keys = relationList.stream().map(relation -> keyPrefix + relation.getAdminId()).collect(Collectors.toList()); + redisService.del(keys); + } + } + + @Override + public void delResourceListByResource(Long resourceId) { + List<Long> adminIdList = umsAdminMapper.getAdminIdList(resourceId); + if (CollUtil.isNotEmpty(adminIdList)) { + String keyPrefix = REDIS_DATABASE + ":" + REDIS_KEY_RESOURCE_LIST + ":"; + List<String> keys = adminIdList.stream().map(adminId -> keyPrefix + adminId).collect(Collectors.toList()); + redisService.del(keys); + } + } + + @Override + public UmsAdmin getAdmin(String username) { + String key = REDIS_DATABASE + ":" + REDIS_KEY_ADMIN + ":" + username; +// return (UmsAdmin) redisService.get(key); + String result = (String) redisService.get(key); + UmsAdmin umsAdmin = JSON.parseObject(result, UmsAdmin.class); + return umsAdmin; + } + + @Override + public void setAdmin(UmsAdmin admin) { + String key = REDIS_DATABASE + ":" + REDIS_KEY_ADMIN + ":" + admin.getUsername(); + redisService.set(key, JSON.toJSONString(admin), REDIS_EXPIRE); + } + + @Override + public List<UmsMenu> getResourceList(Long adminId) { + String key = REDIS_DATABASE + ":" + REDIS_KEY_RESOURCE_LIST + ":" + adminId; + String s = (String) redisService.get(key); +// return (List<UmsResource>) redisService.get(key); + List<UmsMenu> umsResources = JSON.parseArray(s, UmsMenu.class); + return umsResources; + } + + @Override + public void setResourceList(Long adminId, List<UmsMenu> menuList) { + String key = REDIS_DATABASE + ":" + REDIS_KEY_RESOURCE_LIST + ":" + adminId; + redisService.set(key, JSON.toJSONString(menuList), REDIS_EXPIRE); + } + + @Override + public void delBatchAdmin(List<Long> userIds) { + List<UmsAdmin> users = umsAdminService.listByIds(userIds); + if (CollUtil.isNotEmpty(users)) { + String keyPrefix = REDIS_DATABASE + ":" + REDIS_KEY_ADMIN + ":"; + List<String> keys = users.stream().map(umsAdmin->keyPrefix+umsAdmin.getUsername()).collect(Collectors.toList()); + redisService.del(keys); + } + } +} -- Gitblit v1.8.0