fix(spanner): add safecheck to avoid deadlock when creating multiplex…

… session
googleapis · Nov 14, 2024 · bbd8e6c · bbd8e6c
1 parent 4250788
commit bbd8e6c
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/spanner/session.go b/spanner/session.go
@@ -906,7 +906,10 @@ func (p *sessionPool) createMultiplexedSession() {
 				p.mu.Lock()
 				p.multiplexedSessionCreationError = err
 				p.mu.Unlock()
-				p.mayGetMultiplexedSession <- true
+				select {
+				case p.mayGetMultiplexedSession <- true:
+				case <-c.ctx.Done():
+				}
 				continue
 			}
 			p.sc.executeCreateMultiplexedSession(c.ctx, client, p.sc.md, p)