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