Skip to content

Commit d27a332

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

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
@@ -4393,7 +4393,10 @@ def shutdown(self):
43934393
self.join()
43944394

43954395
def schedule(self, delay, fn, *args, **kwargs):
4396-
self._insert_task(delay, (fn, args, tuple(kwargs.items())))
4396+
if delay:
4397+
self._insert_task(delay, (fn, args, tuple(kwargs.items())))
4398+
else:
4399+
self._executor.submit(fn, *args, **kwargs)
43974400

43984401
def schedule_unique(self, delay, fn, *args, **kwargs):
43994402
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
@@ -134,8 +134,8 @@ def test_event_delay_timing(self, *_):
134134
PYTHON-473
135135
"""
136136
sched = _Scheduler(None)
137-
sched.schedule(0, lambda: None)
138-
sched.schedule(0, lambda: None) # pre-473: "TypeError: unorderable types: function() < function()"t
137+
sched.schedule(1, lambda: None)
138+
sched.schedule(1, lambda: None) # pre-473: "TypeError: unorderable types: function() < function()"t
139139

140140

141141
class SessionTest(unittest.TestCase):

0 commit comments

Comments
 (0)