Merge pull request #213 from mehaase/trio_fix

gabor-boros · web-flow · commit 75c4afe0147a · 2020-06-15T17:08:25.000+02:00
Fix bug in Trio's `_read_exactly()`
diff --git a/rethinkdb/trio_net/net_trio.py b/rethinkdb/trio_net/net_trio.py
@@ -237,8 +237,11 @@ async def _read_until(self, delimiter):
         return bytes(buffer)
 
     async def _read_exactly(self, num):
+        data = b''
         try:
-            return await self._stream.receive_some(num)
+            while len(data) < num:
+                data += await self._stream.receive_some(num - len(data))
+            return data
         except (trio.BrokenResourceError, trio.ClosedResourceError):
             self._closed = True
 
