mirror of
https://github.com/zeromicro/go-zero.git
synced 2025-01-24 18:00:24 +08:00
36 lines
874 B
Go
36 lines
874 B
Go
package gen
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/zeromicro/go-zero/tools/goctl/model/sql/template"
|
|
"github.com/zeromicro/go-zero/tools/goctl/util"
|
|
"github.com/zeromicro/go-zero/tools/goctl/util/pathx"
|
|
)
|
|
|
|
func genNew(table Table, withCache, postgreSql bool) (string, error) {
|
|
text, err := pathx.LoadTemplate(category, modelNewTemplateFile, template.New)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
t := fmt.Sprintf(`"%s"`, wrapWithRawString(table.Name.Source(), postgreSql))
|
|
if postgreSql {
|
|
t = "`" + fmt.Sprintf(`"%s"."%s"`, table.Db.Source(), table.Name.Source()) + "`"
|
|
}
|
|
|
|
output, err := util.With("new").
|
|
Parse(text).
|
|
Execute(map[string]any{
|
|
"table": t,
|
|
"withCache": withCache,
|
|
"upperStartCamelObject": table.Name.ToCamel(),
|
|
"data": table,
|
|
})
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
return output.String(), nil
|
|
}
|