me.vcoder.httplogger.TraceableHttpServletRequest Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of http-logger Show documentation
HTTP Logger for Spring Library
The newest version!
package me.vcoder.httplogger;

import java.io.BufferedReader;
import java.io.IOException;
import java.io.StringWriter;
import java.net.URI;
import java.util.ArrayList;
import java.util.Enumeration;
import java.util.LinkedHashMap;
import java.util.List;
import java.util.Map;

import javax.servlet.http.HttpServletRequest;

import org.apache.commons.io.IOUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.util.StringUtils;

/**
 * @author baodn
 * Created on 11 Apr 2018
 */
final class TraceableHttpServletRequest implements TraceableRequest {
    private final static Logger LOGGER = LoggerFactory.getLogger(TraceableHttpServletRequest.class);
    private final HttpServletRequest request;

    TraceableHttpServletRequest(HttpServletRequest request) {
        this.request = request;
    }

    @Override
    public String getMethod() {
        return this.request.getMethod();
    }

    @Override
    public URI getUri() {
        StringBuffer urlBuffer = this.request.getRequestURL();
        if (StringUtils.hasText(this.request.getQueryString())) {
            urlBuffer.append("?");
            urlBuffer.append(this.request.getQueryString());
        }
        return URI.create(urlBuffer.toString());
    }

    @Override
    public Map> getHeaders() {
        return extractHeaders();
    }

    @Override
    public String getRemoteAddress() {
        return this.request.getRemoteAddr();
    }

    private Map> extractHeaders() {
        Map> headers = new LinkedHashMap<>();
        Enumeration names = this.request.getHeaderNames();
        while (names.hasMoreElements()) {
            String name = names.nextElement();
            headers.put(name, toList(this.request.getHeaders(name)));
        }
        return headers;
    }

    private List toList(Enumeration enumeration) {
        List list = new ArrayList<>();
        while (enumeration.hasMoreElements()) {
            list.add(enumeration.nextElement());
        }
        return list;
    }

    @Override
    public String getBody() {
        return getPostData();
    }

    private String getPostData() {
        StringBuilder sb = new StringBuilder();
        try {
            BufferedReader reader = this.request.getReader();
            reader.mark(10000);
            String line;
            do {
                line = reader.readLine();
                if (line != null)
                    sb.append(line).append(System.lineSeparator());
            } while (line != null);
            reader.reset();
            reader.close();
            return sb.toString();
        } catch (IOException e) {
            LOGGER.warn("getPostData couldn't.. get the post data", e);
            return "";
        }
    }
}