Skip to content

Commit 28785ec

Browse files
committed
Make _Scheduler submit right away if delay is 0
This is required by upcoming LimitedConcurrencyShardConnectionBackoffPolicy.
1 parent b976e56 commit 28785ec

File tree

2 files changed

+6
-3
lines changed

2 files changed

+6
-3
lines changed

cassandra/cluster.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4456,7 +4456,10 @@ def shutdown(self):
44564456
self.join()
44574457

44584458
def schedule(self, delay, fn, *args, **kwargs):
4459-
self._insert_task(delay, (fn, args, tuple(kwargs.items())))
4459+
if delay:
4460+
self._insert_task(delay, (fn, args, tuple(kwargs.items())))
4461+
else:
4462+
self._executor.submit(fn, *args, **kwargs)
44604463

44614464
def schedule_unique(self, delay, fn, *args, **kwargs):
44624465
task = (fn, args, tuple(kwargs.items()))

tests/unit/test_cluster.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -148,8 +148,8 @@ def test_event_delay_timing(self, *_):
148148
PYTHON-473
149149
"""
150150
sched = _Scheduler(None)
151-
sched.schedule(0, lambda: None)
152-
sched.schedule(0, lambda: None) # pre-473: "TypeError: unorderable types: function() < function()"t
151+
sched.schedule(1, lambda: None)
152+
sched.schedule(1, lambda: None) # pre-473: "TypeError: unorderable types: function() < function()"t
153153

154154

155155
class SessionTest(unittest.TestCase):

0 commit comments

Comments
 (0)