fix bugs

slackhq · Mar 8, 2024 · d2e999a · d2e999a
1 parent 7e292fa
commit d2e999a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/go/vt/vtgateproxy/gate_balancer.go b/go/vt/vtgateproxy/gate_balancer.go
@@ -110,7 +110,7 @@ func (p *slackAZAffinityPicker) Pick(info balancer.PickInfo) (balancer.PickResul
 	ptr := val.(*uint32)
 	atomic.AddUint32(ptr, 1)
 
-	if len(subConns) >= numConnections {
+	if len(subConns) >= numConnections && numConnections > 0 {
 		fmt.Printf("Limiting to first %v\n", numConnections)
 		return p.pickFromSubconns(subConns[0:numConnections], *ptr)
 	} else {

diff --git a/go/vt/vtgateproxy/vtgateproxy.go b/go/vt/vtgateproxy/vtgateproxy.go
@@ -69,7 +69,7 @@ func (proxy *VTGateProxy) getConnection(ctx context.Context, target string) (*vt
 	proxy.numConnections = targetURL.Query().Get("num_connections")
 	proxy.gateType = targetURL.Host
 
-	fmt.Printf("Getting connection for %v in %v\n", target, proxy.azID)
+	fmt.Printf("Getting connection for %v in %v with %v connections\n", target, proxy.azID, proxy.numConnections)
 
 	// If the connection exists, return it
 	proxy.mu.Lock()