Skip to content

Commit 99a3ae3

Browse files
committed
address style comments
1 parent de7232f commit 99a3ae3

File tree

1 file changed

+10
-10
lines changed

1 file changed

+10
-10
lines changed

redis/cluster.py

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -1955,8 +1955,8 @@ def __hash__(self):
19551955
class 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

19621962
class 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

Comments
 (0)