Support get dataSourceName on ContextHook for monitor which DB executed SQL #1740
|
@ -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
|
||||
|
|
Loading…
Reference in New Issue
Block a user