p

Author: dkropachev

cassandra/policies.py

@@ -872,7 +872,7 @@ def _add_jitter(self, value):
         return min(max(self.base_delay, delay), self.max_delay)
 
 
-class _ShardReconnectionScheduler(ABC):
+class ShardReconnectionScheduler(ABC):
     @abstractmethod
     def schedule(
             self,
@@ -892,24 +892,24 @@ class ShardReconnectionPolicy(ABC):
     """
 
     @abstractmethod
-    def new_scheduler(self, session: Session) -> _ShardReconnectionScheduler:
+    def new_scheduler(self, session: Session) -> ShardReconnectionScheduler:
         raise NotImplementedError()
 
 
 class NoDelayShardReconnectionPolicy(ShardReconnectionPolicy):
     """
-    A shard reconnection policy with no delay between attempts.
+    A shard reconnection policy with no delay between attempts and no concurrency restrictions.
     Ensures at most one pending reconnection per (host, shard) pair — any additional
      reconnection attempts for the same (host, shard) are silently ignored.
 
     On `new_scheduler` instantiate a scheduler that behaves according to the policy
     """
 
-    def new_scheduler(self, session: Session) -> _ShardReconnectionScheduler:
+    def new_scheduler(self, session: Session) -> ShardReconnectionScheduler:
         return _NoDelayShardReconnectionScheduler(session)
 
 
-class _NoDelayShardReconnectionScheduler(_ShardReconnectionScheduler):
+class _NoDelayShardReconnectionScheduler(ShardReconnectionScheduler):
     def __init__(self, session: Session):
         self.session: Session = weakref.proxy(session)
         self.already_scheduled: dict[str, bool] = {}
@@ -970,7 +970,7 @@ def __init__(
         self.shard_reconnection_scope = shard_reconnection_scope
         self.reconnection_policy = reconnection_policy
 
-    def new_scheduler(self, session: Session) -> _ShardReconnectionScheduler:
+    def new_scheduler(self, session: Session) -> ShardReconnectionScheduler:
         return _NoConcurrentShardReconnectionScheduler(session, self.shard_reconnection_scope, self.reconnection_policy)
 
 
@@ -1035,7 +1035,7 @@ def add(self, method, *args, **kwargs):
                 self._schedule()
 
 
-class _NoConcurrentShardReconnectionScheduler(_ShardReconnectionScheduler):
+class _NoConcurrentShardReconnectionScheduler(ShardReconnectionScheduler):
     def __init__(
             self,
             session: Session,