github/SpringBootCodeGenerator
1
0
Fork 0
mirror of https://github.com/moshowgame/SpringBootCodeGenerator.git synced 2025-12-26 05:48:33 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #114 from fuuqiu/master

Browse Source 
add swagger-yml.ftl
This commit is contained in:
Moshow郑锴 2021-08-05 15:31:00 +08:00 committed by GitHub
commit 875f21bc54
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
4 changed files with 37 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

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

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

@ -189,9 +189,24 @@ public class TableParseUtil {
} else {
fieldName = columnName;
}
// field class
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
// 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 class TableParseUtil {
}
FieldInfo fieldInfo = new FieldInfo();
//
fieldInfo.setColumnName(columnName);
fieldInfo.setFieldName(fieldName);
fieldInfo.setFieldClass(fieldClass);
fieldInfo.setSwaggerClass(swaggerClass);
fieldInfo.setFieldComment(fieldComment);
fieldList.add(fieldInfo);

5
generator-web/src/main/resources/template.json
View File

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

12
generator-web/src/main/resources/templates/code-generator/util/swagger-yml.ftl Normal file
View File

@ -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>