feat: 模板方法抽离 不在模板中做逻辑

server/utils/template_funcs.go

@@ -0,0 +1,109 @@
+package utils
+
+import (
+	"strings"
+	"text/template"
+)
+
+// GetTemplateFuncMap 返回模板函数映射，用于在模板中使用
+func GetTemplateFuncMap() template.FuncMap {
+	return template.FuncMap{
+		"formatGormTag":          FormatGormTag,
+		"formatFieldTag":         FormatFieldTag,
+		"formatSearchCondition":  FormatSearchCondition,
+		"formatBetweenCondition": FormatBetweenCondition,
+		"formatLikeCondition":    FormatLikeCondition,
+		"formatModelField":       FormatModelField,
+		"formatRequestField":     FormatRequestField,
+		"indent":                 Indent,
+	}
+}
+
+// FormatGormTag 格式化GORM标签
+func FormatGormTag(fieldIndexType, primaryKey, defaultValue, columnName, comment, dataTypeLong string) string {
+	var tags []string
+
+	if fieldIndexType != "" {
+		tags = append(tags, fieldIndexType)
+	}
+	if primaryKey == "true" {
+		tags = append(tags, "primarykey")
+	}
+	if defaultValue != "" {
+		tags = append(tags, "default:"+defaultValue)
+	}
+	tags = append(tags, "column:"+columnName)
+	if comment != "" {
+		tags = append(tags, "comment:"+comment)
+	}
+	if dataTypeLong != "" {
+		tags = append(tags, "size:"+dataTypeLong)
+	}
+
+	return strings.Join(tags, ";")
+}
+
+// FormatFieldTag 格式化字段标签
+func FormatFieldTag(fieldJson string, required bool) string {
+	result := `json:"` + fieldJson + `" form:"` + fieldJson + `"`
+	if required {
+		result += ` binding:"required"`
+	}
+	return result
+}
+
+// FormatSearchCondition 格式化搜索条件
+func FormatSearchCondition(columnName, fieldSearchType, fieldName, fieldType string) string {
+	var condition string
+	if fieldType == "string" {
+		condition = `if info.` + fieldName + ` != nil && *info.` + fieldName + ` != "" {`
+	} else {
+		condition = `if info.` + fieldName + ` != nil {`
+	}
+
+	var whereClause string
+	if fieldSearchType == "LIKE" {
+		whereClause = `db = db.Where("` + columnName + ` ` + fieldSearchType + ` ?","%"+*info.` + fieldName + `+"%")`
+	} else {
+		whereClause = `db = db.Where("` + columnName + ` ` + fieldSearchType + ` ?",*info.` + fieldName + `)`
+	}
+
+	return condition + "\n    " + whereClause + "\n}"
+}
+
+// FormatBetweenCondition 格式化BETWEEN条件
+func FormatBetweenCondition(columnName, fieldSearchType, fieldName string) string {
+	return `if info.Start` + fieldName + ` != nil && info.End` + fieldName + ` != nil {` + "\n" +
+		`    db = db.Where("` + columnName + ` ` + fieldSearchType + ` ? AND ? ",info.Start` + fieldName + `,info.End` + fieldName + `)` + "\n" +
+		`}`
+}
+
+// FormatLikeCondition 格式化LIKE条件
+func FormatLikeCondition(columnName, fieldName string) string {
+	return `db = db.Where("` + columnName + ` LIKE ?","%"+ *info.` + fieldName + `+"%")`
+}
+
+// FormatModelField 格式化模型字段
+func FormatModelField(fieldName, fieldType, fieldJson, gormTag string, fieldDesc string) string {
+	result := fieldName + "  " + fieldType + " `" + fieldJson + " " + gormTag + "`"
+	if fieldDesc != "" {
+		result += " //" + fieldDesc
+	}
+	return result
+}
+
+// FormatRequestField 格式化请求字段
+func FormatRequestField(fieldName, fieldType, fieldJson string) string {
+	return fieldName + "  " + fieldType + " `json:\"" + fieldJson + "\" form:\"" + fieldJson + "\" `"
+}
+
+// Indent 缩进文本
+func Indent(text string, indent string) string {
+	lines := strings.Split(text, "\n")
+	for i, line := range lines {
+		if line != "" {
+			lines[i] = indent + line
+		}
+	}
+	return strings.Join(lines, "\n")
+}