Fixed review comments - Indentation, switch to if/else

joelittlejohn · Dec 30, 2015 · 641342b · 641342b
1 parent 2b95767
commit 641342b
Showing 1 changed file with 7 additions and 10 deletions.
diff --git a/jsonschema2pojo-core/src/main/java/org/jsonschema2pojo/SchemaMapper.java b/jsonschema2pojo-core/src/main/java/org/jsonschema2pojo/SchemaMapper.java
@@ -122,16 +122,13 @@ public JType generate(JCodeModel codeModel, String className, String packageName
 
         ObjectMapper mapper = new ObjectMapper();
         JsonNode schemaNode = null;
-        switch (ruleFactory.getGenerationConfig().getSourceType()) {
-		case JSON:
-			JsonNode jsonNode = mapper.readTree(json);
-	        schemaNode = schemaGenerator.schemaFromExample(jsonNode);
-			break;
-		default:
-			//Default is JSONSCHEMA, for backward compatibility.
-			schemaNode = mapper.readTree(json);
-			break;
-		}
+        if (ruleFactory.getGenerationConfig().getSourceType() == SourceType.JSON) {
+            JsonNode jsonNode = mapper.readTree(json);
+            schemaNode = schemaGenerator.schemaFromExample(jsonNode);
+        } else {
+            schemaNode = mapper.readTree(json);
+        }
+
         return ruleFactory.getSchemaRule().apply(className, schemaNode, jpackage, new Schema(null, schemaNode, schemaNode));
     }
 }