loafle/openapi-generator
3
0
Fork 0
forked from loafle/openapi-generator-original
Code Pull Requests Activity

added recursion check

Browse Source
This commit is contained in:
Tony Tam 2015-03-25 11:11:04 -07:00
parent e8052f0505
commit d2f5b9ea65
1 changed files with 19 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
modules/swagger-codegen/src/main/java/com/wordnik/swagger/codegen/examples/ExampleGenerator.java
View File

@ -24,6 +24,7 @@ public class ExampleGenerator {
public List<Map<String, String>> generate(Map<String, String> examples, List<String> mediaTypes, Property property) { public List<Map<String, String>> generate(Map<String, String> examples, List<String> mediaTypes, Property property) {
List<Map<String, String>> output = new ArrayList<Map<String, String>>(); List<Map<String, String>> output = new ArrayList<Map<String, String>>();
Set<String> processedModels = new HashSet<String>();
if(examples == null ) { if(examples == null ) {
if(mediaTypes == null) { if(mediaTypes == null) {
// assume application/json for this // assume application/json for this
@ -33,7 +34,7 @@ public class ExampleGenerator {
Map<String, String> kv = new HashMap<String, String>(); Map<String, String> kv = new HashMap<String, String>();
kv.put("contentType", mediaType); kv.put("contentType", mediaType);
if(property != null && mediaType.startsWith("application/json")) { if(property != null && mediaType.startsWith("application/json")) {
String example = Json.pretty(resolvePropertyToExample(mediaType, property)); String example = Json.pretty(resolvePropertyToExample(mediaType, property, processedModels));
if(example != null) { if(example != null) {
example = example.replaceAll("\n", "\\\\n"); example = example.replaceAll("\n", "\\\\n");
@ -69,7 +70,7 @@ public class ExampleGenerator {
return output; return output;
} }
protected Object resolvePropertyToExample(String mediaType, Property property) { protected Object resolvePropertyToExample(String mediaType, Property property, Set<String> processedModels) {
if(property.getExample() != null) { if(property.getExample() != null) {
return property.getExample(); return property.getExample();
} }
@ -83,7 +84,7 @@ public class ExampleGenerator {
Property innerType = ((ArrayProperty)property).getItems(); Property innerType = ((ArrayProperty)property).getItems();
if(innerType != null) { if(innerType != null) {
Object[] output = new Object[]{ Object[] output = new Object[]{
resolvePropertyToExample(mediaType, innerType) resolvePropertyToExample(mediaType, innerType, processedModels)
}; };
return output; return output;
} }
@ -116,11 +117,11 @@ public class ExampleGenerator {
Map<String, Object> mp = new HashMap<String, Object>(); Map<String, Object> mp = new HashMap<String, Object>();
if(property.getName() != null) { if(property.getName() != null) {
mp.put(property.getName(), mp.put(property.getName(),
resolvePropertyToExample(mediaType, ((MapProperty)property).getAdditionalProperties())); resolvePropertyToExample(mediaType, ((MapProperty)property).getAdditionalProperties(), processedModels));
} }
else { else {
mp.put("key", mp.put("key",
resolvePropertyToExample(mediaType, ((MapProperty)property).getAdditionalProperties())); resolvePropertyToExample(mediaType, ((MapProperty)property).getAdditionalProperties(), processedModels));
} }
return mp; return mp;
} }
@ -128,9 +129,10 @@ public class ExampleGenerator {
return "{}"; return "{}";
} }
else if(property instanceof RefProperty) { else if(property instanceof RefProperty) {
Model model = examples.get(((RefProperty)property).getSimpleRef()); String simpleName = ((RefProperty)property).getSimpleRef();
Model model = examples.get(simpleName);
if(model != null) if(model != null)
return resolveModelToExample(mediaType, model); return resolveModelToExample(simpleName, mediaType, model, processedModels);
} }
else if(property instanceof UUIDProperty) { else if(property instanceof UUIDProperty) {
return "046b6c7f-0b8a-43b9-b35d-6489e6daee91"; return "046b6c7f-0b8a-43b9-b35d-6489e6daee91";
@ -139,14 +141,20 @@ public class ExampleGenerator {
return ""; return "";
} }
public Object resolveModelToExample(String mediaType, Model model) { public Object resolveModelToExample(String name, String mediaType, Model model, Set<String> processedModels) {
if(processedModels.contains(name)) {
return "";
}
if(model instanceof ModelImpl) { if(model instanceof ModelImpl) {
processedModels.add(name);
ModelImpl impl = (ModelImpl) model; ModelImpl impl = (ModelImpl) model;
Map<String, Object> values = new HashMap<String, Object>(); Map<String, Object> values = new HashMap<String, Object>();
for(String name : impl.getProperties().keySet()) { if(impl != null && impl.getProperties() != null) {
Property property = impl.getProperties().get(name); for(String propertyName : impl.getProperties().keySet()) {
values.put(name, resolvePropertyToExample(mediaType, property)); Property property = impl.getProperties().get(propertyName);
values.put(propertyName, resolvePropertyToExample(mediaType, property, processedModels));
}
} }
return values; return values;