|
28 | 28 | import java.util.Map; |
29 | 29 | import java.util.Random; |
30 | 30 | import java.util.concurrent.TimeUnit; |
| 31 | +import java.util.zip.Deflater; |
| 32 | +import java.util.zip.DeflaterOutputStream; |
| 33 | +import java.util.zip.GZIPOutputStream; |
31 | 34 |
|
32 | 35 | import javax.servlet.ServletException; |
33 | 36 | import javax.servlet.http.Cookie; |
@@ -125,6 +128,62 @@ private void handleHelper(Request baseRequest, HttpServletRequest request, |
125 | 128 | respondJSON(servletResponse, os, response); |
126 | 129 | baseRequest.setHandled(true); |
127 | 130 | return; |
| 131 | + } else if (method.equals("GET") && uri.equals("/gzip")) { |
| 132 | + Utils.copy(is, Utils.NULL_OUTPUT_STREAM); |
| 133 | + |
| 134 | + JSONObject response = new JSONObject(); |
| 135 | + response.put("args", mapParametersToJSON(request)); |
| 136 | + response.put("headers", mapHeadersToJSON(request)); |
| 137 | + response.put("origin", request.getRemoteAddr()); |
| 138 | + response.put("url", getFullURL(request)); |
| 139 | + response.put("gzipped", true); |
| 140 | + |
| 141 | + byte[] uncompressed = response.toString(/*indent=*/ 2).getBytes( |
| 142 | + StandardCharsets.UTF_8); |
| 143 | + ByteArrayOutputStream baos = new ByteArrayOutputStream( |
| 144 | + uncompressed.length); |
| 145 | + try (GZIPOutputStream gzipos = new GZIPOutputStream(baos)) { |
| 146 | + gzipos.write(uncompressed); |
| 147 | + } |
| 148 | + byte[] compressed = baos.toByteArray(); |
| 149 | + |
| 150 | + servletResponse.setContentLength(compressed.length); |
| 151 | + servletResponse.setHeader("Content-Encoding", "gzip"); |
| 152 | + servletResponse.setContentType("application/json"); |
| 153 | + servletResponse.setStatus(HttpServletResponse.SC_OK); |
| 154 | + os.write(compressed); |
| 155 | + os.flush(); |
| 156 | + baseRequest.setHandled(true); |
| 157 | + return; |
| 158 | + } else if (method.equals("GET") && uri.equals("/deflate")) { |
| 159 | + Utils.copy(is, Utils.NULL_OUTPUT_STREAM); |
| 160 | + |
| 161 | + JSONObject response = new JSONObject(); |
| 162 | + response.put("args", mapParametersToJSON(request)); |
| 163 | + response.put("headers", mapHeadersToJSON(request)); |
| 164 | + response.put("origin", request.getRemoteAddr()); |
| 165 | + response.put("url", getFullURL(request)); |
| 166 | + response.put("deflated", true); |
| 167 | + |
| 168 | + byte[] uncompressed = response.toString(/*indent=*/ 2).getBytes( |
| 169 | + StandardCharsets.UTF_8); |
| 170 | + ByteArrayOutputStream baos = new ByteArrayOutputStream( |
| 171 | + uncompressed.length); |
| 172 | + try (DeflaterOutputStream dos = new DeflaterOutputStream( |
| 173 | + baos, new Deflater(Deflater.DEFAULT_COMPRESSION, |
| 174 | + /*nowrap=*/ true))) { |
| 175 | + dos.write(uncompressed); |
| 176 | + } |
| 177 | + byte[] compressed = baos.toByteArray(); |
| 178 | + |
| 179 | + servletResponse.setContentLength(compressed.length); |
| 180 | + servletResponse.setHeader("Content-Encoding", "deflate"); |
| 181 | + servletResponse.setContentType("application/json"); |
| 182 | + servletResponse.setStatus(HttpServletResponse.SC_OK); |
| 183 | + os.write(compressed); |
| 184 | + os.flush(); |
| 185 | + baseRequest.setHandled(true); |
| 186 | + return; |
128 | 187 | } else if (method.equals("GET") && uri.equals("/cache")) { |
129 | 188 | Utils.copy(is, Utils.NULL_OUTPUT_STREAM); |
130 | 189 |
|
|
0 commit comments