简单使用ThreadLocal,保证线程安全

此案例是在springboot的框架上使用,通过ThreadLocal存储用户信息,并通过UserThreadLocal类中的get()方法获取其中的用户信息。

1.编写一个UserThreadLocal类
public class UserThreadLocal {

    //声明为私有的
    private UserThreadLocal(){}

    private static final ThreadLocal<SysUser> LOCAL = new ThreadLocal<>();

    //放
    public static void put(SysUser sysUser){
        LOCAL.set(sysUser);
    }

    //取
    public static SysUser get(){
        return LOCAL.get();
    }

    //删除
    public static void remove(){
        LOCAL.remove();
    }
}

在此类中新建一个私有唯一的静态ThreadLocal对象LOCAL。还有用于将用户信息存放在ThreadLocal中的put方法,和从ThreadLocal中获取用户信息的get方法,以及删除ThreadLocal的remove方法。

最后还有一个用户信息实体类SysUser。

2.在其他类中使用UserThreadLocal.put(sysUser)存放用户信息
@Slf4j
@Component
public class LoginInterceptor implements HandlerInterceptor {

    @Autowired
    private LoginService loginService;

    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
        //在执行controller方法(Handler)之前进行执行
        /**
         * 1.需要判断请求的接口路径是否为 HandlerMethod (controller方法)
         * 2.判断token是否为空,如果为空未登录
         * 3.如果token不为空,登录验证loginservice checkToken
         * 4.如果认证成功放行即可
         */
        if (!(handler instanceof HandlerMethod)){
            //如果是handler方法,就放行。
            return true;
        }

        //使用slf4j做日志输出
        String token = request.getHeader("Authorization");
        log.info("=================request start===========================");
        String requestURI = request.getRequestURI();
        log.info("request uri:{}",requestURI);
        log.info("request method:{}",request.getMethod());
        log.info("token:{}", token);
        log.info("=================request end===========================");

        //如果token为空,不放行并提示未登录
        if (StringUtils.isBlank(token)){
            Result result = Result.fail(ErrorCode.NO_LOGIN.getCode(), "未登录");
            response.setContentType("application/json;charset=utf-8");
            response.getWriter().print(JSON.toJSONString(result));
            return false;
        }

        //将token解析为sysUser实体类,如果实体类为空,提示不登录,并不放行。
        SysUser sysUser = loginService.checkToken(token);
        if (sysUser == null){
            Result result = Result.fail(ErrorCode.NO_LOGIN.getCode(), "未登录");
            response.setContentType("application/json;charset=utf-8");
            response.getWriter().print(JSON.toJSONString(result));
            return false;
        }
        //登录验证成功 放行
        // 在controller中直接获取用户的信息
        UserThreadLocal.put(sysUser);
        return true;
    }

    @Override
    public void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex) throws Exception {
        //如果不删除ThreadLocal中用完的信息会有内存泄漏的风险
        UserThreadLocal.remove();
    }
}

在我的案例中,我在登录时会经过拦截器拦截进行验证,在登录验证成功后使用UserThreadLocal类中的put方法即UserThreadLocal.put(sysUser),将sysUser用户信息存入ThreadLocal。

特别注意

要使用remove方法删除ThreadLocal中用完的信息,否则会有内存泄漏的风险。本次我是

在LoginInterceptor类中重写afterCompletion方法,即方法执行完毕以后,使用UserThreadLocal.remove删除,防止内存泄漏。

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
ThreadLocal的作用是提供线程内的局部变量,这种变量在多线程环境下访问时能够保证各个线程里变量的独立性,也就是每个线程都拥有自己独立的变量副本。因此,ThreadLocal能够通过线程隔离来确保线程安全。 具体来说,每个线程都拥有一个独立的ThreadLocal对象,用于存储线程的局部变量。在多线程环境下,每个线程通过ThreadLocal对象可以获取自己的局部变量副本,而不会受到其他线程的干扰。 ThreadLocal是通过ThreadLocalMap实现的,每个线程维护一个ThreadLocalMap属性,里面以Map的形式存储了多个ThreadLocal对象。当线程中调用ThreadLocal的操作方法时,通过当前Thread线程对象可以获取到线程的ThreadLocalMap,再通过ThreadLocal对象从ThreadLocalMap中获取对应的局部变量值。这样,每个线程都可以独立地操作自己的局部变量,而不会影响其他线程的局部变量。 相比于其他线程安全的方式,例如使用synchronize加锁方法,ThreadLocal能够有效地减少线程间的竞争和冲突,提高性能和并发性。因为每个线程都拥有自己的局部变量副本,不需要进行额外的同步措施,避免了竞争和冲突的发生。但需要注意的是,ThreadLocal并不能解决所有的线程安全问题,它只是提供了一种线程隔离的手段,让每个线程都拥有自己的变量副本,需要开发人员在使用时仍然要注意线程安全问题
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值