springboot学习-黑马大事件学习 ——Threadlocal优化

最新推荐文章于 2024-11-10 14:44:40 发布

暖季啊

最新推荐文章于 2024-11-10 14:44:40 发布

阅读量561

点赞数 7

分类专栏： spring Boot 文章标签： spring boot 学习后端

本文链接：https://blog.youkuaiyun.com/weixin_50946965/article/details/135707229

版权

spring Boot 专栏收录该内容

6 篇文章

订阅专栏

文章介绍了如何在Java中使用ThreadLocalUtil工具包管理线程局部变量，LoginInterceptor用于处理登录验证，确保数据安全。在Controller层，通过ThreadLocal获取已验证的用户信息并进行后续操作，同时在请求完成后清除ThreadLocal以防止内存泄漏。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

存取数据方法set()/get()

引入ThreadLocalUtil 工具包

public class ThreadLocalUtil {
    //提供ThreadLocal对象,
    private static final ThreadLocal THREAD_LOCAL = new ThreadLocal();

    //根据键获取值
    public static <T> T get(){
        return (T) THREAD_LOCAL.get();
    }
	
    //存储键值对
    public static void set(Object value){
        THREAD_LOCAL.set(value);
    }


    //清除ThreadLocal 防止内存泄漏
    public static void remove(){
        THREAD_LOCAL.remove();
    }
}

LoginInterceptor 写login的处理信息

注意在最后记得调用remove方法释放


@Component
public class LoginInterceptor implements HandlerInterceptor {

    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
        String token = request.getHeader("Authorization");
        //验证token
        try {
            Map<String, Object> claims = JwtUtil.parseToken(token);
            ThreadLocalUtil.set(claims);
            return true;
        } catch (Exception e) {
            //  http响应状态为401
            response.setStatus(401);
            return false;
        }
    }

    @Override
    public void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex) throws Exception {
        //清除ThreadLocal中的数据
        ThreadLocalUtil.remove();
    }
}

在获取信息的controller层中使用

    @GetMapping("/userInfo")
    public Result<User> userInfo(/*@RequestHeader(name = "Authorization") String token*/) {
		//根据用户名查询用户
       /*
        Map<String, Object> map = JwtUtil.parseToken(token);
        String username = (String)map.get("username");*/
        //业务数据存储到Threadlocal中
        Map<String,Object> map =ThreadLocalUtil.get();
        String username = (String) map.get("username");
        User user = userService.findByUserName(username);
        return Result.success(user);

    }