From 47fbd4cead2e881b0250dd5f978caf64c6a5952c Mon Sep 17 00:00:00 2001 From: Florent Le Coz Date: Sat, 3 Jan 2015 05:55:17 +0100 Subject: Delay the handling of stanza for when the process is not busy We use some dirty monkey-patching to add a idle_call() function to the asyncio module. We then use that method to handle each received stanza only when the event loop is not busy with some other IO (mainly, the standard input) --- slixmpp/xmlstream/asyncio.py | 32 ++++++++++++++++++++++++++++++++ 1 file changed, 32 insertions(+) create mode 100644 slixmpp/xmlstream/asyncio.py (limited to 'slixmpp/xmlstream/asyncio.py') diff --git a/slixmpp/xmlstream/asyncio.py b/slixmpp/xmlstream/asyncio.py new file mode 100644 index 00000000..52610d01 --- /dev/null +++ b/slixmpp/xmlstream/asyncio.py @@ -0,0 +1,32 @@ +""" +A module that monkey patches the standard asyncio module to add an +idle_call() method to the main loop. This method is used to execute a +callback whenever the loop is not busy handling anything else. This means +that it is a callback with lower priority than IO, timer, or even +call_soon() ones. These callback are called only once each. +""" + +import asyncio +from asyncio import tasks, events + +def idle_call(self, callback): + if tasks.iscoroutinefunction(callback): + raise TypeError("coroutines cannot be used with idle_call()") + handle = events.Handle(callback, [], self) + self._idle.append(handle) + +def my_run_once(self): + if self._idle: + self._ready.append(events.Handle(lambda: None, (), self)) + real_run_once(self) + if self._idle: + handle = self._idle.pop(0) + handle._run() + +cls = asyncio.get_event_loop().__class__ + +cls._idle = [] +cls.idle_call = idle_call +real_run_once = cls._run_once +cls._run_once = my_run_once + -- cgit v1.2.3