Abort health check loop on done context (#555)

Co-authored-by: Stephen Cathcart <stephen.cathcart@neotechnology.com>
neo4j · Dec 14, 2023 · ac6a50b · ac6a50b
1 parent ddb0635
commit ac6a50b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/neo4j/internal/pool/server.go b/neo4j/internal/pool/server.go
@@ -76,14 +76,14 @@ func (s *server) healthCheck(
 	if itime.Since(connection.IdleDate()) > idlenessTimeout {
 		connection.ForceReset(ctx)
 		if !connection.IsAlive() {
-			return false, nil
+			return false, ctx.Err()
 		}
 	}
 	if err := connection.ReAuth(ctx, auth); err != nil {
 		return false, err
 	}
 	if !connection.IsAlive() {
-		return false, nil
+		return false, ctx.Err()
 	}
 	// Update round-robin counter every time we give away a connection and keep track
 	// of our own round-robin index