Fehler im Zusammenhang mit dem Komprimieren gepufferter Antworten korrigiert
[percentcodec] / cachecontrol / src / main / java / de / halbekunst / juplo / cachecontrol / CacheControl.java
index 77e2004..0512de4 100644 (file)
@@ -1,13 +1,15 @@
 package de.halbekunst.juplo.cachecontrol;
 
+import de.halbekunst.juplo.cachecontrol.AcceleratorFilter.AccelerationWrapper;
 import de.halbekunst.juplo.cachecontrol.annotations.CacheSeconds;
 import de.halbekunst.juplo.cachecontrol.annotations.Accepts;
+import de.halbekunst.juplo.cachecontrol.annotations.AdditionalHeaders;
 import de.halbekunst.juplo.cachecontrol.annotations.LastModified;
 import de.halbekunst.juplo.cachecontrol.annotations.ETag;
 import java.lang.annotation.Annotation;
-import java.lang.reflect.InvocationTargetException;
 import java.lang.reflect.Method;
 import java.util.Date;
+import java.util.HashMap;
 import java.util.Map;
 import java.util.Map.Entry;
 import java.util.TreeMap;
@@ -15,46 +17,43 @@ import javax.servlet.http.HttpServletRequest;
 import javax.servlet.http.HttpServletResponse;
 import org.slf4j.Logger;
 import org.slf4j.LoggerFactory;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.beans.factory.annotation.Qualifier;
+import org.springframework.stereotype.Component;
 
 /**
  *
  * @author kai
  */
