How to insert response size and time into the page itself, at least partially?

前端 未结 3 1386
抹茶落季
抹茶落季 2020-12-02 21:41

I realize it\'s a chicken and egg problem and that it\'s not possible to accurately resolve the time it took to render a page (or the size of response) and insert that numbe

3条回答
  •  夕颜
    夕颜 (楼主)
    2020-12-02 22:10

    This is a perfect use case for the Apache Commons IO CountingOutputStream. You need to create a Filter which uses HttpServletResponseWrapper to replace the OutputStream of the response with this one and replaces the Writer as well which should wrap the wrapped OutputStream. Then get hold of the HttpServletResponseWrapper instance in the request scope so that you can get the getByteCount() from the CountingOutputStream.

    Here's a kickoff example of the CountingFilter:

    public class CountingFilter implements Filter {
    
        @Override
        public void init(FilterConfig arg0) throws ServletException {
            // NOOP.
        }
    
        @Override
        public void doFilter(ServletRequest request, final ServletResponse response, FilterChain chain) throws IOException, ServletException {
            HttpServletResponse httpres = (HttpServletResponse) response;
            CountingServletResponse counter = new CountingServletResponse(httpres);
            HttpServletRequest httpreq = (HttpServletRequest) request;
            httpreq.setAttribute("counter", counter);
            chain.doFilter(request, counter);
            counter.flushBuffer(); // Push the last bits containing HTML comment.
        }
    
        @Override
        public void destroy() {
            // NOOP.
        }
    
    }
    

    The CountingServletResponse:

    public class CountingServletResponse extends HttpServletResponseWrapper {
    
        private final long startTime;
        private final CountingServletOutputStream output;
        private final PrintWriter writer;
    
        public CountingServletResponse(HttpServletResponse response) throws IOException {
            super(response);
            startTime = System.nanoTime();
            output = new CountingServletOutputStream(response.getOutputStream());
            writer = new PrintWriter(output, true);
        }
    
        @Override
        public ServletOutputStream getOutputStream() throws IOException {
            return output;
        }
    
        @Override
        public PrintWriter getWriter() throws IOException {
            return writer;
        }
    
        @Override
        public void flushBuffer() throws IOException {
            writer.flush();
        }
    
        public long getElapsedTime() {
            return System.nanoTime() - startTime;
        }
    
        public long getByteCount() throws IOException {
            flushBuffer(); // Ensure that all bytes are written at this point.
            return output.getByteCount();
        }
    
    }
    

    The CountingServletOutputStream:

    public class CountingServletOutputStream extends ServletOutputStream {
    
        private final CountingOutputStream output;
    
        public CountingServletOutputStream(ServletOutputStream output) {
            this.output = new CountingOutputStream(output);
        }
    
        @Override
        public void write(int b) throws IOException {
            output.write(b);
        }
    
        @Override
        public void flush() throws IOException {
            output.flush();
        }
    
        public long getByteCount() {
            return output.getByteCount();
        }
    
    }
    

    You can use it in any (even non-JSF) page as follows:

    
    
        
            Counting demo
        
        
            

    Hello World

提交回复
热议问题