🐞 fix(gen): pg gen of insert (#1591)

Co-authored-by: kurimi1 <d0n41df@gmail.com>
This commit is contained in:
toutou_o
2022-03-01 19:53:23 +08:00
committed by GitHub
parent fba6543b23
commit cc4c4928e0
5 changed files with 70 additions and 12 deletions

View File

@@ -6,7 +6,7 @@ import (
"github.com/zeromicro/go-zero/tools/goctl/util/pathx"
)
func genImports(withCache, timeImport bool, table Table) (string, error) {
func genImports(withCache, timeImport bool, table Table, postgreSql bool) (string, error) {
if withCache {
text, err := pathx.LoadTemplate(category, importsTemplateFile, template.Imports)
if err != nil {
@@ -23,10 +23,18 @@ func genImports(withCache, timeImport bool, table Table) (string, error) {
return buffer.String(), nil
}
text, err := pathx.LoadTemplate(category, importsWithNoCacheTemplateFile, template.ImportsNoCache)
if err != nil {
return "", err
var text string
var err error
if postgreSql {
text, err = pathx.LoadTemplate(category, importsWithNoCacheTemplateFile, template.ImportsNoCachePg)
if err != nil {
return "", err
}
} else {
text, err = pathx.LoadTemplate(category, importsWithNoCacheTemplateFile, template.ImportsNoCache)
if err != nil {
return "", err
}
}
buffer, err := util.With("import").Parse(text).Execute(map[string]interface{}{