Merge pull request #617 from bgorven/master

Allow setting custom annotator and custom rule factory fields directly in Gradle
joelittlejohn · Sep 8, 2016 · 0c5f7f6 · 0c5f7f6
2 parents 59602f7 + bd023f6
commit 0c5f7f6
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/...2pojo-gradle-plugin/src/main/groovy/org/jsonschema2pojo/gradle/JsonSchemaExtension.groovy b/...2pojo-gradle-plugin/src/main/groovy/org/jsonschema2pojo/gradle/JsonSchemaExtension.groovy
@@ -134,10 +134,18 @@ public class JsonSchemaExtension implements GenerationConfig {
     customAnnotator = Class.forName(clazz, true, this.class.classLoader)
   }
 
+  public void setCustomAnnotator(Class clazz) {
+    customAnnotator = clazz
+  }
+
   public void setCustomRuleFactory(String clazz) {
     customRuleFactory = Class.forName(clazz, true, this.class.classLoader)
   }
 
+  public void setCustomRuleFactory(Class clazz) {
+    customRuleFactory = clazz
+  }
+
   public void setSourceType(String s) {
     sourceType = SourceType.valueOf(s.toUpperCase())
   }