功能描述
在SpringBoot中如要實(shí)現(xiàn)記錄接口被調(diào)用的頻率和生成api日志宛琅，以便查看接口使用情況闪水，那么監(jiān)聽所有api請(qǐng)求的功能就誕生了报破。
功能實(shí)現(xiàn)
一任连、自定義request繼承HttpServletRequestWrapper
import java.io.BufferedReader;
import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.io.InputStreamReader;
import java.nio.charset.Charset;
import javax.servlet.ReadListener;
import javax.servlet.ServletInputStream;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletRequestWrapper;
import com.alibaba.fastjson.JSONObject;
public class MyRequestWrapper extends HttpServletRequestWrapper {
private byte[] body;
private ServletInputStreamWrapper inputStreamWrapper;
private Object requestBody;
public MyRequestWrapper(HttpServletRequest request) throws IOException {
super(request);
int le = request.getContentLength();
if (le > 0) {
body = new byte[le];
request.getInputStream().read(body);
requestBody = JSONObject.parse(body, 0, body.length, Charset.forName("UTF-8").newDecoder(), 0);
} else {
body = new byte[0];
requestBody = "";
}
ByteArrayInputStream byteArrayInputStream = new ByteArrayInputStream(this.body);
this.inputStreamWrapper = new ServletInputStreamWrapper(byteArrayInputStream);
resetInputStream();
}
public String getRequestBody() {
return String.valueOf(requestBody);
}
public void setRequestBody(Object requestBody) {
this.requestBody = requestBody;
}
private void resetInputStream() {
this.inputStreamWrapper.setInputStream(new ByteArrayInputStream(this.body != null ? this.body : new byte[0]));
}
@Override
public ServletInputStream getInputStream() throws IOException {
return this.inputStreamWrapper;
}
@Override
public BufferedReader getReader() throws IOException {
return new BufferedReader(new InputStreamReader(this.inputStreamWrapper));
}
private static class ServletInputStreamWrapper extends ServletInputStream {
private InputStream inputStream;
public ServletInputStreamWrapper(InputStream inputStream) {
super();
this.inputStream = inputStream;
}
public InputStream getInputStream() {
return inputStream;
}
public void setInputStream(InputStream inputStream) {
this.inputStream = inputStream;
}
@Override
public boolean isFinished() {
return true;
}
@Override
public boolean isReady() {
return false;
}
@Override
public void setReadListener(ReadListener readListener) {
}
@Override
public int read() throws IOException {
return this.inputStream.read();
}
}
}
二蚤吹、自定義response繼承HttpServletResponseWrapper
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.OutputStreamWriter;
import java.io.PrintWriter;
import java.nio.charset.Charset;
import javax.servlet.ServletOutputStream;
import javax.servlet.WriteListener;
import javax.servlet.http.HttpServletResponse;
import javax.servlet.http.HttpServletResponseWrapper;
import com.alibaba.fastjson.JSONObject;
public class MyResponseWrapper extends HttpServletResponseWrapper {
private ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream();
private HttpServletResponse response;
private Object reponseBody;
public MyResponseWrapper(HttpServletResponse response) {
super(response);
this.response = response;
}
public String getResponseBody() {
byte[] bytes = byteArrayOutputStream.toByteArray();
reponseBody = JSONObject.parse(bytes, 0, bytes.length, Charset.forName("UTF-8").newDecoder(), 0);
return String.valueOf(reponseBody);
}
@Override
public ServletOutputStream getOutputStream() {
return new ServletOutputStreamWrapper(this.byteArrayOutputStream, this.response);
}
@Override
public PrintWriter getWriter() throws IOException {
return new PrintWriter(
new OutputStreamWriter(this.byteArrayOutputStream, this.response.getCharacterEncoding()));
}
private static class ServletOutputStreamWrapper extends ServletOutputStream {
private ByteArrayOutputStream outputStream;
private HttpServletResponse response;
public ServletOutputStreamWrapper(ByteArrayOutputStream outputStream, HttpServletResponse response) {
this.outputStream = outputStream;
this.response = response;
}
@Override
public boolean isReady() {
return true;
}
@Override
public void setWriteListener(WriteListener listener) {
}
@Override
public void write(int b) throws IOException {
this.outputStream.write(b);
}
@Override
public void flush() throws IOException {
if (!this.response.isCommitted()) {
byte[] body = this.outputStream.toByteArray();
ServletOutputStream outputStream = this.response.getOutputStream();
outputStream.write(body);
outputStream.flush();
}
}
}
三、新建一個(gè)類RequestWrapperFilter繼承OncePerRequestFilter随抠，重寫doFilterInternal()方法裁着，具體如下。
import java.io.IOException;
import java.text.SimpleDateFormat;
import java.util.Date;
import javax.servlet.FilterChain;
import javax.servlet.ServletException;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import org.springframework.stereotype.Component;
import org.springframework.web.filter.OncePerRequestFilter;
import com.lkl.web.maven1.api.help.MyRequestWrapper;
import com.lkl.web.maven1.api.help.MyResponseWrapper;
/**
* 監(jiān)聽所有請(qǐng)求
*
* @author Administrator
*
*/
@Component
public class RequestWrapperFilter extends OncePerRequestFilter {
private MyRequestWrapper requestWrapper;
private MyResponseWrapper reponseWrapper;
@Override
protected void doFilterInternal(HttpServletRequest request, HttpServletResponse response, FilterChain filterChain)
throws ServletException, IOException {
try {
requestWrapper = new MyRequestWrapper(request);
reponseWrapper = new MyResponseWrapper(response);
filterChain.doFilter(requestWrapper, reponseWrapper);
String requestBody = requestWrapper.getRequestBody();
String reponseBody = reponseWrapper.getResponseBody();
//做你想做的事情
} catch (Exception e) {
//失敗時(shí)拱她，默認(rèn)即可
filterChain.doFilter(request, response);
}
}
}
簡(jiǎn)單總結(jié)
1.自定義reques和response無非就是想從流中拿到byte[]二驰，因?yàn)樵趏utputStream和inputStream中read()后再次read()會(huì)讀不到數(shù)據(jù)。
2.有時(shí)候會(huì)因?yàn)楦鞣N請(qǐng)求體或響應(yīng)體為空導(dǎo)致doFilterInternal失敗秉沼，因此加個(gè)try catch塊桶雀。
3.注意RequestWrapperFilter中標(biāo)注@Component。
4.保存日志有可能是高頻率的io操作唬复，可通過定時(shí)器來定時(shí)矗积，選擇性保存日志。
以上都是我的個(gè)人理解敞咧，如有不足棘捣，請(qǐng)?jiān)谠u(píng)論區(qū)指出。不勝感激休建。
