JAVA修改HttpServletRequest中的header信息

最新推荐文章于 2024-09-23 10:27:35 发布

徐晨松

最新推荐文章于 2024-09-23 10:27:35 发布

阅读量50

点赞数 1

文章标签： java 开发语言

import lombok.extern.slf4j.Slf4j;
import org.apache.commons.collections.MapUtils;
import org.apache.tomcat.util.http.MimeHeaders;
import org.springframework.util.StringUtils;
 
import javax.servlet.http.HttpServletRequest;
import java.lang.reflect.Field;
import java.util.Map;
 
/**
 * HTTP请求工具类，提供操作HttpServletRequest的扩展功能
 */
@Slf4j
public class HttpRequestUtil {
 
    /**
     * 设置请求头信息
     *
     * @param request  HttpServletRequest对象，用于获取和设置请求头信息
     * @param headers  包含需要设置的请求头键值对的Map
     */
    public static void setHeaders(HttpServletRequest request, Map<String, String> headers) {
        // 如果请求头信息为空，则直接返回，避免不必要的操作
        if (MapUtils.isEmpty(headers)) {
            return;
        }
        try {
            // 获取请求内部使用的MimeHeaders对象，用于操作HTTP请求头
            MimeHeaders mimeHeaders = HttpRequestUtil.getMimeHeaders(request);
            // 遍历所有需要设置的请求头，逐一添加到MimeHeaders中
            headers.forEach((key, value) -> {
                if (!StringUtils.isEmpty(key)) {
                    mimeHeaders.setValue(key).setString(value);
                }
            });
        } catch (Exception e) {
            // 捕获异常并记录错误日志，避免异常影响业务流程
            log.error(e.getMessage(), e);
        }
    }
 
    /**
     * 反射获取HttpServletRequest内部的MimeHeaders对象
     *
     * @param request  需要获取MimeHeaders的HttpServletRequest对象
     * @return 返回MimeHeaders对象，用于操作HTTP请求头
     * @throws NoSuchFieldException 如果内部字段不存在
     * @throws IllegalAccessException 如果无法访问内部字段
     */
    private static MimeHeaders getMimeHeaders(HttpServletRequest request) throws NoSuchFieldException, IllegalAccessException {
        // 获取HttpServletRequest实现类的Class对象
        Class<? extends HttpServletRequest> requestClass = request.getClass();
        // 获取名为"request"的内部字段，该字段持有实际的请求对象
        Field requestField = requestClass.getDeclaredField("request");
        requestField.setAccessible(true);
        Object requestObject = requestField.get(request);
        // 获取名为"coyoteRequest"的字段，该字段持有CoyoteRequest对象
        Field coyoteRequestField = requestObject.getClass().getDeclaredField("coyoteRequest");
        coyoteRequestField.setAccessible(true);
        Object coyoteRequestObject = coyoteRequestField.get(requestObject);
        // 获取名为"headers"的字段，该字段持有MimeHeaders对象
        Field headerField = coyoteRequestObject.getClass().getDeclaredField("headers");
        headerField.setAccessible(true);
        return (MimeHeaders) headerField.get(coyoteRequestObject);
    }
}