org.jboss.resteasy.plugins.interceptors.ServerContentEncodingAnnotationFeature Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of resteasy-jaxrs
The newest version!
package org.jboss.resteasy.plugins.interceptors;

import org.jboss.resteasy.annotations.ContentEncoding;

import javax.ws.rs.ConstrainedTo;
import javax.ws.rs.RuntimeType;
import javax.ws.rs.container.DynamicFeature;
import javax.ws.rs.container.ResourceInfo;
import javax.ws.rs.core.Configuration;
import javax.ws.rs.core.FeatureContext;
import java.lang.annotation.Annotation;
import java.lang.reflect.Method;
import java.util.HashSet;
import java.util.Set;

/**
 * @author Bill Burke
 * @version $Revision: 1 $
 */
@ConstrainedTo(RuntimeType.SERVER)
public class ServerContentEncodingAnnotationFeature implements DynamicFeature
{
   @Override
   public void configure(ResourceInfo resourceInfo, FeatureContext configurable)
   {
      final Class declaring = resourceInfo.getResourceClass();
      final Method method = resourceInfo.getResourceMethod();

      if (declaring == null || method == null) return;
      Set encodings = getEncodings(method.getAnnotations());
      if (encodings.size() <= 0)
      {
         encodings = getEncodings(declaring.getAnnotations());
         if (encodings.size() <= 0) return;
      }
      // check if GZIP encoder has been registered
      if (!isGZipRegistered(configurable.getConfiguration()))
      {
         encodings.remove("gzip");
      }
      configurable.register(createFilter(encodings));
   }
   
   protected boolean isGZipRegistered(Configuration configuration)
   {
      return configuration.isRegistered(GZIPEncodingInterceptor.class);
   }

   protected ServerContentEncodingAnnotationFilter createFilter(Set encodings)
   {
      return new ServerContentEncodingAnnotationFilter(encodings);
   }

   protected Set getEncodings(Annotation[] annotations)
   {
      Set encodings = new HashSet();
      for (Annotation annotation : annotations)
      {
         if (annotation.annotationType().isAnnotationPresent(ContentEncoding.class))
         {
           encodings.add(annotation.annotationType().getAnnotation(ContentEncoding.class).value().toLowerCase());
         }
      }
      return encodings;
   }
}