feat(gen): support custom templates for model generation

generator.go

@@ -235,6 +235,12 @@ func (g *Generator) ApplyInterface(fc interface{}, models ...interface{}) {
 	g.apply(fc, structs)
 }
 
+func (g *Generator) ApplyCustomTemplateForModel(templateText string) {
+	for _, modelMeta := range g.models {
+		modelMeta.CustomTemplates = append(modelMeta.CustomTemplates, templateText)
+	}
+}
+
 func (g *Generator) apply(fc interface{}, structs []*generate.QueryStructMeta) {
 	interfacePaths, err := parser.GetInterfacePath(fc)
 	if err != nil {
@@ -522,6 +528,14 @@ func (g *Generator) generateModelFile() error {
 				return
 			}
 
+			for _, customTemplate := range data.CustomTemplates {
+				err = render(customTemplate, &buf, data)
+				if err != nil {
+					errChan <- err
+					return
+				}
+			}
+
 			for _, method := range data.ModelMethods {
 				err = render(tmpl.ModelMethod, &buf, method)
 				if err != nil {

internal/generate/query.go

@@ -38,6 +38,7 @@ type QueryStructMeta struct {
 	Source          model.SourceCode
 	ImportPkgPaths  []string
 	ModelMethods    []*parser.Method // user custom method bind to db base struct
+	CustomTemplates []string         // custom template text
 
 	interfaceMode bool