@@ -20,6 +20,7 @@ public class CodegenModel extends CodegenObject {
20
20
public CodegenModel parentModel ;
21
21
public List <CodegenModel > interfaceModels ;
22
22
public List <CodegenModel > children ;
23
+ public List <CodegenModel > subTypes ;
23
24
24
25
public String name , classname , title , description , classVarName , modelJson , dataType , xmlPrefix , xmlNamespace , xmlName ;
25
26
public String classFilename ; // store the class file name, mainly used for import
@@ -115,6 +116,10 @@ public boolean equals(Object o) {
115
116
return false ;
116
117
if (interfaceModels != null ? !interfaceModels .equals (that .interfaceModels ) : that .interfaceModels != null )
117
118
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 ;
118
123
if (name != null ? !name .equals (that .name ) : that .name != null )
119
124
return false ;
120
125
if (classname != null ? !classname .equals (that .classname ) : that .classname != null )
@@ -173,6 +178,8 @@ public int hashCode() {
173
178
result = 31 * result + (parentSchema != null ? parentSchema .hashCode () : 0 );
174
179
result = 31 * result + (interfaces != null ? interfaces .hashCode () : 0 );
175
180
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 );
176
183
result = 31 * result + (interfaceModels != null ? interfaceModels .hashCode () : 0 );
177
184
result = 31 * result + (name != null ? name .hashCode () : 0 );
178
185
result = 31 * result + (classname != null ? classname .hashCode () : 0 );
@@ -251,6 +258,22 @@ public void setChildren(List<CodegenModel> children) {
251
258
this .children = children ;
252
259
}
253
260
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
+
254
277
public String getName () {
255
278
return name ;
256
279
}
0 commit comments