Fix/issue#1447 (#1458)

* Add data for template to render

* fix #1447

Co-authored-by: anqiansong <anqiansong@bytedance.com>
This commit is contained in:
anqiansong
2022-01-18 10:36:38 +08:00
committed by GitHub
parent bf2feee5b7
commit e57fa8ff53
12 changed files with 33 additions and 14 deletions

View File

@@ -9,11 +9,11 @@ import (
"github.com/tal-tech/go-zero/tools/goctl/util/pathx"
)
func genFields(fields []*parser.Field) (string, error) {
func genFields(table Table, fields []*parser.Field) (string, error) {
var list []string
for _, field := range fields {
result, err := genField(field)
result, err := genField(table, field)
if err != nil {
return "", err
}
@@ -24,8 +24,8 @@ func genFields(fields []*parser.Field) (string, error) {
return strings.Join(list, "\n"), nil
}
func genField(field *parser.Field) (string, error) {
tag, err := genTag(field.NameOriginal)
func genField(table Table, field *parser.Field) (string, error) {
tag, err := genTag(table, field.NameOriginal)
if err != nil {
return "", err
}
@@ -43,6 +43,7 @@ func genField(field *parser.Field) (string, error) {
"tag": tag,
"hasComment": field.Comment != "",
"comment": field.Comment,
"data": table,
})
if err != nil {
return "", err