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,
}
if engine.logSessionID {
ctx = context.WithValue(engine.defaultContext, sessionID, session)
session.ctx = context.WithValue(session.ctx, sessionID, session)
}
return session
}
@ -171,6 +171,10 @@ func (session *Session) db() *core.DB {
return session.engine.db
}
func (session *Session) Engine() *Engine {
return session.engine
}
func (session *Session) getQueryer() core.Queryer {
if session.tx != nil {
return session.tx