【线程上下文】

线程上下文

/**
 * 线程上下文
 */
public class ThreadContextHolder {

    private static final InheritableThreadLocal<Object> context = new InheritableThreadLocal<>();//实际<Object>修改为自己需要的对象

    /**
     * 设置上下文
     *
     * @param o
     */
    public static void setContext(Object o) {//实际<Object>修改为自己需要的对象
        context.set(o);
    }

    /**
     * 从上下文获取对象
     *
     * @return
     */
    public static Object get() {//实际<Object>修改为自己需要的对象
        return context.get();
    }

    /**
     * 移除上下文
     */
    public static void remove() {
        context.remove();
    }
}

拦截器

import com.jwt.ThreadContextHolder;
import org.springframework.lang.Nullable;
import org.springframework.stereotype.Component;
import org.springframework.web.servlet.HandlerInterceptor;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

@Component
public class InterCeptor implements HandlerInterceptor {

    @Override
    public boolean preHandle(HttpServletRequest request,
                             HttpServletResponse response,
                             Object handler) throws Exception {
        //todo 可以调用ThreadContextHolder.setContext()设置上下文
        return true;
    }

    @Override
    public void afterCompletion(HttpServletRequest request,
                                HttpServletResponse response,
                                Object handler,
                                @Nullable Exception ex) throws Exception {
        ThreadContextHolder.remove();
    }
}