月球大数据地理空间分析展示平台-【后端】-月球后台服务
13693261870
2023-07-21 f4b1a132caad3002db25c999d931c41e5aa62019
代理服务的权限验证添加缓存功能以优化性能
已修改3个文件
41 ■■■■ 文件已修改
src/main/java/com/moon/server/entity/all/RedisCacheKey.java 7 ●●●●● 补丁 | 查看 | 原始文档 | blame | 历史
src/main/java/com/moon/server/service/all/PermsService.java 6 ●●●● 补丁 | 查看 | 原始文档 | blame | 历史
src/main/java/com/moon/server/service/sys/ProxyService.java 28 ●●●●● 补丁 | 查看 | 原始文档 | blame | 历史
src/main/java/com/moon/server/entity/all/RedisCacheKey.java
@@ -62,6 +62,13 @@
    }
    /**
     * 代理授权实体键
     */
    public static String permsProxy(String key) {
        return "perms:proxy:" + key;
    }
    /**
     * 黑/白名单键
     */
    public static String blacklistKey(String key) {
src/main/java/com/moon/server/service/all/PermsService.java
@@ -4,9 +4,9 @@
import com.moon.server.entity.sys.MenuEntity;
import com.moon.server.entity.sys.ResEntity;
import com.moon.server.mapper.all.PermsMapper;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import javax.annotation.Resource;
import java.util.List;
import java.util.concurrent.TimeUnit;
@@ -16,10 +16,10 @@
 */
@Service
public class PermsService implements PermsMapper {
    @Autowired
    @Resource
    PermsMapper permsMapper;
    @Autowired
    @Resource
    RedisService redisService;
    @Override
src/main/java/com/moon/server/service/sys/ProxyService.java
@@ -1,9 +1,7 @@
package com.moon.server.service.sys;
import com.alibaba.fastjson.JSON;
import com.moon.server.entity.all.HttpStatus;
import com.moon.server.entity.all.ResponseMsg;
import com.moon.server.entity.all.StaticData;
import com.moon.server.entity.all.*;
import com.moon.server.entity.sys.ResEntity;
import com.moon.server.entity.sys.ResLogEntity;
import com.moon.server.entity.sys.TokenEntity;
@@ -14,6 +12,7 @@
import com.moon.server.helper.WebHelper;
import com.moon.server.interceptor.AuthInterceptor;
import com.moon.server.service.all.PermsService;
import com.moon.server.service.all.RedisService;
import com.moon.server.service.all.SysService;
import org.springframework.stereotype.Service;
@@ -22,6 +21,7 @@
import javax.servlet.http.HttpServletResponse;
import java.util.List;
import java.util.TimerTask;
import java.util.concurrent.TimeUnit;
/**
 * 代理服务类
@@ -30,6 +30,8 @@
 */
@Service
public class ProxyService {
    @Resource
    RedisService redisService;
    @Resource
    private SysService sysService;
@@ -47,11 +49,8 @@
     */
    public void proxyUrl(String token, int resId, HttpServletRequest req, HttpServletResponse res) throws Exception {
        // 3.获取用户
        UserEntity ue = getUser(res, token);
        UserEntity ue = getUser(req, res, token);
        if (null == ue) {
            return;
        }
        if (!check(req, res, ue, token)) {
            return;
        }
@@ -72,12 +71,23 @@
    /**
     * 获取用户
     */
    private UserEntity getUser(HttpServletResponse res, String token) {
    private UserEntity getUser(HttpServletRequest req, HttpServletResponse res, String token) {
        String key = RedisCacheKey.permsProxy(token);
        Object obj = redisService.get(key);
        if (obj instanceof UserEntity) {
            return (UserEntity) obj;
        }
        UserEntity ue = sysService.tokenService.getUserByToken(token);
        if (ue == null) {
        if (null == ue) {
            WebHelper.writeStr2Page(res, AuthInterceptor.NO_LOGIN);
            return null;
        }
        if (!check(req, res, ue, token)) {
            return null;
        }
        redisService.put(key, ue, SettingData.CACHE_EXPIRE, TimeUnit.MINUTES);
        return ue;
    }