X-Git-Url: https://juplo.de/gitweb/?p=percentcodec;a=blobdiff_plain;f=cachecontrol%2Fsrc%2Fmain%2Fjava%2Fde%2Fhalbekunst%2Fjuplo%2Fcachecontrol%2FAcceleratorFilter.java;h=5fee73674e2240348fb0d2760e461f84a0966dfe;hp=7c250a24f47988721b73e2340db392d57cc222ea;hb=84bf389af59e95353a5c4c631decc0c3ea5129ae;hpb=7b440f7f3568028e63280646cfb071652bd81e94 diff --git a/cachecontrol/src/main/java/de/halbekunst/juplo/cachecontrol/AcceleratorFilter.java b/cachecontrol/src/main/java/de/halbekunst/juplo/cachecontrol/AcceleratorFilter.java index 7c250a24..5fee7367 100644 --- a/cachecontrol/src/main/java/de/halbekunst/juplo/cachecontrol/AcceleratorFilter.java +++ b/cachecontrol/src/main/java/de/halbekunst/juplo/cachecontrol/AcceleratorFilter.java @@ -1,5 +1,6 @@ package de.halbekunst.juplo.cachecontrol; +import java.io.ByteArrayOutputStream; import java.io.IOException; import java.io.OutputStream; import java.io.PrintWriter; @@ -72,8 +73,6 @@ public class AcceleratorFilter implements Filter { httpRequest.setAttribute(RESPONSE_WRAPPER, wrapper); cacheControl.init(wrapper); chain.doFilter(request, wrapper); - /** Dekoration auslösen, falls sie bisher nicht ausgelöst wurde... */ - cacheControl.decorate(httpRequest, httpResponse, wrapper); wrapper.finish(); } @@ -92,9 +91,9 @@ public class AcceleratorFilter implements Filter { private final HttpServletResponse response; boolean zipped; // CacheControll greift direkt auf dieses Flag zu! + boolean forceCompression = false; - private int buffer; - private GZIPServletOutputStream out; + private CountingServletOutputStream out; private ServletOutputStream stream; private PrintWriter writer; @@ -124,7 +123,6 @@ public class AcceleratorFilter implements Filter { weak = AcceleratorFilter.this.weak; cacheParams = new HashMap(); - buffer = AcceleratorFilter.this.buffer; zipped = false; Enumeration values = request.getHeaders(Headers.HEADER_ACCEPT_ENCODING); while (values.hasMoreElements()) { @@ -134,12 +132,14 @@ public class AcceleratorFilter implements Filter { break; } } + if (zipped) + out = new GZIPServletOutputStream(); + else + out = new CountingServletOutputStream(); } - public void finish() throws IOException { - if (zipped && out != null) - out.zout.finish(); + out.close(); } @@ -278,7 +278,6 @@ public class AcceleratorFilter implements Filter { throw new IllegalStateException("ServletOutputStream and PrintWriter cannot be requested both!"); if (stream == null) { - out = new GZIPServletOutputStream(); stream = out; } @@ -292,7 +291,6 @@ public class AcceleratorFilter implements Filter { throw new IllegalStateException("ServletOutputStream and PrintWriter cannot be requested both!"); if (writer == null) { - out = new GZIPServletOutputStream(); writer = new PrintWriter(out); } @@ -310,21 +308,14 @@ public class AcceleratorFilter implements Filter { @Override public void setBufferSize(int size) { - if (out != null && this.buffer != out.left) - throw new IllegalStateException("setBufferSize() cannot be called after content has been written!"); - - if (size < 0) - size = 0; - - this.buffer = size; - if (out != null) - out.left = size; + out.setBuffer(size); response.setBufferSize(size); } @Override public void flushBuffer() throws IOException { + forceCompression = true; cacheControl.decorate(request, response, this); response.flushBuffer(); } @@ -333,6 +324,15 @@ public class AcceleratorFilter implements Filter { public void resetBuffer() { response.resetBuffer(); + try { + if (zipped) + out = new GZIPServletOutputStream(); + else + out = new CountingServletOutputStream(); + } + catch (IOException e) { + throw new IllegalStateException(e); + } stream = null; writer = null; } @@ -341,11 +341,20 @@ public class AcceleratorFilter implements Filter { public void reset() { response.reset(); - out = null; + try { + if (zipped) + out = new GZIPServletOutputStream(); + else + out = new CountingServletOutputStream(); + } + catch (IOException e) { + throw new IllegalStateException(e); + } stream = null; writer = null; /** Cookies has been cleared! Reinitialize decorator... */ + forceCompression = false; cacheControl.init(this); } @@ -353,7 +362,7 @@ public class AcceleratorFilter implements Filter { @Override public boolean isZipped() { - return zipped && out != null && !out.empty; + return out.isZipped(); } @Override @@ -545,29 +554,31 @@ public class AcceleratorFilter implements Filter { } - class GZIPServletOutputStream extends ServletOutputStream { + class CountingServletOutputStream extends ServletOutputStream { - private final OutputStream out; - private final GZIPOutputStream zout; + private OutputStream out; int left; boolean empty; - public GZIPServletOutputStream() throws IOException { - if (zipped) { - this.zout = new GZIPOutputStream(response.getOutputStream(), buffer); - this.out = this.zout; - } - else { - this.zout = null; - this.out = response.getOutputStream(); - } - empty = true; + + CountingServletOutputStream() throws IOException { + out = response.getOutputStream(); left = buffer; + empty = true; } - @Override - public void close() throws IOException { + void setBuffer(int size) throws IllegalStateException {} + + boolean isZipped() { + return false; + } + + void finish() throws IOException { + decorate(); + } + + void decorate() throws IOException { try { AcceleratorFilter.this.cacheControl.decorate(AccelerationWrapper.this.request, response, AccelerationWrapper.this); } @@ -575,39 +586,113 @@ public class AcceleratorFilter implements Filter { log.error("Error while guessing Cache-Header's", e); response.sendError(HttpServletResponse.SC_INTERNAL_SERVER_ERROR); } + } - if (!empty && zout != null) - zout.finish(); + + @Override + public void close() throws IOException { + decorate(); out.close(); } @Override public void flush() throws IOException { - try { - AcceleratorFilter.this.cacheControl.decorate(AccelerationWrapper.this.request, response, AccelerationWrapper.this); - } - catch (Exception e) { - log.error("Error while guessing Cache-Header's", e); - response.sendError(HttpServletResponse.SC_INTERNAL_SERVER_ERROR); - } - + decorate(); out.flush(); } @Override public void write(int i) throws IOException { - if (left == 0) { + empty = false; + if (left == 0) + decorate(); + left--; + out.write(i); + } + } + + + final class GZIPServletOutputStream extends CountingServletOutputStream { + + final static int MINIMAL_BUFFER_SIZE = 128; + + private ByteArrayOutputStream buffer; + private OutputStream out; + private GZIPOutputStream zout; + private int bufferSize; + private boolean decorated = false; + + + public GZIPServletOutputStream() throws IOException { + setBuffer(AcceleratorFilter.this.buffer); + } + + + @Override + void finish() throws IOException { + decorate(); + zout.finish(); + super.out.write(buffer.toByteArray()); + } + + @Override + boolean isZipped() { + if (decorated) + return true; + return !empty; + } + + @Override + void setBuffer(int size) throws IllegalStateException { + if (!empty) + throw new IllegalStateException("attemp to change buffer size after writing data to response!"); + + if (size > MINIMAL_BUFFER_SIZE) { + left = size; try { - AcceleratorFilter.this.cacheControl.decorate(AccelerationWrapper.this.request, response, buffer); + bufferSize = size; + out = response.getOutputStream(); + buffer = new ByteArrayOutputStream(size); + zout = new GZIPOutputStream(buffer, size); + super.out = zout; } - catch (Exception e) { - log.error("Error while guessing Cache-Header's", e); - response.sendError(HttpServletResponse.SC_INTERNAL_SERVER_ERROR); + catch (IOException e) { + throw new IllegalStateException(e); } } + } + + + + @Override + public void close() throws IOException { + decorate(); + zout.close(); + out.write(buffer.toByteArray()); + out.close(); + } + + @Override + public void flush() throws IOException { + decorate(); + out.write(buffer.toByteArray()); + out.flush(); + } + + @Override + public void write(int i) throws IOException { empty = false; + if (left == 0) { + if (!decorated) { + decorate(); + decorated = true; + } + out.write(buffer.toByteArray()); + buffer.reset(); + left = bufferSize; + } left--; - out.write(i); + zout.write(i); } } }