feat: generate with Session method

go-gorm · Oct 18, 2022 · 3c1f921 · 3c1f921
1 parent 6cbdcbe
commit 3c1f921
Show file tree

Hide file tree

Showing 31 changed files with 3,313 additions and 3 deletions.
diff --git a/internal/template/method.go b/internal/template/method.go
@@ -43,6 +43,10 @@ func ({{.S}} {{.QueryStructName}}Do) WriteDB() {{.ReturnObject}} {
 	return {{.S}}.Clauses(dbresolver.Write)
 }
 
+func ({{.S}} {{.QueryStructName}}Do) Session(config *gorm.Session) {{.ReturnObject}} {
+	return {{.S}}.withDO({{.S}}.DO.Session(config))
+}
+
 func ({{.S}} {{.QueryStructName}}Do) Clauses(conds ...clause.Expression) {{.ReturnObject}} {
 	return {{.S}}.withDO({{.S}}.DO.Clauses(conds...))
 }

diff --git a/internal/template/struct.go b/internal/template/struct.go
@@ -7,7 +7,7 @@ const (
 		{{.QueryStructName}}Do
 		` + fields + `
 	}
-	` + tableMethod + asMethond + updateFieldMethod + getFieldMethod + fillFieldMapMethod + cloneMethod+ replaceMethod + relationship + defineMethodStruct
+	` + tableMethod + asMethond + updateFieldMethod + getFieldMethod + fillFieldMapMethod + cloneMethod + replaceMethod + relationship + defineMethodStruct
 
 	// TableQueryStructWithContext table query struct with context
 	TableQueryStructWithContext = createMethod + `
@@ -151,6 +151,7 @@ type I{{.ModelStructName}}Do interface {
 	ReadDB() I{{.ModelStructName}}Do
 	WriteDB() I{{.ModelStructName}}Do
 	As(alias string) gen.Dao
+	Session(config *gorm.Session) I{{.ModelStructName}}Do
 	Columns(cols ...field.Expr) gen.Columns
 	Clauses(conds ...clause.Expression) I{{.ModelStructName}}Do
 	Not(conds ...gen.Condition) I{{.ModelStructName}}Do

diff --git a/tests/.expect/dal_1/query/banks.gen.go b/tests/.expect/dal_1/query/banks.gen.go
diff --git a/tests/.expect/dal_1/query/credit_cards.gen.go b/tests/.expect/dal_1/query/credit_cards.gen.go
diff --git a/tests/.expect/dal_1/query/customers.gen.go b/tests/.expect/dal_1/query/customers.gen.go
diff --git a/tests/.expect/dal_1/query/people.gen.go b/tests/.expect/dal_1/query/people.gen.go
diff --git a/tests/.expect/dal_1/query/users.gen.go b/tests/.expect/dal_1/query/users.gen.go
diff --git a/tests/.expect/dal_2/query/banks.gen.go b/tests/.expect/dal_2/query/banks.gen.go
diff --git a/tests/.expect/dal_2/query/credit_cards.gen.go b/tests/.expect/dal_2/query/credit_cards.gen.go
diff --git a/tests/.expect/dal_2/query/customers.gen.go b/tests/.expect/dal_2/query/customers.gen.go
diff --git a/tests/.expect/dal_2/query/people.gen.go b/tests/.expect/dal_2/query/people.gen.go
diff --git a/tests/.expect/dal_2/query/users.gen.go b/tests/.expect/dal_2/query/users.gen.go
diff --git a/tests/.expect/dal_3/model/banks.gen.go b/tests/.expect/dal_3/model/banks.gen.go
diff --git a/tests/.expect/dal_3/model/credit_cards.gen.go b/tests/.expect/dal_3/model/credit_cards.gen.go
diff --git a/tests/.expect/dal_3/model/customers.gen.go b/tests/.expect/dal_3/model/customers.gen.go
diff --git a/tests/.expect/dal_3/model/people.gen.go b/tests/.expect/dal_3/model/people.gen.go
diff --git a/tests/.expect/dal_3/model/users.gen.go b/tests/.expect/dal_3/model/users.gen.go