Merge pull request #11299 from swagger-api/codegen-generators-issue-966

Codegen generators issue 966

Author: HugoMario

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

@@ -20,6 +20,7 @@ public class CodegenModel extends CodegenObject {
     public CodegenModel parentModel;
     public List<CodegenModel> interfaceModels;
     public List<CodegenModel> children;
+    public List<CodegenModel> subTypes;
 
     public String name, classname, title, description, classVarName, modelJson, dataType, xmlPrefix, xmlNamespace, xmlName;
     public String classFilename; // store the class file name, mainly used for import
@@ -115,6 +116,8 @@ public boolean equals(Object o) {
             return false;
         if (interfaceModels != null ? !interfaceModels.equals(that.interfaceModels) : that.interfaceModels != null)
             return false;
+        if (subTypes != null ? !subTypes.equals(that.subTypes) : that.subTypes != null)
+            return false;
         if (name != null ? !name.equals(that.name) : that.name != null)
             return false;
         if (classname != null ? !classname.equals(that.classname) : that.classname != null)
@@ -173,6 +176,7 @@ public int hashCode() {
         result = 31 * result + (parentSchema != null ? parentSchema.hashCode() : 0);
         result = 31 * result + (interfaces != null ? interfaces.hashCode() : 0);
         result = 31 * result + (parentModel != null ? parentModel.hashCode() : 0);
+        result = 31 * result + (subTypes != null ? subTypes.hashCode() : 0);
         result = 31 * result + (interfaceModels != null ? interfaceModels.hashCode() : 0);
         result = 31 * result + (name != null ? name.hashCode() : 0);
         result = 31 * result + (classname != null ? classname.hashCode() : 0);
@@ -251,6 +255,22 @@ public void setChildren(List<CodegenModel> children) {
         this.children = children;
     }
 
+    public List<CodegenModel> getSubTypes() {
+        return subTypes;
+    }
+
+    public void setSubTypes(List<CodegenModel> subTypes) {
+        this.subTypes = subTypes;
+    }
+
+    public CodegenModel addSubType(CodegenModel subType) {
+        if (this.subTypes == null) {
+            this.subTypes = new ArrayList<>();
+        }
+        this.subTypes.add(subType);
+        return this;
+    }
+
     public String getName() {
         return name;
     }