com.manymobi.servlet.http.log.LogHttpServletRequestWrapper Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of servlet-http-log-spring-boot-starter Show documentation
servlet http log
There is a newer version: 2.0.1
Show newest version
package com.manymobi.servlet.http.log;

import com.manymobi.servlet.http.log.io.PartByteArrayOutputStream;

import javax.servlet.ReadListener;
import javax.servlet.ServletInputStream;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletRequestWrapper;
import java.io.BufferedReader;
import java.io.ByteArrayOutputStream;
import java.io.IOException;
import java.io.InputStreamReader;
import java.io.UnsupportedEncodingException;
import java.util.function.Consumer;

/**
 * @author 梁建军
 * 创建日期： 2018/10/22
 * 创建时间： 14:14
 * @version 1.0
 * @since 1.0
 */
public class LogHttpServletRequestWrapper extends HttpServletRequestWrapper {


    private ServletInputStream servletInputStream;

    private final LogStrategy logStrategy;
    private final Consumer print;

    public LogHttpServletRequestWrapper(HttpServletRequest request, LogStrategy logStrategy, Consumer print) {
        super(request);
        this.logStrategy = logStrategy;
        this.print = print;
    }

    @Override
    public BufferedReader getReader() throws IOException {
        return new BufferedReader(new InputStreamReader(getInputStream(), getContentType()));
    }

    @Override
    public ServletInputStream getInputStream() throws IOException {
        if (servletInputStream == null) {
            servletInputStream = new RetainServletInputStream(super.getInputStream(), this);
        }
        return servletInputStream;
    }

    public void print() throws UnsupportedEncodingException {
        if (servletInputStream instanceof RetainServletInputStream) {
            ((RetainServletInputStream) servletInputStream).print(true);
        }
    }


    /**
     * 保留
     */
    private class RetainServletInputStream extends ServletInputStream {

        private final ServletInputStream servletInputStream;

        private final LogHttpServletRequestWrapper logHttpServletRequestWrapper;
        /**
         * 用于记录请求体，打印完日志就释放
         */
        private ByteArrayOutputStream byteArrayOutputStream;

        private final int readContentLength;

        public RetainServletInputStream(ServletInputStream servletInputStream, LogHttpServletRequestWrapper logHttpServletRequestWrapper) {

            this.servletInputStream = servletInputStream;
            this.logHttpServletRequestWrapper = logHttpServletRequestWrapper;
            int contentLength = logHttpServletRequestWrapper.getContentLength();
            int requestBodyMaxSize = logStrategy.getRequestBodyMaxSize();
            int requestBodyInitialSize = logStrategy.getRequestBodyInitialSize();
            if (requestBodyInitialSize <= 0) {
                if (requestBodyMaxSize <= 0) {
                    byteArrayOutputStream = new ByteArrayOutputStream(contentLength);
                    readContentLength = contentLength;
                } else {
                    int min = Math.min(contentLength, requestBodyMaxSize);
                    byteArrayOutputStream = new PartByteArrayOutputStream(min, min);
                    readContentLength = min;
                }
            } else {
                if (contentLength < requestBodyInitialSize) {
                    byteArrayOutputStream = new PartByteArrayOutputStream(contentLength, requestBodyMaxSize);
                    readContentLength = contentLength;
                } else {
                    byteArrayOutputStream = new PartByteArrayOutputStream(requestBodyInitialSize, requestBodyMaxSize);
                    readContentLength = Math.min(requestBodyMaxSize, contentLength);
                }
            }
        }

        @Override
        public boolean isFinished() {
            return servletInputStream.isFinished();
        }

        @Override
        public boolean isReady() {
            return servletInputStream.isReady();
        }

        @Override
        public void setReadListener(ReadListener listener) {
            servletInputStream.setReadListener(listener);
        }

        @Override
        public int read() throws IOException {
            int b = servletInputStream.read();
            if (b != -1) {
                byteArrayOutputStream.write(b);
            } else {
                print(true);
            }
            print(false);
            return b;
        }


        @Override
        public int read(byte[] b, int off, int len) throws IOException {
            int read = servletInputStream.read(b, off, len);
            if (read != -1) {
                byteArrayOutputStream.write(b, off, read);
            } else {
                print(true);
            }
            print(false);
            return read;
        }

        /**
         * @param force 强制
         */
        private void print(boolean force) throws UnsupportedEncodingException {
            if (byteArrayOutputStream != null) {
                if (force || byteArrayOutputStream.size() >= readContentLength) {
                    print.accept(byteArrayOutputStream.toString(logHttpServletRequestWrapper.getCharacterEncoding()));
                    byteArrayOutputStream = null;
                }
            }
        }
    }
}