ThreadLocal +拦截器处理参数

1、构建ThreadContext

100

101

102

103

104

package com.ne.ice.boot.common;
 
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
 
import java.util.HashMap;
import java.util.Map;
 
 
public class ThreadContext {
    private static final Logger log = LoggerFactory.getLogger(ThreadContext.class);
 
    public static final String SUBJECT_KEY = ThreadContext.class.getName() + "_SUBJECT_KEY";
 
    private static final ThreadLocal<Map<Object, Object>> resources = new InheritableThreadLocalMap<Map<Object, Object>>();
 
    protected ThreadContext() {
    }
 
    public static Map<Object, Object> getResources() {
        return resources != null ? new HashMap<Object, Object>(resources.get()) : null;
    }
 
    public static void setResources(Map<Object, Object> newResources) {
        if (isEmpty(newResources)) {
            return;
        }
        resources.get().clear();
        resources.get().putAll(newResources);
    }
 
    public static boolean isEmpty(Map<?, ?> map) {
        return (map == null || map.isEmpty());
    }
 
    private static Object getValue(Object key) {
        return resources.get().get(key);
    }
 
    public static <T> T get(Object key) {
        if (log.isTraceEnabled()) {
            String msg = "get() - in thread [" + Thread.currentThread().getName() + "]";
            log.trace(msg);
        }
 
        Object value = getValue(key);
        if ((value != null) && log.isTraceEnabled()) {
            String msg = "Retrieved value of type [" + value.getClass().getName() + "] for key [" +
                    key + "] " + "bound to thread [" + Thread.currentThread().getName() + "]";
            log.trace(msg);
        }
        return (T) value;
    }
 
    public static void put(Object key, Object value) {
        if (key == null) {
            throw new IllegalArgumentException("key cannot be null");
        }
 
        if (value == null) {
            remove(key);
            return;
        }
 
        resources.get().put(key, value);
 
        if (log.isTraceEnabled()) {
            String msg = "Bound value of type [" + value.getClass().getName() + "] for key [" +
                    key + "] to thread " + "[" + Thread.currentThread().getName() + "]";
            log.trace(msg);
        }
    }
 
    public static Object remove(Object key) {
        Object value = resources.get().remove(key);
 
        if ((value != null) && log.isTraceEnabled()) {
            String msg = "Removed value of type [" + value.getClass().getName() + "] for key [" +
                    key + "]" + "from thread [" + Thread.currentThread().getName() + "]";
            log.trace(msg);
        }
 
        return value;
    }
 
    public static void remove() {
        resources.remove();
    }
 
    private static final class InheritableThreadLocalMap<T extends Map<Object, Object>> extends InheritableThreadLocal<Map<Object, Object>> {
        protected Map<Object, Object> initialValue() {
            return new HashMap<Object, Object>();
        }
 
        @SuppressWarnings({"unchecked"})
        protected Map<Object, Object> childValue(Map<Object, Object> parentValue) {
            if (parentValue != null) {
                return (Map<Object, Object>) ((HashMap<Object, Object>) parentValue).clone();
            } else {
                return null;
            }
        }
    }
}

2、编写拦截器

package com.gw.manage.cms.interceptor;
 
import com.ne.ice.boot.common.ThreadContext;
import org.apache.commons.lang.StringUtils;
import org.springframework.web.servlet.HandlerInterceptor;
 
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
 
import static com.gw.manage.cms.interceptor.DataFilterInterceptor.*;
 
public class DataFilterHandlerInterceptor implements HandlerInterceptor {
    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) {
        ThreadContext.put(CLIENT_ID_KEY, request.getHeader(CLIENT_ID_KEY));
        ThreadContext.put(BRAND_ID_KEY, request.getHeader(BRAND_ID_KEY));
        ThreadContext.put(DEALER_SHOP_IDS_KEY, StringUtils.split(request.getHeader(DEALER_SHOP_IDS_KEY), ","));
        ThreadContext.put(BRAND_IDS_KEY, StringUtils.split(request.getHeader(BRAND_IDS_KEY), ","));
        ThreadContext.put(EMPLOYEE_ID_KEY, request.getHeader(EMPLOYEE_ID_KEY));
        return true;
    }
 
    @Override
    public void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex) {
        ThreadContext.remove(CLIENT_ID_KEY);
        ThreadContext.remove(BRAND_ID_KEY);
        ThreadContext.remove(DEALER_SHOP_IDS_KEY);
        ThreadContext.remove(BRAND_IDS_KEY);
        ThreadContext.remove(EMPLOYEE_ID_KEY);
    }
}