package com.qianwen.core.boot.request;
|
|
import java.io.BufferedReader;
|
import java.io.ByteArrayInputStream;
|
import java.io.IOException;
|
import java.io.InputStreamReader;
|
import javax.servlet.ReadListener;
|
import javax.servlet.ServletInputStream;
|
import javax.servlet.http.HttpServletRequest;
|
import javax.servlet.http.HttpServletRequestWrapper;
|
import com.qianwen.core.tool.utils.WebUtil;
|
|
/* loaded from: blade-core-boot-9.3.0.0-SNAPSHOT.jar:org/springblade/core/boot/request/BladeHttpServletRequestWrapper.class */
|
public class BladeHttpServletRequestWrapper extends HttpServletRequestWrapper {
|
private final HttpServletRequest orgRequest;
|
private byte[] body;
|
|
public BladeHttpServletRequestWrapper(HttpServletRequest request) {
|
super(request);
|
this.orgRequest = request;
|
}
|
|
public BufferedReader getReader() throws IOException {
|
return new BufferedReader(new InputStreamReader(getInputStream()));
|
}
|
|
public ServletInputStream getInputStream() throws IOException {
|
if (super.getHeader("Content-Type") == null) {
|
return super.getInputStream();
|
}
|
if (super.getHeader("Content-Type").startsWith("multipart/form-data")) {
|
return super.getInputStream();
|
}
|
if (super.getHeader("Content-Type").startsWith("application/octet-stream")) {
|
return super.getInputStream();
|
}
|
if (this.body == null) {
|
this.body = WebUtil.getRequestBody(super.getInputStream()).getBytes();
|
}
|
final ByteArrayInputStream byteArrayInputStream = new ByteArrayInputStream(this.body);
|
return new ServletInputStream() { // from class: com.qianwen.core.boot.request.BladeHttpServletRequestWrapper.1
|
public int read() {
|
return byteArrayInputStream.read();
|
}
|
|
public boolean isFinished() {
|
return false;
|
}
|
|
public boolean isReady() {
|
return false;
|
}
|
|
public void setReadListener(ReadListener readListener) {
|
}
|
};
|
}
|
|
public HttpServletRequest getOrgRequest() {
|
return this.orgRequest;
|
}
|
|
public static HttpServletRequest getOrgRequest(HttpServletRequest request) {
|
if (request instanceof BladeHttpServletRequestWrapper) {
|
return ((BladeHttpServletRequestWrapper) request).getOrgRequest();
|
}
|
return request;
|
}
|
}
|