WheelTimer: Don't scan list, use index.

This commit is contained in:
Erik Johnston 2016-02-18 16:33:07 +00:00
parent 48b652bcbe
commit b8cdec92c7

View File

@ -46,11 +46,14 @@ class WheelTimer(object):
then (int): When to return the object strictly after. then (int): When to return the object strictly after.
""" """
then_key = int(then / self.bucket_size) + 1 then_key = int(then / self.bucket_size) + 1
for entry in self.entries:
# Add to first bucket we find. This should gracefully handle inserts if self.entries:
# for times in the past. min_key = self.entries[0].end_key
if entry.end_key >= then_key: max_key = self.entries[-1].end_key
entry.queue.append(obj)
if then_key <= max_key:
# The max here is to protect against inserts for times in the past
self.entries[max(min_key, then_key) - min_key].queue.append(obj)
return return
next_key = int(now / self.bucket_size) + 1 next_key = int(now / self.bucket_size) + 1