自动化代码结构体无需手动添加time包

piexlmax · piexlmax · commit e0c15cc1a470 · 2022-06-30T11:57:24.000+08:00
diff --git a/server/model/system/sys_auto_code.go b/server/model/system/sys_auto_code.go
@@ -17,6 +17,7 @@ type AutoCodeStruct struct {
 	AutoCreateApiToSql bool     `json:"autoCreateApiToSql"` // 是否自动创建api
 	AutoMoveFile       bool     `json:"autoMoveFile"`       // 是否自动移动文件
 	Fields             []*Field `json:"fields,omitempty"`
+	HasTimer           bool
 	DictTypes          []string `json:"-"`
 	Package            string   `json:"package"`
 	PackageT           string   `json:"-"`
diff --git a/server/resource/autocode_template/server/model.go.tpl b/server/resource/autocode_template/server/model.go.tpl
@@ -3,10 +3,10 @@ package {{.Package}}
 
 import (
 	"github.com/flipped-aurora/gin-vue-admin/server/global"
+	{{ if .HasTimer }}"time"{{ end }}
 )
 
 // {{.StructName}} 结构体
-// 如果含有time.Time 请自行import time包
 type {{.StructName}} struct {
       global.GVA_MODEL {{- range .Fields}}
             {{- if eq .FieldType "enum" }}
diff --git a/server/service/system/sys_auto_code.go b/server/service/system/sys_auto_code.go
@@ -149,6 +149,12 @@ var AutoCodeServiceApp = new(AutoCodeService)
 
 func (autoCodeService *AutoCodeService) PreviewTemp(autoCode system.AutoCodeStruct) (map[string]string, error) {
 	makeDictTypes(&autoCode)
+	for i := range autoCode.Fields {
+		if autoCode.Fields[i].FieldType == "time.Time" {
+			autoCode.HasTimer = true
+			break
+		}
+	}
 	dataList, _, needMkdir, err := autoCodeService.getNeedList(&autoCode)
 	if err != nil {
 		return nil, err
@@ -228,6 +234,12 @@ func makeDictTypes(autoCode *system.AutoCodeStruct) {
 
 func (autoCodeService *AutoCodeService) CreateTemp(autoCode system.AutoCodeStruct, ids ...uint) (err error) {
 	makeDictTypes(&autoCode)
+	for i := range autoCode.Fields {
+		if autoCode.Fields[i].FieldType == "time.Time" {
+			autoCode.HasTimer = true
+			break
+		}
+	}
 	// 增加判断: 重复创建struct
 	if autoCode.AutoMoveFile && AutoCodeHistoryServiceApp.Repeat(autoCode.StructName, autoCode.Package) {
 		return RepeatErr
