e57fa8ff53
* Add data for template to render * fix #1447 Co-authored-by: anqiansong <anqiansong@bytedance.com>
42 lines
936 B
Go
42 lines
936 B
Go
package gen
|
|
|
|
import (
|
|
"github.com/tal-tech/go-zero/tools/goctl/model/sql/template"
|
|
"github.com/tal-tech/go-zero/tools/goctl/util"
|
|
"github.com/tal-tech/go-zero/tools/goctl/util/pathx"
|
|
)
|
|
|
|
func genImports(withCache, timeImport bool, table Table) (string, error) {
|
|
if withCache {
|
|
text, err := pathx.LoadTemplate(category, importsTemplateFile, template.Imports)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
buffer, err := util.With("import").Parse(text).Execute(map[string]interface{}{
|
|
"time": timeImport,
|
|
"data": table,
|
|
})
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
return buffer.String(), nil
|
|
}
|
|
|
|
text, err := pathx.LoadTemplate(category, importsWithNoCacheTemplateFile, template.ImportsNoCache)
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
buffer, err := util.With("import").Parse(text).Execute(map[string]interface{}{
|
|
"time": timeImport,
|
|
"data": table,
|
|
})
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
return buffer.String(), nil
|
|
}
|