Use timeout constants instead of magic numbers in scheduler and event loop

Set default wait timeout as max() of previous values
This commit is contained in:
Anton Ryzhov 2013-06-20 13:07:57 +04:00
parent 7430a8ca40
commit 805f1c0e39
2 changed files with 16 additions and 8 deletions

View File

@ -20,6 +20,11 @@ import itertools
from sleekxmpp.util import Queue, QueueEmpty from sleekxmpp.util import Queue, QueueEmpty
#: The time in seconds to wait for events from the event queue, and also the
#: time between checks for the process stop signal.
WAIT_TIMEOUT = 1.0
log = logging.getLogger(__name__) log = logging.getLogger(__name__)
@ -120,6 +125,10 @@ class Scheduler(object):
#: Lock for accessing the task queue. #: Lock for accessing the task queue.
self.schedule_lock = threading.RLock() self.schedule_lock = threading.RLock()
#: The time in seconds to wait for events from the event queue,
#: and also the time between checks for the process stop signal.
self.wait_timeout = WAIT_TIMEOUT
def process(self, threaded=True, daemon=False): def process(self, threaded=True, daemon=False):
"""Begin accepting and processing scheduled tasks. """Begin accepting and processing scheduled tasks.
@ -143,7 +152,7 @@ class Scheduler(object):
if self.schedule: if self.schedule:
wait = self.schedule[0].next - time.time() wait = self.schedule[0].next - time.time()
else: else:
wait = 0.1 wait = self.wait_timeout
try: try:
if wait <= 0.0: if wait <= 0.0:
newtask = self.addq.get(False) newtask = self.addq.get(False)
@ -156,8 +165,8 @@ class Scheduler(object):
not self.stop.is_set() and \ not self.stop.is_set() and \
newtask is None and \ newtask is None and \
elapsed < wait: elapsed < wait:
newtask = self.addq.get(True, 0.1) newtask = self.addq.get(True, self.wait_timeout)
elapsed += 0.1 elapsed += self.wait_timeout
except QueueEmpty: # Time to run some tasks, and no new tasks to add. except QueueEmpty: # Time to run some tasks, and no new tasks to add.
self.schedule_lock.acquire() self.schedule_lock.acquire()
#select only those tasks which are to be executed now #select only those tasks which are to be executed now

View File

@ -49,7 +49,7 @@ RESPONSE_TIMEOUT = 30
#: The time in seconds to wait for events from the event queue, and also the #: The time in seconds to wait for events from the event queue, and also the
#: time between checks for the process stop signal. #: time between checks for the process stop signal.
WAIT_TIMEOUT = 0.1 WAIT_TIMEOUT = 1.0
#: The number of threads to use to handle XML stream events. This is not the #: The number of threads to use to handle XML stream events. This is not the
#: same as the number of custom event handling threads. #: same as the number of custom event handling threads.
@ -1629,8 +1629,7 @@ class XMLStream(object):
try: try:
while not self.stop.is_set(): while not self.stop.is_set():
try: try:
wait = self.wait_timeout event = self.event_queue.get(True, timeout=self.wait_timeout)
event = self.event_queue.get(True, timeout=wait)
except QueueEmpty: except QueueEmpty:
event = None event = None
if event is None: if event is None:
@ -1693,13 +1692,13 @@ class XMLStream(object):
while not self.stop.is_set(): while not self.stop.is_set():
while not self.stop.is_set() and \ while not self.stop.is_set() and \
not self.session_started_event.is_set(): not self.session_started_event.is_set():
self.session_started_event.wait(timeout=0.1) self.session_started_event.wait(timeout=0.1) # Wait for session start
if self.__failed_send_stanza is not None: if self.__failed_send_stanza is not None:
data = self.__failed_send_stanza data = self.__failed_send_stanza
self.__failed_send_stanza = None self.__failed_send_stanza = None
else: else:
try: try:
data = self.send_queue.get(True, 1) data = self.send_queue.get(True, timeout=self.wait_timeout) # Wait for data to send
except QueueEmpty: except QueueEmpty:
continue continue
log.debug("SEND: %s", data) log.debug("SEND: %s", data)