ThreadLocal 方式存储用户信息

因为每次业务层 或者 其他非 Controller 层,需要用到用户信息的时候,就需要把当前用户信息先查询出来在传递,或者是直接把当前的 session 直接往下层传递。这样很繁琐。
可以使用 线程封闭 性来优化这个问题。

最关键的地方在于:

1.每次请求的时候,可能不是同一个线程去执行,会到导致用户数据获取不到。
2.线程要释放绑定的用户数据,不然会出现内存泄露的问题。

先创建 ThreadLocal 操作类

........
import lombok.extern.slf4j.Slf4j;

/**
 * session 操作
 * @author weiximei
 */
@Slf4j
public class SessionLoad {

    private static ThreadLocal<SysUser> threadLocal = new ThreadLocal<>();

    /**
     * 设置用户信息
     * @param user
     */
    public static void setUser(SysUser user){
        log.info("当前线程 --- [{}] --- 设置用户 {} ", Thread.currentThread().getName(), user);
        threadLocal.set(user);
    }

    /**
     * 获取用户信息
     * @return
     */
    public static SysUser getUser() {
        SysUser user = threadLocal.get();
        log.info("当前线程 --- [{}] --- 获取用户 {} ", Thread.currentThread().getName(), user);
        return user;
    }

    /**
     * 移除用户
     */
    public static void removeUser() {
        threadLocal.remove();
    }

}

用户session操作类,针对 ThreadLocal 操作二次封装

........
import org.springframework.stereotype.Component;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;

import javax.servlet.http.HttpSession;

/**
 * 用户session
 * @author weiximei
 */
@Component
@Slf4j
public class UserControl {

    public static final String SESSION_USER = "user";

    /**
     * 设置用户
     * @param user
     */
    public static void setUser(SysUser user) {
        if (null == user) {
            throw new FoyoException("用户信息不能为空");
        }
        HttpSession session = getSession();
        user.setToken(session.getId());
        session.setAttribute(SESSION_USER, user);
        log.info("当前用户sessionId: {}", session.getId());
        SessionLoad.setUser(user);
    }

    /**
     * 获取当前用户
     * @return
     */
    public static SysUser getUser() {
        HttpSession session = getSession();
        log.info("当前用户sessionId: {}", session.getId());

        SysUser sysUser = SessionLoad.getUser();
        if (null == sysUser) {
            throw new FoyoException("用户未登录");
        }

        return sysUser;
    }

    /**
     * 移除用户
     */
    public static void removeUser() {
        HttpSession session = getSession();
        session.removeAttribute(SESSION_USER);
        log.info("当前用户sessionId: {}, 退出登录", session.getId());

        SessionLoad.removeUser();
    }

    /**
     * 获取当前 Request
     */
    private static HttpSession getSession() {
        HttpSession session = ((ServletRequestAttributes)RequestContextHolder.currentRequestAttributes()).getRequest().getSession();
        return session;
    }
}

定义一个 Filter 解决,每次请求不是同一个线程,造成用户信息获取不到问题

.......
import org.springframework.stereotype.Component;

import javax.servlet.*;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;

/**
 * 过滤器
 * @author weiximei
 */
@Component
public class FoyoFilter implements Filter {

    @Override
    public void doFilter(ServletRequest servletRequest, ServletResponse servletResponse, FilterChain filterChain) throws IOException, ServletException {
        HttpServletRequest request = (HttpServletRequest) servletRequest;
        HttpServletResponse response = (HttpServletResponse) servletResponse;

        if (request.getRequestURI().equals("/admin/login")) {
            filterChain.doFilter(servletRequest, servletResponse);
            return;
        }

        SysUser user = SessionLoad.getUser();
        if (null == user) {
            user = (SysUser) request.getSession().getAttribute(UserControl.SESSION_USER);
            if (null == user) {
                // 重定向登录
                request.getRequestDispatcher("/admin/toLogin").forward(servletRequest, servletResponse);
                return;
            }
        }

        // 设置用户
        UserControl.setUser(user);

        // 继续执行
        filterChain.doFilter(servletRequest, servletResponse);
    }
}

定义一个 Request 上下文监听器,用于释放线程绑定的用户信息,避免内存泄露
这里也可以选择 Spring Mvc提供的拦截器。但是我这里选择用 Servlet 的原生接口实现

.......
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Component;

import javax.servlet.ServletRequestEvent;
import javax.servlet.ServletRequestListener;
import javax.servlet.http.HttpServletRequest;

/**
 * web 请求监听器
 * @author weiximei
 */
@Component
@Slf4j
public class RequestListener implements ServletRequestListener {

