封装request和response对象使用

原创

已注销 2022-11-01 11:56:21 博主文章分类：java ©著作权

文章标签 封装request json java spring 文章分类 后端开发

©著作权归作者所有：来自51CTO博客作者已注销的原创作品，请联系作者获取转载授权，否则将追究法律责任

import cn.hutool.json.JSONUtil;
import com.anchnet.core.model.ContextUserHolder;
import lombok.extern.slf4j.Slf4j;
import org.springframework.web.context.ContextLoader;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;

import javax.servlet.ServletContext;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.HttpSession;
import java.util.Enumeration;
import java.util.HashMap;
import java.util.Map;

@Slf4j
public class RequestContextHolderUtils {

    public static HttpServletRequest getRequest() {
        return getRequestAttributes().getRequest();
    }

    public static HttpServletResponse getResponse() {
        return getRequestAttributes().getResponse();
    }

    public static HttpSession getSession() {
        return getRequest().getSession();
    }

    public static ServletRequestAttributes getRequestAttributes() {
        return ((ServletRequestAttributes) RequestContextHolder.getRequestAttributes());
    }

    public static ServletContext getServletContext() {
        return ContextLoader.getCurrentWebApplicationContext().getServletContext();
    }

    public static Map<String, Object> getMapHolder() {
        Map<String, Object> map = new HashMap<>();
        Enumeration<String> headerNames = RequestContextHolderUtils.getRequest().getHeaderNames();
        while (headerNames.hasMoreElements()) {
            String key = headerNames.nextElement();
            // 排除Cookie字段
            if (key.equalsIgnoreCase("Cookie")) {
                continue;
            }
            String value = RequestContextHolderUtils.getRequest().getHeader(key);
            log.info("key = " + key + ", value = " + value);
            map.put(key, value);
        }

        return map;
    }

    public static ContextUserHolder getUserHolder() {
        String jsonStr = JSONUtil.toJsonStr(getMapHolder());
        log.info("ContextUserHolder : " + jsonStr);
        ContextUserHolder userHolder = JSONUtil.toBean(jsonStr, ContextUserHolder.class);
        return userHolder;
    }
}