asyncio_micro: Implement proper write() handling.

Paul Sokolovsky · Paul Sokolovsky · commit 1d9f7856f68c · 2014-10-11T05:19:51.000+03:00
TODO: Test!
diff --git a/asyncio_micro/asyncio_micro.py b/asyncio_micro/asyncio_micro.py
@@ -223,10 +223,19 @@ def __init__(self, s):
         self.s = s
 
     def write(self, buf):
-        res = self.s.write(buf)
-        log.debug("StreamWriter.write(): %d", res)
-        s = yield IOWrite(self.s)
-        log.debug("StreamWriter.write(): returning")
+        sz = len(buf)
+        while True:
+            res = self.s.write(buf)
+            log.debug("StreamWriter.write(): %d", res)
+            # If we spooled everything, (just) return
+            if res == sz:
+                return
+            if res is None:
+                res = 0
+            buf = buf[res:]
+            sz -= res
+            s = yield IOWrite(self.s)
+            log.debug("StreamWriter.write(): can write more")
 
     def close(self):
         yield IODone(IO_WRITE, self.s)
