zeromicro/go-zero
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

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

Browse Source 
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
Download Patch File Download Diff File Expand all files Collapse all files

30
tools/goctl/model/sql/gen/insert.go
View File

@@ -46,9 +46,19 @@ func genInsert(table Table, withCache, postgreSql bool) (string, string, error)
}
camel := table.Name.ToCamel()
text, err := pathx.LoadTemplate(category, insertTemplateFile, template.Insert)
if err != nil {
return "", "", err
var text string
var err error
// if database is postgresql
if postgreSql {
text, err = pathx.LoadTemplate(category, insertTemplateFile, template.InsertPg)
if err != nil {
return "", "", err
}
} else {
text, err = pathx.LoadTemplate(category, insertTemplateFile, template.Insert)
if err != nil {
return "", "", err
}
}
output, err := util.With("insert").
@@ -69,9 +79,17 @@ func genInsert(table Table, withCache, postgreSql bool) (string, string, error)
}
// interface method
text, err = pathx.LoadTemplate(category, insertTemplateMethodFile, template.InsertMethod)
if err != nil {
return "", "", err
// if database is postgresql
if postgreSql {
text, err = pathx.LoadTemplate(category, insertTemplateMethodFile, template.InsertMethodPg)
if err != nil {
return "", "", err
}
} else {
text, err = pathx.LoadTemplate(category, insertTemplateMethodFile, template.InsertMethod)
if err != nil {
return "", "", err
}
}
insertMethodOutput, err := util.With("insertMethod").Parse(text).Execute(map[string]interface{}{