package com.qmtt.tools;

import java.io.BufferedReader;

import java.io.ByteArrayInputStream;

import java.io.IOException;

import java.io.InputStream;

import java.io.InputStreamReader;

import javax.servlet.ReadListener;

import javax.servlet.ServletInputStream;

import javax.servlet.http.HttpServletRequest;

import javax.servlet.http.HttpServletRequestWrapper;

/**

 *

 * @author 流的方式获取JSON数据

 */

public class RequestWrapper extends HttpServletRequestWrapper {

    // 存放JSON数据主体

    private final String body;

    public RequestWrapper(HttpServletRequest request) throws IOException {

        super(request);

        StringBuilder stringBuilder = new StringBuilder();

        BufferedReader bufferedReader = null;

        try {

            InputStream inputStream = request.getInputStream();

            if (inputStream != null) {

                bufferedReader = new BufferedReader(new InputStreamReader(inputStream));

                char[] charBuffer = new char[128];

                int bytesRead = -1;

                while ((bytesRead = bufferedReader.read(charBuffer)) > 0) {

                    stringBuilder.append(charBuffer, 0, bytesRead);

                }

            } else {

                stringBuilder.append("");

            }

        } catch (IOException ex) {

            throw ex;

        } finally {

            if (bufferedReader != null) {

                try {

                    bufferedReader.close();

                } catch (IOException ex) {

                    throw ex;

                }

            }

        }

        body = stringBuilder.toString();

    }

    @Override

    public ServletInputStream getInputStream() throws IOException {

        final ByteArrayInputStream byteArrayInputStream = new ByteArrayInputStream(body.getBytes("UTF-8"));

        ServletInputStream servletInputStream = new ServletInputStream() {

            @Override

            public int read() throws IOException {

                return byteArrayInputStream.read();

            }

            @Override

            public boolean isFinished() {

                return false;

            }

            @Override

            public boolean isReady() {

                return false;

            }

            @Override

            public void setReadListener(ReadListener listener) {

            }

        };

        return servletInputStream;

    }

    @Override

    public BufferedReader getReader() throws IOException {

        return new BufferedReader(new InputStreamReader(this.getInputStream()));

    }

    public String getBody() {

        return this.body;

    }

}

    @Override

    public void doFilter(ServletRequest request, ServletResponse response, FilterChain chain)

            throws IOException, ServletException {

        HttpServletRequest req = (HttpServletRequest) request;

        String path = req.getContextPath();

        String method = req.getServletPath();

        Map map = request.getParameterMap();

        String msg = "";

        RequestWrapper wrapper = new RequestWrapper(req);

        if (map != null)

            msg = JsonUtils.toJsonString(map);

        log.info(String.format("请求路径:%s,请求方法%s,请求参数：%s-%s", path, method, msg, wrapper.getBody()));

        chain.doFilter(wrapper, response);

    }