feat(ReverseEngine):给生成的文件添加表注释,当表注释不为空的时候会在import下面生成该注释信息

This commit is contained in:
huangzj
2020-05-26 09:12:17 +08:00
parent e9502e1ac8
commit f74ba73d29
4 changed files with 17 additions and 9 deletions
+12 -4
View File
@@ -8,13 +8,14 @@ import (
type FileContentSupport struct{}
func (FileContentSupport) GenerateFileContent(filePath string, fileName string, fms []entity.FieldMessage, hasTime bool, tagKey string) string {
func (FileContentSupport) GenerateFileContent(filePath string, fileName string, fms []entity.FieldMessage, hasTime bool, tagKey string, tableComment string) string {
var build strings.Builder
pName := splitLast(filePath, static.Splice)
writePackage(&build, pName) //写文件的引用包
writeImportIfTime(&build, hasTime) //如果有time.Time类型的需要写入import
buildStructHead(&build, fileName) //写入结构体头
writePackage(&build, pName) //写文件的引用包
writeImportIfTime(&build, hasTime) //如果有time.Time类型的需要写入import
writeTableComment(&build, tableComment) //写入表注释
buildStructHead(&build, fileName) //写入结构体头
//对于没有字段的返回空定义的文件
if len(fms) == 0 {
@@ -99,3 +100,10 @@ func splitLast(filePath string, split string) string {
ss := strings.Split(filePath, split)
return ss[len(ss)-1]
}
func writeTableComment(builder *strings.Builder, s string) {
if s != "" {
builder.WriteString("//" + s)
builder.WriteString("\n")
}
}