ThreadLocal

1.1 ThreadLocal

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

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

ThreadLocal常用方法：

A. public void set(T value) : 设置当前线程的线程局部变量的值

B. public T get() : 返回当前线程所对应的线程局部变量的值

C. public void remove() : 删除当前线程所对应的线程局部变量的值

我们可以在LoginInterceptor的preHandle方法中获取当前登录用户id，并调用ThreadLocal的set方法来设置当前线程的线程局部变量的值（用户id），然后在MyMetaObjectHandler的updateFill方法中调用ThreadLocal的get方法来获得当前线程所对应的线程局部变量的值（用户id）。 如果在后续的操作中, 我们需要在Controller / Service中要使用当前登录用户的ID, 可以直接从ThreadLocal直接获取。

1.2 操作步骤

实现步骤：

1). 编写BaseContext工具类，基于ThreadLocal封装的工具类

2). 在LoginInterceptor的preHandle方法中调用BaseContext来设置当前登录用户的id

3). 在MyMetaObjectHandler的方法中调用BaseContext获取登录用户的id

1.3 代码实现

1). BaseContext工具类

/**
 * 工具类，维护ThreadLocal
 */
public class BaseContext {
​
    private static final ThreadLocal<Long> threadLocal = new ThreadLocal<>();
​
    public static void setCurrentId(Long id) {
        threadLocal.set(id);
    }
​
    public static Long getCurrentId() {
        return threadLocal.get();
    }
}

​

 

2).LoginInterceptor的preHandle中存放当前登录用户到ThreadLocal

在preHandle方法中, 判定用户是否登录, 如果用户登录, 在放行之前, 获取HttpSession中的登录用户信息, 调用BaseContext的setCurrentId方法将当前登录用户ID存入ThreadLocal。

/**
 * 判断用户登录的拦截器
 */
@Slf4j
public class LoginInterceptor implements HandlerInterceptor {
​
    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response,
                             Object handler) throws Exception {
        //1. 判断用户是否登录，获取session中的用户
        Long employeeId = (Long) request.getSession().getAttribute("employee");
        if (employeeId != null) {
            //将当前登录的员工id存入线程
            BaseContext.setCurrentId(employeeId);
            return true;
        }
        //2. 用户未登录，响应code=0 && msg=NOTLOGIN
        log.info("用户未登录");
        R<String> r = R.error("NOTLOGIN");
        String jsonString = JSON.toJSONString(r);
        response.getWriter().write(jsonString);
        return false;
    }
​
}

​

3). MyMetaObjectHandler中从ThreadLocal中获取

将之前在代码中固定的当前登录用户1， 修改为动态调用BaseContext中的getCurrentId方法获取当前登录用户ID

@Component
@Slf4j
public class MyMetaObjectHandler implements MetaObjectHandler {
​
    //添加时自动填充
    @Override
    public void insertFill(MetaObject metaObject) {
        setFieldValByName("createTime", LocalDateTime.now(), metaObject);
        setFieldValByName("updateTime", LocalDateTime.now(), metaObject);
​
        setFieldValByName("createUser", BaseContext.getCurrentId(), metaObject);
        setFieldValByName("updateUser", BaseContext.getCurrentId(), metaObject);
    }
​
    //更新时自动填充
    @Override
    public void updateFill(MetaObject metaObject) {
        setFieldValByName("updateTime", LocalDateTime.now(), metaObject);
        setFieldValByName("updateUser", BaseContext.getCurrentId(), metaObject);
    }
}