签入脚本

templates/dao_internal.tpl

@@ -0,0 +1,83 @@
+// ==========================================================================
+// Code generated and maintained by GoFrame CLI tool. DO NOT EDIT.
+// ==========================================================================
+
+package internal
+
+import (
+	"context"
+
+	"github.com/gogf/gf/v2/database/gdb"
+	"github.com/gogf/gf/v2/frame/g"
+)
+
+// {{.DaoName}}Dao is the data access object for the table {{.TableNameFull}}.
+type {{.DaoName}}Dao struct {
+	table    string               // table is the underlying table name of the DAO.
+	group    string               // group is the database configuration group name of the current DAO.
+	columns  {{.DaoName}}Columns  // columns contains all the column names of Table for convenient usage.
+	handlers []gdb.ModelHandler   // handlers for customized model modification.
+}
+
+// {{.DaoName}}Columns defines and stores column names for the table {{.TableNameFull}}.
+type {{.DaoName}}Columns struct {
+{{- range .Fields}}
+	{{.CamelExport}} string // {{.Comment}}
+{{- end}}
+}
+
+// {{.ModuleCamel}}Columns holds the columns for the table {{.TableNameFull}}.
+var {{.ModuleCamel}}Columns = {{.DaoName}}Columns{
+{{- range .Fields}}
+	{{.CamelExport}}: "{{.Name}}",
+{{- end}}
+}
+
+// New{{.DaoName}}Dao creates and returns a new DAO object for table data access.
+func New{{.DaoName}}Dao(handlers ...gdb.ModelHandler) *{{.DaoName}}Dao {
+	return &{{.DaoName}}Dao{
+		group:    "default",
+		table:    "{{.TableNameFull}}",
+		columns:  {{.ModuleCamel}}Columns,
+		handlers: handlers,
+	}
+}
+
+// DB retrieves and returns the underlying raw database management object of the current DAO.
+func (dao *{{.DaoName}}Dao) DB() gdb.DB {
+	return g.DB(dao.group)
+}
+
+// Table returns the table name of the current DAO.
+func (dao *{{.DaoName}}Dao) Table() string {
+	return dao.table
+}
+
+// Columns returns all column names of the current DAO.
+func (dao *{{.DaoName}}Dao) Columns() {{.DaoName}}Columns {
+	return dao.columns
+}
+
+// Group returns the database configuration group name of the current DAO.
+func (dao *{{.DaoName}}Dao) Group() string {
+	return dao.group
+}
+
+// Ctx creates and returns a Model for the current DAO. It automatically sets the context for the current operation.
+func (dao *{{.DaoName}}Dao) Ctx(ctx context.Context) *gdb.Model {
+	model := dao.DB().Model(dao.table)
+	for _, handler := range dao.handlers {
+		model = handler(model)
+	}
+	return model.Safe().Ctx(ctx)
+}
+
+// Transaction wraps the transaction logic using function f.
+// It rolls back the transaction and returns the error if function f returns a non-nil error.
+// It commits the transaction and returns nil if function f returns nil.
+//
+// Note: Do not commit or roll back the transaction in function f,
+// as it is automatically handled by this function.
+func (dao *{{.DaoName}}Dao) Transaction(ctx context.Context, f func(ctx context.Context, tx gdb.TX) error) (err error) {
+	return dao.Ctx(ctx).Transaction(ctx, f)
+}