feat: let model customizable (#1738)

tools/goctl/model/sql/template/model.go

@@ -6,6 +6,33 @@ import (
 	"github.com/zeromicro/go-zero/tools/goctl/util"
 )
 
+// ModelCustom defines a template for extension
+const ModelCustom = `package {{.pkg}}
+
+import "github.com/zeromicro/go-zero/core/stores/sqlx"
+
+var _ {{.upperStartCamelObject}}Model = (*custom{{.upperStartCamelObject}}Model)(nil)
+
+type (
+	// {{.upperStartCamelObject}}Model is an interface to be customized, add more methods here,
+	// and implement the added methods in custom{{.upperStartCamelObject}}Model.
+	{{.upperStartCamelObject}}Model interface {
+		{{.lowerStartCamelObject}}Model
+	}
+
+	custom{{.upperStartCamelObject}}Model struct {
+		*default{{.upperStartCamelObject}}Model
+	}
+)
+
+// New{{.upperStartCamelObject}}Model returns a model for the database table.
+func New{{.upperStartCamelObject}}Model(conn sqlx.SqlConn{{if .withCache}}, c cache.CacheConf{{end}}) {{.upperStartCamelObject}}Model {
+	return &custom{{.upperStartCamelObject}}Model{
+		default{{.upperStartCamelObject}}Model: new{{.upperStartCamelObject}}Model(conn{{if .withCache}}, c{{end}}),
+	}
+}
+`
+
 // ModelGen defines a template for model
 var ModelGen = fmt.Sprintf(`%s
 
@@ -19,11 +46,5 @@ package {{.pkg}}
 {{.update}}
 {{.delete}}
 {{.extraMethod}}
+{{.tableName}}
 `, util.DoNotEditHead)
-
-// ModelCustom defines a template for extension
-var ModelCustom = fmt.Sprintf(`package {{.pkg}}
-type {{.upperStartCamelObject}}Model interface {
-	{{.lowerStartCamelObject}}Model
-}
-`)