前言
在开发Spring Boot项目时,我们很多的Controller层代码都需要获取一下,HttpServletRequest、HttpServletResponse 和 HttpSession 等对象,我们普遍的方式是在Controller类下的方法参数中直接获取,例如:
@Slf4j
@ResponseResult
@RestController
@RequestMapping("/users")
public class UserController {
@PostMapping
@ResponseStatus(HttpStatus.CREATED)
public User addUser(@Validated @RequestBody User user, HttpServletRequest request, HttpServletResponse response) {
log.info(JsonUtil.object2Json(request));
log.info(JsonUtil.object2Json(response));
return user;
}
}
但是这样获取,我们有时需要在Service层代码中获取他们的附带信息时,就需要不断的将这两个参数下传到其使用的方法内,这导致有一些方法参数看起来特别别扭,其实获取这几参数的方式有很多种,今天给大家分享一个获取他们的工具类。
代码实现
package com.wenjianbao.util;
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;
/**
* 获取Request/Response工具类
*/
public class RequestContextUtil {
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();
}
}
代码解释
代码的逻辑是很简单的,主要其实就是用了ContextLoader这个工具类,在它的内部会把这些 Request、Response、Session对象都放入ThreadLocal下,当你想获取的时候自然也就是当前线程下的对象信息了。