跳至主要內容

ThreadLocal

sixkey大约 2 分钟后端ThreadLocal实战

ThreadLocal

1、使用前景

因为自动填充公共字段时需要获取当前登录用户的Id,所以引入了ThreadLocal。

在学习ThreadLocal之前,我们需要确认一个事情,就是客户端发送的每次http请求,对应的在服务端都会分配一个新的线程来处理,要确保以下所执行到的方法都是同一个线程:

1LoginCheckFilter的doFilter方法
2EmployeeController的update方法
3MyMetaObjectHandler的updateFill方法
可以使用以下代码查看当前线程Id
long id = ThreadLocal.currentThread().getId();

2 什么是ThreadLocal?

由JDK所提供。ThreadLocal并不是一个Thread,而是Thread的局部变量,当使用ThreadLocal维护变量时,ThreadLocal为每个使用该变量的线程提供独立的变量副本,所以每一个线程都可以独立地改变自己的副本,而不会影响其它线程所对应的副本。

ThreadLocal为每一个线程提供单独的一份存储空间,具有线程隔离的效果,只有在线程内才能获取到对应的值,线程外则不能访问。

ThreadLocal常用方法:

public void set(T value) 设置当前线程的线程局部变量的值
public T get()           返回当前线程所对应的线程局部变量的值

具体使用:

我们可以在LoginCheckFilter的doFilter方法中获取当前登录的用户Id,并调用ThreadLocal的set方法设置当前线程的线程局部变量的值 ( 用户Id ) ,然后在MyMetaObjectHandler的updateFill方法中调用ThreadLocal的get方法来获取当前的线程所对应的线程局部变量的值 ( 用户Id )。

代码实现:

/**
 * 基于ThreadLocal封装工具类,用来保存和获取当前登录的用户Id
 */
public class BaseContext {
    private static ThreadLocal<Long> threadLocal = new ThreadLocal<Long>();

    public static void setCurrentId(Long id){
        threadLocal.set(id);
    }
    public static Long getCurrentId(){
        return threadLocal.get();
    }
}

在过滤器方法里面一旦用户登录就设置用户Id

//4.查看登录状态,如果已登录则放行
        if(request.getSession().getAttribute("employee") != null){
            log.info("用户已登录,id为:"+request.getSession().getAttribute("employee"));
            Long empId = (Long) request.getSession().getAttribute("employee");
            //同一个线程保存用户Id
            BaseContext.setCurrentId(empId);
            filterChain.doFilter(request,response);
            return;
        }