add swagger-yml.ftl

Author: fuuqiu

generator-web/src/main/java/com/softdev/system/generator/entity/FieldInfo.java

@@ -13,6 +13,7 @@ public class FieldInfo {
     private String columnName;
     private String fieldName;
     private String fieldClass;
+    private String swaggerClass;
     private String fieldComment;
 
 }

generator-web/src/main/java/com/softdev/system/generator/util/TableParseUtil.java

@@ -189,9 +189,24 @@ public static ClassInfo processTableIntoClassInfo(ParamInfo paramInfo)
                     } else {
                         fieldName = columnName;
                     }
+                    columnLine = columnLine.substring(columnLine.indexOf("`") + 1).trim();
 
+                    //swagger class
+                    String swaggerClass = "string" ;
+                    if (columnLine.contains(" tinyint")) {
+                        swaggerClass = "integer";
+                    } else if (columnLine.contains(" int") || columnLine.contains(" smallint")) {
+                        swaggerClass = "integer";
+                    } else if (columnLine.contains(" bigint")) {
+                        swaggerClass = "integer";
+                    } else if (columnLine.contains(" float")) {
+                        swaggerClass = "number";
+                    } else if (columnLine.contains(" double")) {
+                        swaggerClass = "number";
+                    }  else if (columnLine.contains(" boolean")) {
+                        swaggerClass = "boolean";
+                    }
                     // field class
-                    columnLine = columnLine.substring(columnLine.indexOf("`") + 1).trim();
                     // int(11) NOT NULL AUTO_INCREMENT COMMENT '用户ID',
                     String fieldClass = Object.class.getSimpleName();
                     //2018-9-16 zhengk 补充char/clob/blob/json等类型，如果类型未知，默认为String
@@ -287,9 +302,11 @@ public static ClassInfo processTableIntoClassInfo(ParamInfo paramInfo)
                     }
 
                     FieldInfo fieldInfo = new FieldInfo();
+                    //
                     fieldInfo.setColumnName(columnName);
                     fieldInfo.setFieldName(fieldName);
                     fieldInfo.setFieldClass(fieldClass);
+                    fieldInfo.setSwaggerClass(swaggerClass);
                     fieldInfo.setFieldComment(fieldComment);
 
                     fieldList.add(fieldInfo);

generator-web/src/main/resources/template.json

@@ -167,6 +167,11 @@
         "id": "73",
         "name": "sql",
         "description": "sql"
+      },
+      {
+        "id": "74",
+        "name": "swagger-yml",
+        "description": "swagger-yml"
       }
     ]
   }

generator-web/src/main/resources/templates/code-generator/util/swagger-yml.ftl

@@ -0,0 +1,12 @@
+
+<#if classInfo.fieldList?exists && classInfo.fieldList?size gt 0>
+${classInfo.className}:
+  type: "object"
+  properties:
+<#list classInfo.fieldList as fieldItem >
+    ${fieldItem.fieldName}:
+      type: ${fieldItem.swaggerClass}
+      description:  <#if isComment?exists && isComment==true>"${fieldItem.fieldComment}"</#if>
+</#list>
+</#if>
+