asyncio: Implement subclass implementing filedes watching interface.

Paul Sokolovsky · Paul Sokolovsky · commit 78fef25eeaa3 · 2014-04-24T01:07:22.000+03:00
diff --git a/asyncio/asyncio.py b/asyncio/asyncio.py
@@ -6,10 +6,6 @@ def coroutine(f):
     return f
 
 
-def get_event_loop():
-    return EventLoop()
-
-
 class EventLoop:
 
     def __init__(self):
@@ -38,15 +34,18 @@ def call_at(self, time, callback, *args):
 #            c = self.q.pop(0)
 #            c[0](*c[1])
 
+    def wait(self, delay):
+#        print("Sleeping for:", delay)
+        time.sleep(delay)
+
     def run_forever(self):
         while self.q:
 #            t, cnt, cb, args = self.q.pop(0)
             t, cnt, cb, args = heapq.heappop(self.q)
             tnow = self.time()
             delay = t - tnow
             if delay > 0:
-#                print("Sleeping for:", delay)
-                time.sleep(delay)
+                self.wait(delay)
             delay = 0
             try:
                 ret = next(cb)
@@ -74,6 +73,26 @@ def run_until_complete(self, coro):
     def close(self):
         pass
 
+import select
+
+class EpollEventLoop(EventLoop):
+
+    def __init__(self):
+        EventLoop.__init__(self)
+        self.poller = select.epoll(1)
+
+    def add_reader(self, fd, cb, *args):
+        self.poller.register(fd, select.EPOLLIN, (cb, args))
+
+    def add_writer(self, fd, cb, *args):
+        self.poller.register(fd, select.EPOLLOUT, (cb, args))
+
+    def wait(self, delay):
+        res = self.poller.poll(int(delay * 1000))
+        print("poll: ", res)
+        for cb, ev in res:
+            cb[0](*cb[1])
+
 
 class SysCall:
 
@@ -87,6 +106,9 @@ def handle(self):
         time.sleep(self.args[0])
 
 
+def get_event_loop():
+    return EpollEventLoop()
+
 def sleep(secs):
     yield Sleep("sleep", secs)
 
