sprintboot伪造HttpServletRequest

import org.springframework.mock.web.MockHttpServletRequest;
import org.springframework.mock.web.MockHttpSession;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;

public class YourJobClass {

    public void yourJobMethod() {
        // 伪造ServletRequestAttributes
        ServletRequestAttributes requestAttributes = createServletRequestAttributes();

        try {
            // 将伪造的ServletRequestAttributes设置为当前线程的上下文
            RequestContextHolder.setRequestAttributes(requestAttributes);

            // 在这里调用需要request信息的接口
            // 这里可以获取Session并进行相关操作
            MockHttpServletRequest request = (MockHttpServletRequest) requestAttributes.getRequest();
            MockHttpSession session = (MockHttpSession) request.getSession();
            
            // 设置请求参数
            request.setParameter("param1", "value1");
            request.setParameter("param2", "value2");

            // 设置Session属性
            session.setAttribute("yourAttribute", "yourValue");

            // 在这里可以继续进行其他操作

        } finally {
            // 清除当前线程的上下文信息
            RequestContextHolder.resetRequestAttributes();
        }
    }

    private ServletRequestAttributes createServletRequestAttributes() {
        // 在这里创建伪造的ServletRequestAttributes
        // 使用MockHttpServletRequest模拟HttpServletRequest
        MockHttpServletRequest request = new MockHttpServletRequest();
        request.setMethod("GET");
        request.setRequestURI("/your/endpoint");
        // 设置其他需要的request信息

        // 创建MockHttpSession并将其设置到HttpServletRequest中
        MockHttpSession session = new MockHttpSession();
        request.setSession(session);

        // 创建ServletRequestAttributes并将模拟的HttpServletRequest设置进去
        ServletRequestAttributes requestAttributes = new ServletRequestAttributes(request);

        return requestAttributes;
    }
}