@@ -1955,8 +1955,8 @@ def __hash__(self):
19551955class LoadBalancingStrategy (Enum ):
19561956 ROUND_ROBIN = "round_robin"
19571957 ROUND_ROBIN_REPLICAS = "round_robin_replicas"
1958- RANDOM_REPLICA = "random_replica"
19591958 RANDOM = "random"
1959+ RANDOM_REPLICA = "random_replica"
19601960
19611961
19621962class LoadBalancer :
@@ -1975,15 +1975,15 @@ def get_server_index(
19751975 list_size : int ,
19761976 load_balancing_strategy : LoadBalancingStrategy = LoadBalancingStrategy .ROUND_ROBIN ,
19771977 ) -> int :
1978- if (
1979- load_balancing_strategy == LoadBalancingStrategy .RANDOM_REPLICA
1980- or load_balancing_strategy == LoadBalancingStrategy .RANDOM
1981- ):
1982- return self ._get_random_replica_index (
1978+ if load_balancing_strategy == LoadBalancingStrategy .RANDOM_REPLICA :
1979+ return self ._get_random_server_index (
1980+ list_size ,
1981+ replicas_only = True ,
1982+ )
1983+ elif load_balancing_strategy == LoadBalancingStrategy .RANDOM :
1984+ return self ._get_random_server_index (
19831985 list_size ,
1984- replicas_only = (
1985- load_balancing_strategy == LoadBalancingStrategy .RANDOM_REPLICA
1986- ),
1986+ replicas_only = False ,
19871987 )
19881988 else :
19891989 return self ._get_round_robin_index (
@@ -1996,7 +1996,7 @@ def reset(self) -> None:
19961996 with self ._lock :
19971997 self .primary_to_idx .clear ()
19981998
1999- def _get_random_replica_index (self , list_size : int , replicas_only : bool ) -> int :
1999+ def _get_random_server_index (self , list_size : int , replicas_only : bool ) -> int :
20002000 return random .randint (1 if replicas_only else 0 , list_size - 1 )
20012001
20022002 def _get_round_robin_index (
0 commit comments