trezor/loop: optimization

Re-cycle queue of paused tasks by clear()ing it rather than pop()ing it out of the dict.
This commit is contained in:
Jan Pochyla 2017-10-02 11:42:56 +02:00
parent 7d31bdcf78
commit cfa1705a88
1 changed files with 6 additions and 4 deletions

View File

@ -53,7 +53,7 @@ def unschedule(task):
def pause(task, iface): def pause(task, iface):
tasks = _paused.get(iface, None) tasks = _paused.get(iface)
if tasks is None: if tasks is None:
tasks = _paused[iface] = [] tasks = _paused[iface] = []
tasks.append(task) tasks.append(task)
@ -94,9 +94,11 @@ def run():
if io.poll(_paused, msg_entry, delay): if io.poll(_paused, msg_entry, delay):
# message received, run tasks paused on the interface # message received, run tasks paused on the interface
msg_tasks = _paused.pop(msg_entry[0], ()) msg_tasks = _paused.get(msg_entry[0])
for task in msg_tasks: if msg_tasks is not None:
_step(task, msg_entry[1]) for task in msg_tasks:
_step(task, msg_entry[1])
msg_tasks.clear()
else: else:
# timeout occurred, run the first scheduled task # timeout occurred, run the first scheduled task
if _queue: if _queue: