update idle count on changing pool capacity after the capacity is app…

…lied to the pool

Signed-off-by: Harshit Gangal <[email protected]>

go/pools/smartconnpool/pool.go

@@ -671,7 +671,9 @@ func (pool *ConnPool[C]) setCapacity(ctx context.Context, newcap int64) error {
 	if oldcap == newcap {
 		return nil
 	}
-	pool.setIdleCount()
+	// update the idle count to match the new capacity if necessary
+	// wait for connections to be returned to the pool if we're reducing the capacity.
+	defer pool.setIdleCount()
 
 	const delay = 10 * time.Millisecond