feat: session method for relation field (go-gorm#779)

m430 · Apr 12, 2023 · e9e31a7 · e9e31a7
1 parent 6df282a
commit e9e31a7
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/internal/template/struct.go b/internal/template/struct.go
@@ -237,6 +237,11 @@ func (a {{$.QueryStructName}}{{$relationship}}{{$relation.Name}}) WithContext(ct
 	return &a
 }
 
+func (a {{$.QueryStructName}}{{$relationship}}{{$relation.Name}}) Session(session *gorm.Session) *{{$.QueryStructName}}{{$relationship}}{{$relation.Name}} {
+	a.db = a.db.Session(session)
+	return &a
+}
+
 func (a {{$.QueryStructName}}{{$relationship}}{{$relation.Name}}) Model(m *{{$.StructInfo.Package}}.{{$.StructInfo.Type}}) *{{$.QueryStructName}}{{$relationship}}{{$relation.Name}}Tx {
 	return &{{$.QueryStructName}}{{$relationship}}{{$relation.Name}}Tx{a.db.Model(m).Association(a.Name())}
 }