From 1b0c40a6e2621977cd4476f92fc1563e9ca94322 Mon Sep 17 00:00:00 2001
From: undefined_ss <sununiq@163.com>
Date: Mon, 20 Sep 2021 22:16:30 +0800
Subject: [PATCH] fix ctx override bug

---
 session.go | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/session.go b/session.go
index f5b45a73..f51fd41b 100644
--- a/session.go
+++ b/session.go
@@ -732,6 +732,12 @@ func (session *Session) incrVersionFieldValue(fieldValue *reflect.Value) {
 
 // Context sets the context on this session
 func (session *Session) Context(ctx context.Context) *Session {
+	if session.ctx != nil {
+		ctx = context.WithValue(ctx, log.SessionIDKey, session.ctx.Value(log.SessionIDKey))
+		ctx = context.WithValue(ctx, log.SessionKey, session.ctx.Value(log.SessionKey))
+		ctx = context.WithValue(ctx, log.SessionShowSQLKey, session.ctx.Value(log.SessionShowSQLKey))
+	}
+
 	session.ctx = ctx
 	return session
 }