fix for #939

Author: fehguy

modules/swagger-jaxrs/src/main/java/com/wordnik/swagger/jaxrs/config/BeanConfig.java

@@ -5,6 +5,8 @@
 import com.wordnik.swagger.config.*;
 
 import com.wordnik.swagger.annotations.Api;
+import com.wordnik.swagger.core.filter.*;
+import com.wordnik.swagger.config.FilterFactory;
 
 import org.reflections.Reflections;
 import org.reflections.scanners.ResourcesScanner;
@@ -13,10 +15,15 @@
 import org.reflections.util.ClasspathHelper;
 import org.reflections.util.ConfigurationBuilder;
 
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
 import java.util.Set;
 import java.util.HashSet;
 
 public class BeanConfig extends AbstractScanner implements Scanner, SwaggerConfig {
+  Logger LOGGER = LoggerFactory.getLogger(BeanConfig.class);
+
   Reader reader = new Reader(new Swagger());
 
   String resourcePackage;
@@ -202,6 +209,17 @@ public Swagger configure(Swagger swagger) {
       for(String scheme : schemes)
         swagger.scheme(Scheme.forValue(scheme));
     }
+    if(filterClass != null) {
+      try {
+        SwaggerSpecFilter filter = (SwaggerSpecFilter) Class.forName(filterClass).newInstance();
+        if(filter != null) {
+          FilterFactory.setFilter(filter);
+        }
+      }
+      catch (Exception e) {
+        LOGGER.error("failed to load filter", e);
+      }
+    }
     return swagger.info(info)
       .host(host)
       .basePath(basePath);

modules/swagger-jaxrs/src/main/java/com/wordnik/swagger/jaxrs/config/ReflectiveJaxrsScanner.java

@@ -3,7 +3,9 @@
 import com.wordnik.swagger.annotations.Api;
 import com.wordnik.swagger.config.Scanner;
 import com.wordnik.swagger.config.SwaggerConfig;
+import com.wordnik.swagger.config.FilterFactory;
 import com.wordnik.swagger.models.Swagger;
+import com.wordnik.swagger.core.filter.*;
 
 import org.reflections.Reflections;
 import org.reflections.scanners.ResourcesScanner;
@@ -58,6 +60,18 @@ public void setResourcePackage(String resourcePackage) {
   }
 
   public Swagger configure(Swagger swagger) {
+    if(filterClass != null) {
+      try {
+        SwaggerSpecFilter filter = (SwaggerSpecFilter) Class.forName(filterClass).newInstance();
+        if(filter != null) {
+          FilterFactory.setFilter(filter);
+        }
+      }
+      catch (Exception e) {
+        LOGGER.error("failed to load filter", e);
+      }
+    }
+
     return swagger;
   }