diff --git a/engine_group.go b/engine_group.go index 6796075e..6ef68f9c 100644 --- a/engine_group.go +++ b/engine_group.go @@ -5,6 +5,7 @@ package xorm import ( + "context" "time" "github.com/go-xorm/core" @@ -74,6 +75,13 @@ func (eg *EngineGroup) Close() error { return nil } +// Context returned a group session +func (eg *EngineGroup) Context(ctx context.Context) *Session { + sess := eg.NewSession() + sess.isAutoClose = true + return sess.Context(ctx) +} + // NewSession returned a group session func (eg *EngineGroup) NewSession() *Session { sess := eg.Engine.NewSession()