jshiffer/slixmpp
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Try/except around outbound stanza processing

Browse Source 
to avoid killing the send loop when a filter has an error
This commit is contained in:
mathieui 2019-08-22 16:13:24 +02:00
parent a32794ec35
commit 27d3ae958b
No known key found for this signature in database
GPG Key ID: C59F84CEEFD616E3
1 changed files with 26 additions and 23 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

49
slixmpp/xmlstream/xmlstream.py
View File

@ -902,30 +902,33 @@ class XMLStream(asyncio.BaseProtocol):
""" """
while True: while True:
(data, use_filters) = await self.waiting_queue.get() (data, use_filters) = await self.waiting_queue.get()
if isinstance(data, ElementBase): try:
if use_filters: if isinstance(data, ElementBase):
for filter in self.__filters['out']: if use_filters:
if iscoroutinefunction(filter): for filter in self.__filters['out']:
data = await filter(data) if iscoroutinefunction(filter):
else: data = await filter(data)
data = filter(data) else:
if data is None: data = filter(data)
return if data is None:
return
if isinstance(data, ElementBase): if isinstance(data, ElementBase):
if use_filters: if use_filters:
for filter in self.__filters['out_sync']: for filter in self.__filters['out_sync']:
if iscoroutinefunction(filter): if iscoroutinefunction(filter):
data = await filter(data) data = await filter(data)
else: else:
data = filter(data) data = filter(data)
if data is None: if data is None:
return return
str_data = tostring(data.xml, xmlns=self.default_ns, str_data = tostring(data.xml, xmlns=self.default_ns,
stream=self, top_level=True) stream=self, top_level=True)
self.send_raw(str_data) self.send_raw(str_data)
else: else:
self.send_raw(data) self.send_raw(data)
except:
log.error('Could not send stanza %s', data, exc_info=True)
self.waiting_queue.task_done() self.waiting_queue.task_done()
def send(self, data, use_filters=True): def send(self, data, use_filters=True):