test request.getOutputStream

Author: netbuffer

src/main/java/cn/netbuffer/filter/ModifyHttpResponseFilter.java

@@ -3,11 +3,13 @@
 import cn.netbuffer.filter.wrapper.ModifyHttpResponseWrapper;
 import com.alibaba.fastjson.JSON;
 import com.alibaba.fastjson.JSONObject;
+import org.apache.commons.lang3.StringUtils;
 
 import javax.servlet.*;
 import javax.servlet.annotation.WebFilter;
 import javax.servlet.http.HttpServletResponse;
 import java.io.IOException;
+import java.io.PrintWriter;
 
 @WebFilter(filterName = "ModifyHttpResponseFilter", urlPatterns = {"/json/*"})
 public class ModifyHttpResponseFilter implements Filter {
@@ -23,11 +25,14 @@ public void doFilter(ServletRequest servletRequest, ServletResponse servletRespo
         System.out.println("-----------------before ModifyHttpResponseFilter------------------");
         filterChain.doFilter(servletRequest, modifyHttpResponseWrapper);
         System.out.println("-----------------after ModifyHttpResponseFilter------------------");
-        JSONObject jsonObject = JSON.parseObject(modifyHttpResponseWrapper.getResult());
+        JSONObject jsonObject = JSON.parseObject(StringUtils.isBlank(modifyHttpResponseWrapper.getResult()) ? modifyHttpResponseWrapper.getContent() : modifyHttpResponseWrapper.getResult());
         jsonObject.put("msg", "wrapper[" + jsonObject.getString("msg") + "]");
         HttpServletResponse response = (HttpServletResponse) modifyHttpResponseWrapper.getResponse();
         response.setContentType("application/json; charset=UTF-8");
-        response.getWriter().write(jsonObject.toJSONString());
+        PrintWriter printWriter = response.getWriter();
+        printWriter.write(jsonObject.toJSONString());
+        printWriter.flush();
+        printWriter.close();
     }
 
     @Override

src/main/java/cn/netbuffer/filter/wrapper/ModifyHttpResponseWrapper.java

@@ -1,39 +1,81 @@
 package cn.netbuffer.filter.wrapper;
 
+import javax.servlet.ServletOutputStream;
+import javax.servlet.WriteListener;
 import javax.servlet.http.HttpServletResponse;
 import javax.servlet.http.HttpServletResponseWrapper;
+import java.io.ByteArrayOutputStream;
 import java.io.CharArrayWriter;
 import java.io.IOException;
 import java.io.PrintWriter;
 
 public class ModifyHttpResponseWrapper extends HttpServletResponseWrapper {
 
-    private PrintWriter cacheWriter;
     private CharArrayWriter bufferWriter;
+    private ByteArrayOutputStream byteArrayOutputStream = new ByteArrayOutputStream();
+    private HttpServletResponse httpServletResponse;
 
     public ModifyHttpResponseWrapper(HttpServletResponse response) {
         super(response);
         bufferWriter = new CharArrayWriter();
-        cacheWriter = new PrintWriter(bufferWriter);
+        httpServletResponse = response;
+    }
+
+    @Override
+    public ServletOutputStream getOutputStream() {
+        return new ServletOutputStreamWrapper(this.byteArrayOutputStream, this.httpServletResponse);
     }
 
     @Override
     public PrintWriter getWriter() throws IOException {
-        return cacheWriter;
+        return new PrintWriter(bufferWriter);
     }
 
-    /**
-     * 获取原始的HTML页面内容。
-     *
-     * @return
-     */
     public String getResult() {
         return bufferWriter.toString();
     }
 
+    public String getContent() {
+        return new String(byteArrayOutputStream.toByteArray());
+    }
+
     @Override
     public void setContentType(String type) {
         super.setContentType(type);
     }
 
+    private class ServletOutputStreamWrapper extends ServletOutputStream {
+
+        private ByteArrayOutputStream outputStream;
+        private HttpServletResponse response;
+
+        public ServletOutputStreamWrapper(ByteArrayOutputStream byteArrayOutputStream, HttpServletResponse response) {
+            this.outputStream = byteArrayOutputStream;
+            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();
+            }
+        }
+    }
 }

src/main/java/cn/netbuffer/servlet/JSONServlet.java

@@ -18,6 +18,7 @@ protected void doGet(HttpServletRequest request, HttpServletResponse response) t
         data.put("msg", "invoke success for " + this.getClass().getName());
         System.out.println("get json=" + data);
         response.getWriter().write(data.toString());
+//        response.getOutputStream().write(data.toString().getBytes());
     }
 
 }