Skip to content

Commit 04ceaea

Browse files
committed
added array of subtypes property for codegen model
1 parent 452e77b commit 04ceaea

File tree

1 file changed

+23
-0
lines changed

1 file changed

+23
-0
lines changed

modules/swagger-codegen/src/main/java/io/swagger/codegen/v3/CodegenModel.java

Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@ public class CodegenModel extends CodegenObject {
2020
public CodegenModel parentModel;
2121
public List<CodegenModel> interfaceModels;
2222
public List<CodegenModel> children;
23+
public List<CodegenModel> subTypes;
2324

2425
public String name, classname, title, description, classVarName, modelJson, dataType, xmlPrefix, xmlNamespace, xmlName;
2526
public String classFilename; // store the class file name, mainly used for import
@@ -115,6 +116,10 @@ public boolean equals(Object o) {
115116
return false;
116117
if (interfaceModels != null ? !interfaceModels.equals(that.interfaceModels) : that.interfaceModels != null)
117118
return false;
119+
if (children != null ? !children.equals(that.children) : that.children != null)
120+
return false;
121+
if (subTypes != null ? !subTypes.equals(that.subTypes) : that.subTypes != null)
122+
return false;
118123
if (name != null ? !name.equals(that.name) : that.name != null)
119124
return false;
120125
if (classname != null ? !classname.equals(that.classname) : that.classname != null)
@@ -173,6 +178,8 @@ public int hashCode() {
173178
result = 31 * result + (parentSchema != null ? parentSchema.hashCode() : 0);
174179
result = 31 * result + (interfaces != null ? interfaces.hashCode() : 0);
175180
result = 31 * result + (parentModel != null ? parentModel.hashCode() : 0);
181+
result = 31 * result + (children != null ? children.hashCode() : 0);
182+
result = 31 * result + (subTypes != null ? subTypes.hashCode() : 0);
176183
result = 31 * result + (interfaceModels != null ? interfaceModels.hashCode() : 0);
177184
result = 31 * result + (name != null ? name.hashCode() : 0);
178185
result = 31 * result + (classname != null ? classname.hashCode() : 0);
@@ -251,6 +258,22 @@ public void setChildren(List<CodegenModel> children) {
251258
this.children = children;
252259
}
253260

261+
public List<CodegenModel> getSubTypes() {
262+
return subTypes;
263+
}
264+
265+
public void setSubTypes(List<CodegenModel> subTypes) {
266+
this.subTypes = subTypes;
267+
}
268+
269+
public CodegenModel addSubType(CodegenModel subType) {
270+
if (this.subTypes == null) {
271+
this.subTypes = new ArrayList<>();
272+
}
273+
this.subTypes.add(subType);
274+
return this;
275+
}
276+
254277
public String getName() {
255278
return name;
256279
}

0 commit comments

Comments
 (0)