You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
46 lines
1.1 KiB
Go
46 lines
1.1 KiB
Go
package gen
|
|
|
|
import (
|
|
"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 genImports(table Table, withCache, timeImport, decimalImport bool) (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]any{
|
|
"time": timeImport,
|
|
"containsPQ": table.ContainsPQ,
|
|
"data": table,
|
|
"decimal": decimalImport,
|
|
})
|
|
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]any{
|
|
"time": timeImport,
|
|
"containsPQ": table.ContainsPQ,
|
|
"data": table,
|
|
"decimal": decimalImport,
|
|
})
|
|
if err != nil {
|
|
return "", err
|
|
}
|
|
|
|
return buffer.String(), nil
|
|
}
|