asyncio_micro: StreamReader, StreamWriter: add more methods.

pfalcon · pfalcon · commit 4e3964d3fff6 · 2014-10-11T05:19:50.000+03:00
diff --git a/asyncio_micro/asyncio_micro.py b/asyncio_micro/asyncio_micro.py
@@ -186,6 +186,13 @@ class StreamReader:
     def __init__(self, s):
         self.s = s
 
+    def read(self, n):
+        s = yield IORead(self.s)
+        res = self.s.read(n)
+        if not res:
+            yield IODone(IO_READ, self.s)
+        return res
+
     def readline(self):
         log.debug("StreamReader.readline()")
         s = yield IORead(self.s)
@@ -208,6 +215,10 @@ def write(self, buf):
         s = yield IOWrite(self.s)
         log.debug("StreamWriter.write(): returning")
 
+    def close(self):
+        yield IODone(IO_WRITE, self.s)
+        self.s.close()
+
 
 def open_connection(host, port):
     log.debug("open_connection(%s, %s)", host, port)