+@Component
 public class CacheControl {
   private final static Logger log = LoggerFactory.getLogger(CacheControl.class);
 
-  public static final String HEADER_DATE = "Date";
-  public static final String HEADER_CACHE_CONTROL = "Cache-Control";
-  public static final String HEADER_LAST_MODIFIED = "Last-Modified";
-  public static final String HEADER_ETAG = "ETag";
-  public static final String HEADER_EXPIRES = "Expires";
-  public static final String HEADER_PRAGMA = "Pragma";
-  public static final String HEADER_IF_MODIFIED_SINCE = "If-Modified-Since";
-  public static final String HEADER_IF_NONE_MATCH = "If-None-Match";
-  public static final String HEADER_RANGE = "Range";
-
   private static final ThreadLocal<CacheMethodHandle> tl = new ThreadLocal<CacheMethodHandle>();
 
-  private Integer defaultCacheSeconds;
-  private Long defaultLastModified;
+  @Autowired @Qualifier("cacheSeconds") private Integer defaultCacheSeconds;
+  @Autowired @Qualifier("lastModified") private Long defaultLastModified;
+
 
+  public void init(CacheMethodHandle handle) {
+    CacheControl.tl.set(handle);
+  }
 
-  public void init(Object handler) throws Exception {
-    if (CacheControl.tl.get() == null)
-      CacheControl.tl.set(new ReflectionCacheMethodHandle(handler));
+  void init(Object handler, AccelerationWrapper wrapper) throws NoSuchMethodException {
+    CacheControl.tl.set(new ReflectionCacheMethodHandle(handler, wrapper == null ? false : wrapper.zipped));
   }
 
   public boolean decorate(
       HttpServletRequest request,
       HttpServletResponse response,
       Object handler
-      ) throws Exception
+      )
   {
     try {
-    CacheMethodHandle controller = CacheControl.tl.get();
+    CacheMethodHandle handle = CacheControl.tl.get();
 
     /** Doppelte Ausführung verhindern... */
-    if (controller == null) {
+    if (handle == null) {
       /** Dekoration wurde bereits durchgeführt! */
       return true;
     }
@@ -64,10 +63,10 @@ public class CacheControl {
      * http://www.w3.org/Protocols/rfc2616/rfc2616-sec14.html#sec14.18 RFC 2616,
      * Abschnitt 14.18} einen Date-Header enthalten
      */
-    response.setDateHeader(HEADER_DATE, controller.getTimestamp());
+    response.setDateHeader(Headers.HEADER_DATE, handle.getTimestamp());
 
     /** Besondere Maßnahmen für besondere HTTP-Status-Codes ?!? */
-    int status = controller.accepts(request);
+    int status = handle.accepts(request);
     switch (status) {
       case HttpServletResponse.SC_OK: // 200
       case HttpServletResponse.SC_NO_CONTENT: // 204
@@ -95,11 +94,6 @@ public class CacheControl {
       case HttpServletResponse.SC_NOT_IMPLEMENTED: // 501
       case HttpServletResponse.SC_SERVICE_UNAVAILABLE: // 503
       case HttpServletResponse.SC_HTTP_VERSION_NOT_SUPPORTED: // 505
-        /**
-         * Ein Fehlercode kann stellvertretend für den Handler gesendet werden,
-         * da im Fehlerfall eh keine weiteren Daten ausgegeben werden!
-         */
-        response.sendError(status);
         return true;
       default:
         /**
@@ -109,6 +103,13 @@ public class CacheControl {
         return false;
     }
 
+    if (handle.isZipped())
+      response.addHeader(Headers.HEADER_CONTENT_ENCODING, "gzip");
+
+    Map<String,String> headers = handle.getAdditionalHeaders(request);
+    for (String name : headers.keySet())
+      response.addHeader(name, headers.get(name));
+
     String url = null;
     if (log.isDebugEnabled()) {
       if (request.getQueryString() == null) {
@@ -123,29 +124,29 @@ public class CacheControl {
       }
     }
 
-    int cacheSeconds = controller.getCacheSeconds(request);
+    int cacheSeconds = handle.getCacheSeconds(request);
     if (cacheSeconds < 1) {
       log.debug("{}: caching disabled!", url);
-      response.setDateHeader(HEADER_DATE, controller.getTimestamp());
-      response.setDateHeader(HEADER_EXPIRES, 0);
-      response.addHeader(HEADER_PRAGMA, "no-cache");
-      response.addHeader(HEADER_CACHE_CONTROL, "private");
-      response.addHeader(HEADER_CACHE_CONTROL, "no-cache");
-      response.addHeader(HEADER_CACHE_CONTROL, "no-store");
-      response.addHeader(HEADER_CACHE_CONTROL, "max-age=0");
-      response.addHeader(HEADER_CACHE_CONTROL, "s-max-age=0");
+      response.setDateHeader(Headers.HEADER_DATE, handle.getTimestamp());
+      response.setDateHeader(Headers.HEADER_EXPIRES, 0);
+      response.addHeader(Headers.HEADER_PRAGMA, "no-cache");
+      response.addHeader(Headers.HEADER_CACHE_CONTROL, "private");
+      response.addHeader(Headers.HEADER_CACHE_CONTROL, "no-cache");
+      response.addHeader(Headers.HEADER_CACHE_CONTROL, "no-store");
+      response.addHeader(Headers.HEADER_CACHE_CONTROL, "max-age=0");
+      response.addHeader(Headers.HEADER_CACHE_CONTROL, "s-max-age=0");
       return true;
     }
 
     long ifModifiedSince = -1;
     try {
-      ifModifiedSince = request.getDateHeader(HEADER_IF_MODIFIED_SINCE);
+      ifModifiedSince = request.getDateHeader(Headers.HEADER_IF_MODIFIED_SINCE);
     }
     catch (Exception e) {
       log.error("Exception while fetching If-Modified-Since: {}", e);
     }
 
-    long lastModified = controller.getLastModified(request);
+    long lastModified = handle.getLastModified(request);
 
     /**
      * Sicherstellen, dass der Wert keine Millisekunden enthält, da die
@@ -154,8 +155,8 @@ public class CacheControl {
      */
     lastModified = lastModified - (lastModified % 1000);
 
-    String ifNoneMatch = request.getHeader(HEADER_IF_NONE_MATCH);
-    String eTag = controller.getETag(request);
+    String ifNoneMatch = request.getHeader(Headers.HEADER_IF_NONE_MATCH);
+    String eTag = handle.getETag(request);
 
     /**
      * 304-Antworten sollen nach dem {@plainlink
@@ -165,12 +166,12 @@ public class CacheControl {
      */
     if (eTag != null) {
       StringBuilder builder = new StringBuilder();
-      if (controller.isETagWeak())
+      if (handle.isETagWeak())
         builder.append("W/");
       builder.append('"');
       builder.append(eTag);
       builder.append('"');
-      response.setHeader(HEADER_ETAG, builder.toString());
+      response.setHeader(Headers.HEADER_ETAG, builder.toString());
     }
 
 
@@ -204,13 +205,13 @@ public class CacheControl {
         ifNoneMatch = ifNoneMatch.substring(1, ifNoneMatch.length() - 1);
       }
 
-      if (!weak || (request.getMethod().equals("GET") && request.getHeader(HEADER_RANGE) == null)) {
+      if (!weak || (request.getMethod().equals("GET") && request.getHeader(Headers.HEADER_RANGE) == null)) {
         /**
          * Die Gleichheit gilt nur, wenn die ETag's der Anfrage _und_ der
          * Antwort stark sind (starke Gleichheit!), oder wenn die Antwort nur
          * schwache Gleichheit fordert...
          */
-        if (ifNoneMatch.equals(eTag) && (controller.isETagWeak() || !weak)) {
+        if (ifNoneMatch.equals(eTag) && (handle.isETagWeak() || !weak)) {
           log.debug("{}: ETag {} not changed -> 304 ", url, ifNoneMatch);
           response.setStatus(HttpServletResponse.SC_NOT_MODIFIED);
           return false;
@@ -225,7 +226,7 @@ public class CacheControl {
     log.debug("{}: first up!", url);
 
     /** HTTP/1.1-Caching-Header richtig setzen!! */
-    response.setDateHeader(HEADER_LAST_MODIFIED, lastModified);
+    response.setDateHeader(Headers.HEADER_LAST_MODIFIED, lastModified);
 
     /** Cache-Control für HTTP/1.1-Clients generieren */
     Map<String, String> cacheControl = new TreeMap<String, String>();
@@ -248,11 +249,11 @@ public class CacheControl {
        * <code>Expires</code>-Header für HTTP/1.0-Clients setzen.
        */
       cacheControl.put("max-age", Integer.toString(cacheSeconds));
-      response.setDateHeader(HEADER_EXPIRES, (controller.getTimestamp() + (long) cacheSeconds * 1000));
+      response.setDateHeader(Headers.HEADER_EXPIRES, (handle.getTimestamp() + (long) cacheSeconds * 1000));
     }
 
     /** Dem Handler die Gelegenheit geben, den Cache-Controll-Header anzupassen */
-    controller.cacheControl(request, cacheControl);
+    handle.cacheControl(request, cacheControl);
 
 
     if (cacheControl.containsKey("private")) {
@@ -264,8 +265,8 @@ public class CacheControl {
        * http://www.w3.org/Protocols/rfc2616/rfc2616-sec14.html#sec14.32
        * Abschnitt 14.32})
        */
-      response.setDateHeader(HEADER_EXPIRES, 0l);
-      response.addHeader(HEADER_PRAGMA, "no-cache");
+      response.setDateHeader(Headers.HEADER_EXPIRES, 0l);
+      response.addHeader(Headers.HEADER_PRAGMA, "no-cache");
     }
 
     StringBuilder builder = new StringBuilder();
@@ -276,7 +277,7 @@ public class CacheControl {
         builder.append('=');
         builder.append(entry.getValue());
       }
-      response.addHeader(HEADER_CACHE_CONTROL, builder.toString());
+      response.addHeader(Headers.HEADER_CACHE_CONTROL, builder.toString());
     }
 
     return true;
@@ -296,17 +297,6 @@ public class CacheControl {
   }
 
 
-  interface CacheMethodHandle {
-    long getTimestamp();
-    int accepts(HttpServletRequest request) throws Exception;
-    int getCacheSeconds(HttpServletRequest request) throws Exception;
-    long getLastModified(HttpServletRequest request) throws Exception;
-    String getETag(HttpServletRequest request) throws Exception;
-    boolean isETagWeak();
-    void cacheControl(HttpServletRequest request, Map<String, String> cacheControlMap) throws Exception;
-  }
-
-
   class ReflectionCacheMethodHandle implements CacheMethodHandle {
 
     private Object handler;
@@ -314,22 +304,27 @@ public class CacheControl {
     private Integer cacheSeconds;
     private Long lastModified;
     private String eTag;
+    private Map<String,String> additionalHeaders;
     private Method acceptsMethod;
     private Method cacheSecondsMethod;
     private Method lastModifiedMethod;
     private Method eTagMethod;
     private Method cacheControlMethod;
+    private Method additionalHeadersMethod;
     private boolean isAcceptsMethodDefined;
     private boolean isCacheSecondsMethodDefined;
     private boolean isLastModifiedMethodDefined;
     private boolean isETagMethodDefined;
     private boolean isCacheControlMethodDefined;
+    private boolean isAdditionalHeadersMethodDefined;
     private boolean weak;
+    private boolean zipped;
 
 
-    ReflectionCacheMethodHandle(Object handler) throws NoSuchMethodException {
+    ReflectionCacheMethodHandle(Object handler, boolean zipped) throws NoSuchMethodException {
 
       this.handler = handler;
+      this.zipped = zipped;
 
       cacheSeconds = CacheControl.this.defaultCacheSeconds;
       lastModified = CacheControl.this.defaultLastModified;
@@ -362,6 +357,23 @@ public class CacheControl {
           isETagMethodDefined = true;
           continue;
         }
+        if (annotation.annotationType().equals(AdditionalHeaders.class)) {
+          AdditionalHeaders additionalHeadersAnnotation = (AdditionalHeaders)annotation;
+          additionalHeaders = new HashMap<String,String>();
+          for (String header : additionalHeadersAnnotation.value()) {
+            int i = header.indexOf(':');
+            if (i < 0) {
+              log.error("invalid header: [{}]", header);
+            }
+            else {
+              String name = header.substring(0,i).trim();
+              String value = header.substring(i+1,header.length()).trim();
+              additionalHeaders.put(name, value);
+            }
+          }
+          isAdditionalHeadersMethodDefined = true;
+          continue;
+        }
       }
 
       /** Method-Level-Annotations auslesen */
@@ -403,62 +415,89 @@ public class CacheControl {
             isCacheControlMethodDefined = true;
             continue;
           }
+          if (annotation.annotationType().equals(AdditionalHeaders.class)) {
+            if (isAdditionalHeadersMethodDefined)
+              throw new IllegalArgumentException("Die Annotation @AdditionalHeaders wurde in der Klasse " + handler.getClass().getSimpleName() + " mehrfach verwendet!");
+            additionalHeadersMethod = method;
+            isAdditionalHeadersMethodDefined = true;
+            continue;
+          }
         }
       }
+
+      if (!isAdditionalHeadersMethodDefined)
+        additionalHeaders = new HashMap<String,String>();
     }
 
 
+    @Override
+    public boolean isZipped() {
+      return zipped;
+    }
+
     @Override
     public long getTimestamp() {
       return now;
     }
 
     @Override
-    public int accepts(HttpServletRequest request)
-        throws IllegalAccessException,
-               IllegalArgumentException,
-               InvocationTargetException
-    {
-      if (acceptsMethod == null)
+    public int accepts(HttpServletRequest request) throws IllegalArgumentException {
+      if (acceptsMethod == null) {
         return HttpServletResponse.SC_OK;
-      else
-        return (Integer)acceptsMethod.invoke(handler, request);
+      }
+      else {
+        try {
+          return (Integer)acceptsMethod.invoke(handler, request);
+        }
+        catch (Exception e) {
+          throw new IllegalArgumentException(e);
+        }
+      }
     }
 
     @Override
-    public int getCacheSeconds(HttpServletRequest request)
-        throws IllegalAccessException,
-               IllegalArgumentException,
-               InvocationTargetException
-    {
-      if (cacheSecondsMethod == null)
+    public int getCacheSeconds(HttpServletRequest request) throws IllegalArgumentException {
+      if (cacheSecondsMethod == null) {
         return cacheSeconds;
-      else
-        return (Integer)cacheSecondsMethod.invoke(handler, request);
+      }
+      else {
+        try {
+          return (Integer)cacheSecondsMethod.invoke(handler, request);
+        }
+        catch (Exception e) {
+          throw new IllegalArgumentException(e);
+        }
+      }
     }
 
     @Override
-    public long getLastModified(HttpServletRequest request)
-        throws IllegalAccessException,
-               IllegalArgumentException,
-               InvocationTargetException
-    {
-      if (lastModifiedMethod == null)
+    public long getLastModified(HttpServletRequest request) throws IllegalArgumentException {
+      if (lastModifiedMethod == null) {
         return lastModified;
-      else
-        return (Long)lastModifiedMethod.invoke(handler, request);
+      }
+      else {
+        try {
+          return (Long)lastModifiedMethod.invoke(handler, request);
+        }
+        catch (Exception e) {
+          throw new IllegalArgumentException(e);
+        }
+      }
     }
 
     @Override
-    public String getETag(HttpServletRequest request)
-        throws IllegalAccessException,
-               IllegalArgumentException,
-               InvocationTargetException
-    {
-      if (eTagMethod == null)
+    public String getETag(HttpServletRequest request) throws IllegalArgumentException {
+      if (eTagMethod == null) {
         return eTag;
-      else
-        return (String)eTagMethod.invoke(handler, request);
+      }
+      else {
+        try {
+          return (String)eTagMethod.invoke(handler, request);
+        }
+        catch (Exception e) {
+          throw new IllegalArgumentException(e);
+        }
+      }
     }
 
     @Override
@@ -471,21 +510,31 @@ public class CacheControl {
         HttpServletRequest request,
         Map<String, String> cacheControlMap
         )
-        throws IllegalAccessException,
-               IllegalArgumentException,
-               InvocationTargetException
+        throws IllegalArgumentException
     {
-      if (cacheControlMethod != null)
-        cacheControlMethod.invoke(handler, request, cacheControlMap);
+      if (cacheControlMethod != null) {
+        try {
+          cacheControlMethod.invoke(handler, request, cacheControlMap);
+        }
+        catch (Exception e) {
+          throw new IllegalArgumentException(e);
+        }
+      }
     }
-  }
-
-
-  public void setDefaultCacheSeconds(Integer defaultCacheSeconds) {
-    this.defaultCacheSeconds = defaultCacheSeconds;
-  }
 
-  public void setDefaultLastModified(Long defaultLastModified) {
-    this.defaultLastModified = defaultLastModified;
+    @Override
+    public Map<String,String> getAdditionalHeaders(HttpServletRequest request) throws IllegalArgumentException {
+      if (additionalHeadersMethod == null) {
+        return additionalHeaders;
+      }
+      else {
+        try {
+          return (Map<String,String>)additionalHeadersMethod.invoke(handler, request);
+        }
+        catch (Exception e) {
+          throw new IllegalArgumentException(e);
+        }
+      }
+    }
   }
 }