session add Engine func

This commit is contained in:
yong.an 2020-07-09 17:40:37 +08:00
parent da55c35b25
commit 3a48920a98
1 changed files with 5 additions and 1 deletions

View File

@ -139,7 +139,7 @@ func newSession(engine *Engine) *Session {
sessionType: engineSession, sessionType: engineSession,
} }
if engine.logSessionID { if engine.logSessionID {
ctx = context.WithValue(engine.defaultContext, sessionID, session) session.ctx = context.WithValue(session.ctx, sessionID, session)
} }
return session return session
} }
@ -171,6 +171,10 @@ func (session *Session) db() *core.DB {
return session.engine.db return session.engine.db
} }
func (session *Session) Engine() *Engine {
return session.engine
}
func (session *Session) getQueryer() core.Queryer { func (session *Session) getQueryer() core.Queryer {
if session.tx != nil { if session.tx != nil {
return session.tx return session.tx