ThreadLocal优化

测试类证明一下ThreadLocal存储的数据是线程程安全的

java 复制代码
    package com.lin.springboot01;

    import org.junit.jupiter.api.Test;

    public class testThreadLocal {
        @Test
        public void testThreadLocalSetAndGet(){
            //提供一个ThreadLocal对象
            ThreadLocal t1 = new ThreadLocal();
            new Thread(()->{
                t1.set("张三");
                System.out.println(Thread.currentThread().getName()+":"+t1.get());
                System.out.println(Thread.currentThread().getName()+":"+t1.get());
                System.out.println(Thread.currentThread().getName()+":"+t1.get());
                System.out.println(Thread.currentThread().getName()+":"+t1.get());
            },"绿色").start();

            new Thread(()->{
                t1.set("李四");
                System.out.println(Thread.currentThread().getName()+":"+t1.get());
                System.out.println(Thread.currentThread().getName()+":"+t1.get());
                System.out.println(Thread.currentThread().getName()+":"+t1.get());
                System.out.println(Thread.currentThread().getName()+":"+t1.get());
            },"黄色").start();
        }

    }

接上一篇获取用户详细信息,用ThreadLocal进行优化

ThreadLocalUtil:工具类

java 复制代码
package com.lin.springboot01.utils;

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:ThreadLocalUtil.set(claims)将数据存储到ThreadLocal中

java 复制代码
package com.lin.springboot01.interceptors;

import com.lin.springboot01.pojo.Result;
import com.lin.springboot01.utils.JwtUtil;
import com.lin.springboot01.utils.ThreadLocalUtil;
import jakarta.servlet.http.HttpServletRequest;
import jakarta.servlet.http.HttpServletResponse;
import org.springframework.stereotype.Component;
import org.springframework.web.servlet.HandlerInterceptor;

import java.util.Map;

@Component
public class LoginInterceptor implements HandlerInterceptor {
    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
        String token = request.getHeader("Authorization");
        try {
            //能否解析成功
            Map<String, Object> claims = JwtUtil.parseToken(token);
            //把业务数据存储到ThreadLocal中
            ThreadLocalUtil.set(claims);
            //放行
            return true;
        } catch (Exception e) {
            //解析失败,httpServletResponse响应码设置为401
            response.setStatus(401);
            return false;
        }
    }

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

UserController:改用ThreadLocalUtil获取用户name,在把参数传给findByName方法

java 复制代码
    @GetMapping("/userInfo")
    public Result<User> userInfo(/*@RequestHeader(name="Authorization") String token*/){
       /* Map<String, Object> map = JwtUtil.parseToken(token);
        String username = (String) map.get("username");*/
        Map<String,Object> map = ThreadLocalUtil.get();
        String username = (String) map.get("username");
        User user = userService.findByName(username);
        return Result.success(user);
    }
相关推荐
JS.Huang7 分钟前
【JavaScript】原生函数
开发语言·javascript·ecmascript
Olrookie13 分钟前
若依前后端分离版学习笔记(二十)——实现滑块验证码(vue3)
java·前端·笔记·后端·学习·vue·ruoyi
CoderCodingNo1 小时前
【GESP】C++五级考试大纲知识点梳理, (5) 算法复杂度估算(多项式、对数)
开发语言·c++·算法
倚栏听风雨1 小时前
java.lang.SecurityException异常
java
星河队长1 小时前
VS创建C++动态库和C#访问过程
java·c++·c#
鼠鼠我捏,要死了捏2 小时前
Java虚拟线程原理与性能优化实战
java·performance-optimization·virtual-thread
ftpeak2 小时前
JavaScript性能优化实战
开发语言·javascript·性能优化
艾菜籽2 小时前
Spring MVC练习:留言板
java·spring·mvc
一个很帅的帅哥2 小时前
JavaScript事件循环
开发语言·前端·javascript
驰羽2 小时前
[GO]gin框架:ShouldBindJSON与其他常见绑定方法
开发语言·golang·gin