Fix "invalid argument to Intn"

Signed-off-by: Bin Shi <[email protected]>
tikv · Apr 19, 2023 · a5c61ff · a5c61ff
1 parent fb0cf06
commit a5c61ff
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/client/tso_service_discovery.go b/client/tso_service_discovery.go
@@ -516,8 +516,9 @@ func (c *tsoServiceDiscovery) getTSOServer() (string, error) {
 
 	// Pick a TSO server in a round-robin way.
 	len := len(c.tsoServerDiscovery.addrs)
+	c.tsoServerDiscovery.selectIdx++
 	tsoServerAddr := c.tsoServerDiscovery.addrs[rand.Intn(c.tsoServerDiscovery.selectIdx)]
-	c.tsoServerDiscovery.selectIdx = (c.tsoServerDiscovery.selectIdx + 1) % len
+	c.tsoServerDiscovery.selectIdx %= len
 	if _, err := c.GetOrCreateGRPCConn(tsoServerAddr); err != nil {
 		log.Warn("[tso] failed to connect the tso server",
 			zap.String("tso-server", tsoServerAddr), errs.ZapError(err))