|
|
@@ -305,15 +305,22 @@ func GenTemplate(templateText string, templateData interface{}, params map[strin
|
|
|
"TrimSuffix": strings.TrimSuffix,
|
|
|
"Contains": strings.Contains,
|
|
|
"ParamName": ParamName,
|
|
|
- "ParamsContainDT": func(params []XmlColumn) bool {
|
|
|
- isContain := false
|
|
|
+ "ToImports": func(params []XmlColumn) []template.HTML {
|
|
|
+ importToExist := make(map[template.HTML]bool)
|
|
|
+ ret := make([]template.HTML, 0)
|
|
|
for _, v := range params {
|
|
|
- if strings.ToLower(v.Type) == "local_time" || strings.ToLower(v.DbType) == "datetime" {
|
|
|
- isContain = true
|
|
|
- break
|
|
|
+ var importName template.HTML
|
|
|
+ if strings.ToLower(v.Type) == "local_time" {
|
|
|
+ importName = template.HTML(`sysmodel "git.qianqiusoft.com/qianqiusoft/light-apiengine/models"`)
|
|
|
+ } else if strings.ToLower(v.Type) == "datetime" {
|
|
|
+ importName = template.HTML(`"time"`)
|
|
|
+ }
|
|
|
+ if importName != "" && !importToExist[importName] {
|
|
|
+ ret = append(ret, importName)
|
|
|
+ importToExist[importName] = true
|
|
|
}
|
|
|
}
|
|
|
- return isContain
|
|
|
+ return ret
|
|
|
},
|
|
|
"AllParams": func(params []XmlApiParam) string {
|
|
|
str := ""
|