ThreadLocal全局获取用户信息

场景需求 : 虽然在登录之后,前端可以获取到我所返回的jwt,然后通过它保持登录状态和获取信息。然后在调用接口的时候将信息传递给我进行各种操作。
但是种种原因问题,我需要很简便地在执行操作的过程中,在任何一个方法中都能获取到当前用户的一些基本信息
而不纯粹依赖前端传值。
因为并非所有方法都可以拿到jwt,如果没有请求头作为参数的话。
我设想了一种简便的实现方式,就是在请求通过jwtFilter的时候,获取jwt的同时,把解析出来的用户信息存储到ThreadLocal之中。
使用ThreadLocal的话,可以在用户调用的过程中任何方法里作为一个全局的参数,随时得到想要的信息。

 

 

1、登陆成功后把用户信息存储到redis缓存中

伪代码如下

 @RequestMapping("/login")
    public Map login(String name){
        Map result = new HashMap<>();
        result.put(ExecutionContext.USER_ID,name);
        result.put(ExecutionContext.USER_NAME,name);
        result.put(ExecutionContext.ACCESS_TOKEN,name);
        //存储用户信息到redis中 key值Token  value是用户信息
        RMap map = redissonClient.getMap(name);

        map.put(ExecutionContext.USER_ID,name);
        map.put(ExecutionContext.USER_NAME,name);
        map.put(ExecutionContext.ACCESS_TOKEN,name);

        return map;
    }

2.在过滤其中进行存放用户登录信息

伪代码如下


package com.ahies.test.module1.filter;


import com.ahies.test.module1.util.ExecutionContext;
import org.apache.commons.lang3.StringUtils;
import org.redisson.Redisson;
import org.redisson.api.RedissonClient;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.data.redis.core.StringRedisTemplate;

import javax.annotation.Resource;
import javax.servlet.*;
import javax.servlet.annotation.WebFilter;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.util.List;
import java.util.Map;
import java.util.Objects;

/**
 * @author hwt
 * @description TODO
 * @Date 2018/12/12 10:40
 */


@WebFilter(urlPatterns = "/*")
public class ValidationFilter implements Filter {
    private final Logger LOGGER = LoggerFactory.getLogger(ValidationFilter.class);

    @Autowired
    RedissonClient redissonClient;


    @Override
    public void init(FilterConfig filterConfig) throws ServletException {
        System.out.println("----------------------->过滤器被创建");
    }

    @Override
    public void doFilter(ServletRequest servletRequest, ServletResponse servletResponse, FilterChain filterChain) throws IOException, ServletException {
        HttpServletRequest req = (HttpServletRequest) servletRequest;
        HttpServletResponse httpResponse = (HttpServletResponse) servletResponse;
        //防止中文乱码
        httpResponse.setContentType("text/html;charset=utf-8");
        String requestURI = req.getRequestURI();
        LOGGER.info("当前请求url为*************** {}", requestURI);
        String token = req.getParameter("Token");
        if (StringUtils.isBlank(token)) {
            token = req.getHeader("Token");
        }
        Map map = redissonClient.getMap(token);
        // 获取用户信息 用户信息存储到ThreadLocal之中。
        ExecutionContext.setContextMap(map);
        filterChain.doFilter(servletRequest,servletResponse);


    }

    @Override
    public void destroy() {
        LOGGER.info("----------------------->过滤器被销毁");
    }


}

3、ThreadLocal工具类

ExecutionContext
package com.ahies.test.module1.util;

import java.util.HashMap;
import java.util.Map;

/**
 * 用于在应用内及应用间保存、传递执行上下文信息。
 * 
 * @author LIU Fangran
 * 
 */
public class ExecutionContext {
	/**
	 * 用于保存线程相关信息
	 */
	transient static ThreadLocal> threadLocal = new ThreadLocal>();


	public static final String USER_ID = "user_id";
	public static final String USER_NAME = "user_name";
	public static final String ACCESS_TOKEN = "access_token";

	/**
	 * 构造函数
	 */
	public ExecutionContext() {
	}

	/**
	 * 从 ThreadLocal中获取名值Map(不包含appCode)
	 * 
	 * @return 名值Map
	 */
	public static Map getContextMap() {
		Map map = threadLocal.get();
		return map;
	}

	/**
	 * 从 ThreadLocal 获取名值Map
	 * 
	 * @param contextMap
	 *            名值Map
	 */
	public static void setContextMap(Map contextMap) {
		threadLocal.set(contextMap);
	}

	/**
	 * (获取键下的值.如果不存在,返回null;如果名值Map未初始化,也返回null) Get the value of key. Would
	 * return null if context map hasn't been initialized.
	 * 
	 * @param key
	 *            键
	 * @return 键下的值
	 */
	public static String get(String key) {
		Map contextMap = getContextMap();
		if (contextMap == null) {
			return null;
		}
		return contextMap.get(key);
	}

	/**
	 * (设置名值对。如果Map之前为null,则会被初始化) Put the key-value into the context map;
	 * 

* Initialize the map if the it doesn't exist. * * @param key * 键 * @param value * 值 * @return 之前的值 */ public static String put(String key, String value) { Map contextMap = getContextMap(); if (contextMap == null) { contextMap = new HashMap(); setContextMap(contextMap); } return contextMap.put(key, value); } public static String getUserId() { return get(USER_ID); } public static void setUserId(String userId) { put(USER_ID, userId); } public static String getUserName() { return get(USER_NAME); } public static void setUserName(String userName) { put(USER_NAME, userName); } public static String getAccessToken() { return get(ACCESS_TOKEN); } public static void setAccessToken(String requestDomain) { put(ACCESS_TOKEN, requestDomain); } }

4、使用如下

  @RequestMapping("/get")
    public Page get(){
        Page page = new Page(1,10);
        List des = new ArrayList<>();
        des.add("create_time");
        page.setDescs(des);
        //获取用户id
        System.out.println(ExecutionContext.getUserId());
        stringRedisTemplate.opsForValue().set("11","22");
        return  holidayService.selectPage(page,null);
    }

 

你可能感兴趣的:(Java和Jvm)