import lombok.AccessLevel;
import lombok.NoArgsConstructor;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;
import java.util.Optional;
import javax.annotation.Nullable;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
/**
* 描述:Web相关工具类
*/
@NoArgsConstructor(access = AccessLevel.PRIVATE)
public final class WebUtils {
/**
* 获取当前请求 request
*
* @return 当前请求对象 current request
* @throws NullPointerException 该方法可能会返回 null 值
*/
public static Optional<HttpServletRequest> getCurrentRequest() {
var servletRequestAttributes = (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();
return servletRequestAttributes == null ? Optional.empty() :
Optional.of(servletRequestAttributes.getRequest());
}
/**
* 获取当前请求 request
*
* @return 当前请求对象 current request
* @throws NullPointerException 该方法可能会返回 null 值
*/
public static Optional<HttpServletResponse> getCurrentResponse() {
var servletRequestAttributes = (ServletRequestAttributes)RequestContextHolder.getRequestAttributes();
return servletRequestAttributes == null ? Optional.empty()
: Optional.ofNullable(servletRequestAttributes.getResponse());
}
/**
* 获取当前请求的 url
*
* @return 如果不存在返回 null
*/
@Nullable
public static String getCurrentRequestUrl() {
return getCurrentRequest().map(HttpServletRequest::getRequestURI).orElse(null);
}
/**
* 获取当前请求的 header 头内容
*
* @return 如果不存在返回 null
*/
@Nullable
public static String getCurrentRequestHeader(String headerKey) {
return getCurrentRequest().map((HttpServletRequest s) -> s.getHeader(headerKey)).orElse(null);
}
}