diff --git a/internal/app/module/beegopro/container.go b/internal/app/module/beegopro/container.go index 2d2cea7..ac0df98 100644 --- a/internal/app/module/beegopro/container.go +++ b/internal/app/module/beegopro/container.go @@ -18,7 +18,7 @@ const MDateFormat = "20060102_150405" var DefaultBeegoPro = &Container{ BeegoProFile: system.CurrentDir + "/beegopro.toml", - TimestampFile: system.CurrentDir + "/beegopro.timestamp", + TimestampFile: system.CurrentDir + "/.beegopro.timestamp", GoModFile: system.CurrentDir + "/go.mod", UserOption: UserOption{ Debug: false, diff --git a/internal/app/module/beegopro/util.go b/internal/app/module/beegopro/util.go index d7dd2a0..79f02a3 100644 --- a/internal/app/module/beegopro/util.go +++ b/internal/app/module/beegopro/util.go @@ -76,7 +76,7 @@ func (c *RenderFile) write(filename string, buf string) (err error) { func isNeedOverwrite(fileName string) (flag bool) { seg := "//" ext := filepath.Ext(fileName) - if ext == "sql" { + if ext == ".sql" { seg = "--" } @@ -179,7 +179,7 @@ func getModelType(orm string) (inputType, goType, mysqlType, tag string) { } // user set orm tag if len(kv) == 2 { - tag = orm + tag = kv[1] } return }