improve template for command gen dao

gogf · Jun 3, 2021 · 4c84e08 · 4c84e08
1 parent 4a7d304
commit 4c84e08
Showing 1 changed file with 4 additions and 9 deletions.
diff --git a/commands/gen/gen_dao_template_dao.go b/commands/gen/gen_dao_template_dao.go
@@ -15,7 +15,7 @@ import (
 // and custom defined data operations functions management. You can define
 // methods on it to extend its functionality as you wish.
 type {TplTableNameCamelLowerCase}Dao struct {
-	internal.{TplTableNameCamelCase}Dao
+	*internal.{TplTableNameCamelCase}Dao
 }
 
 var (
@@ -25,7 +25,7 @@ var (
 
 func init() {
 	{TplTableNameCamelCase} = {TplTableNameCamelLowerCase}Dao{
-		internal.{TplTableNameCamelCase},
+		internal.New{TplTableNameCamelCase}Dao(),
 	}
 }
 
@@ -60,13 +60,8 @@ type {TplTableNameCamelLowerCase}Columns struct {
 	{TplColumnDefine}
 }
 
-var (
-	// {TplTableNameCamelCase} is globally public accessible object for table {TplTableName} operations.
-	{TplTableNameCamelCase} {TplTableNameCamelCase}Dao
-)
-
-func init() {
-	{TplTableNameCamelCase} = {TplTableNameCamelCase}Dao{
+func New{TplTableNameCamelCase}Dao() *{TplTableNameCamelCase}Dao {
+	return &{TplTableNameCamelCase}Dao{
 		M:     g.DB("{TplGroupName}").Model("{TplTableName}").Safe(),
 		DB:    g.DB("{TplGroupName}"),
 		Table: "{TplTableName}",