open-im-server/pkg/discoveryregistry/zookeeper/load_balancing.go

package zookeeper

import (
	"sync"

	"google.golang.org/grpc"
)

type RoundRobin struct {
	index int
	lock  sync.Mutex
}

func (r *RoundRobin) getConnBalance(conns []*grpc.ClientConn) (conn *grpc.ClientConn, err error) {
	if len(conns) == 0 {
		return nil, ErrConnIsNil
	}
	r.lock.Lock()
	defer r.lock.Unlock()
	if r.index < len(conns)-1 {
		r.index++
	} else {
		r.index = 0
	}
	return conns[r.index], nil
}