diff options
author | mathieui <mathieui@mathieui.net> | 2021-04-30 19:43:05 +0200 |
---|---|---|
committer | mathieui <mathieui@mathieui.net> | 2021-04-30 19:43:05 +0200 |
commit | 6f4ac7e7cec0c15ba5d6732082b5270b046a377f (patch) | |
tree | 66193a366bc973261eebcc009e4477707daff22c /slixmpp | |
parent | 027545eb710c6a760a4ddc1f7ce6c193a47b1cec (diff) | |
parent | aaab58d229379492e4a2d5c9972db793662a29da (diff) | |
download | slixmpp-6f4ac7e7cec0c15ba5d6732082b5270b046a377f.tar.gz slixmpp-6f4ac7e7cec0c15ba5d6732082b5270b046a377f.tar.bz2 slixmpp-6f4ac7e7cec0c15ba5d6732082b5270b046a377f.tar.xz slixmpp-6f4ac7e7cec0c15ba5d6732082b5270b046a377f.zip |
Merge branch 'fix-slow-tasks' into 'master'
xmlstream: Fix slow tasks
See merge request poezio/slixmpp!162
Diffstat (limited to 'slixmpp')
-rw-r--r-- | slixmpp/xmlstream/xmlstream.py | 12 |
1 files changed, 8 insertions, 4 deletions
diff --git a/slixmpp/xmlstream/xmlstream.py b/slixmpp/xmlstream/xmlstream.py index 7a94bf50..ab9b781d 100644 --- a/slixmpp/xmlstream/xmlstream.py +++ b/slixmpp/xmlstream/xmlstream.py @@ -1053,11 +1053,13 @@ class XMLStream(asyncio.BaseProtocol): """ data = await task self.__slow_tasks.remove(task) - for filter in self.__filters['out']: + if data is None: + return + for filter in self.__filters['out'][:]: if filter in already_used: continue if iscoroutinefunction(filter): - data = await task + data = await filter(data) else: data = filter(data) if data is None: @@ -1093,7 +1095,7 @@ class XMLStream(asyncio.BaseProtocol): timeout=1, ) if pending: - self.slow_tasks.append(task) + self.__slow_tasks.append(task) asyncio.ensure_future( self._continue_slow_send( task, @@ -1101,7 +1103,9 @@ class XMLStream(asyncio.BaseProtocol): ), loop=self.loop, ) - raise Exception("Slow coro, rescheduling") + raise ContinueQueue( + "Slow coroutine, rescheduling filters" + ) data = task.result() else: data = filter(data) |