    @Override
    public void requestDestroyed(ServletRequestEvent sre) {
        HttpServletRequest request  = (HttpServletRequest) sre.getServletRequest();
        log.info("当前线程 --- [{}] --- 销毁请求:{}", Thread.currentThread().getName() , request.getRequestURI());

       /**
         * 移除当前线程绑定的用户数据,防止内存泄露
         */
        SessionLoad.removeUser();
    }

    @Override
    public void requestInitialized(ServletRequestEvent sre) {
        HttpServletRequest request  = (HttpServletRequest) sre.getServletRequest();
        log.info("当前线程 --- [{}] --- 创建请求:{}", Thread.currentThread().getName() ,request.getRequestURI());
    }
}

看看结果:

2019-09-02 15:51:57.392  INFO 15848 --- [nio-8080-exec-3] com.huiju.foyo.filter.RequestListener    : 当前线程 ---  [http-nio-8080-exec-3] --- 创建请求:/admin/user/info
2019-09-02 15:51:57.392  INFO 15848 --- [nio-8080-exec-3] com.huiju.foyo.session.SessionLoad       : 当前线程 --- [http-nio-8080-exec-3] --- 获取用户 null 
2019-09-02 15:51:57.392  INFO 15848 --- [nio-8080-exec-3] com.huiju.foyo.session.UserControl       : 当前用户sessionId: 037D8E592F00E126463DA83A73EA4111
2019-09-02 15:51:57.392  INFO 15848 --- [nio-8080-exec-3] com.huiju.foyo.session.SessionLoad       : 当前线程 --- [http-nio-8080-exec-3] --- 设置用户 SysUser(userId=1, name=admin, avatar=https://wpimg.wallstcn.com/f778738c-e4f8-4870-b634-56703b4acafe.gif, loginName=admin, passwd=null, status=1, token=037D8E592F00E126463DA83A73EA4111) 
2019-09-02 15:51:57.393  INFO 15848 --- [nio-8080-exec-3] com.huiju.foyo.session.UserControl       : 当前用户sessionId: 037D8E592F00E126463DA83A73EA4111
2019-09-02 15:51:57.393  INFO 15848 --- [nio-8080-exec-3] com.huiju.foyo.session.SessionLoad       : 当前线程 --- [http-nio-8080-exec-3] --- 获取用户 SysUser(userId=1, name=admin, avatar=https://wpimg.wallstcn.com/f778738c-e4f8-4870-b634-56703b4acafe.gif, loginName=admin, passwd=null, status=1, token=037D8E592F00E126463DA83A73EA4111) 
2019-09-02 15:51:57.394  INFO 15848 --- [nio-8080-exec-3] com.huiju.foyo.filter.RequestListener    : 当前线程 ---  [http-nio-8080-exec-3] --- 销毁请求:/admin/user/info

监听器Listener 和 Filter

Servlet监听器用于监听一些重要事件的发生,监听器对象可以在事情发生前、发生后可以做一些必要的处理。下面将介绍几种常用的监听器,以及它们都适合运用于那些环境。
分类及介绍:
1.ServletContextListener:用于监听WEB 应用启动和销毁的事件,监听器类需要实现javax.servlet.ServletContextListener 接口。
2.ServletContextAttributeListener:用于监听WEB应用属性改变的事件,包括:增加属性、删除属性、修改属性,监听器类需要实现javax.servlet.ServletContextAttributeListener接口。
3.ServletRequestListener : 用于监听请求的创建于销毁, 需要实现 javax.servlet.ServletRequestListener​ 接口
4.ServletRequestAttributeListener:请求属性事件监听器。用于监听请求中的属性改变的事件。
5.HttpSessionListener:用于监听Session对象的创建和销毁,监听器类需要实现javax.servlet.http.HttpSessionListener接口或者javax.servlet.http.HttpSessionActivationListener接口,或者两个都实现。
6.HttpSessionActivationListener:用于监听Session对象的钝化/活化事件,监听器类需要实现javax.servlet.http.HttpSessionListener接口或者javax.servlet.http.HttpSessionActivationListener接口,或者两个都实现。
7.HttpSessionAttributeListener:用于监听Session对象属性的改变事件,监听器类需要实现
8.HttpSessionBindingListener:会话值绑定事件监听器。这是唯一不需要在web.xml中设定的Listener。
9.Filter:用于在客户端的请求访问后端资源之前,拦截这些请求 或者 在服务器的响应发送回客户端之前,处理这些响应

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 202,802评论 5 476
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 85,109评论 2 379
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 149,683评论 0 335
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 54,458评论 1 273
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 63,452评论 5 364
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,505评论 1 281
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 37,901评论 3 395
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,550评论 0 256
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,763评论 1 296
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,556评论 2 319
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,629评论 1 329
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 33,330评论 4 318
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 38,898评论 3 307
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,897评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 31,140评论 1 259
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 42,807评论 2 349
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 42,339评论 2 342

推荐阅读更多精彩内容