Add new asyncio-based implementation.

Author: aaugustin

src/websockets/asyncio/client.py

@@ -2,7 +2,7 @@
 
 import asyncio
 import codecs
-from typing import AsyncIterator, List, Optional
+from typing import AsyncIterable, List, Optional
 
 from ..frames import Frame, Opcode
 from ..typing import Data
@@ -101,7 +101,7 @@ async def get(self) -> Data:
 
         return message
 
-    async def get_iter(self) -> AsyncIterator[Data]:
+    async def get_iter(self) -> AsyncIterable[Data]:
         """
         Stream the next message.
 

src/websockets/asyncio/connection.py

@@ -24,7 +24,7 @@
 
 class ClientConnection(Connection):
     """
-    Threaded implementation of a WebSocket client connection.
+    :mod:`threading` implementation of a WebSocket client connection.
 
     :class:`ClientConnection` provides :meth:`recv` and :meth:`send` methods for
     receiving and sending messages.

src/websockets/asyncio/messages.py

@@ -26,7 +26,7 @@
 
 class Connection:
     """
-    Threaded implementation of a WebSocket connection.
+    :mod:`threading` implementation of a WebSocket connection.
 
     :class:`Connection` provides APIs shared between WebSocket servers and
     clients.
@@ -229,7 +229,8 @@ def recv_streaming(self) -> Iterator[Data]:
 
         """
         try:
-            yield from self.recv_messages.get_iter()
+            for frame in self.recv_messages.get_iter():
+                yield frame
         except EOFError:
             raise self.protocol.close_exc from self.recv_events_exc
         except RuntimeError:
@@ -273,7 +274,7 @@ def send(self, message: Union[Data, Iterable[Data]]) -> None:
 
         Raises:
             ConnectionClosed: When the connection is closed.
-            RuntimeError: If a connection is busy sending a fragmented message.
+            RuntimeError: If the connection is sending a fragmented message.
             TypeError: If ``message`` doesn't have a supported type.
 
         """
@@ -552,7 +553,7 @@ def recv_events(self) -> None:
 
                 # Acquire the connection lock.
                 with self.protocol_mutex:
-                    # Feed incoming data to the connection.
+                    # Feed incoming data to the protocol.
                     self.protocol.receive_data(data)
 
                     # This isn't expected to raise an exception.
@@ -595,7 +596,7 @@ def recv_events(self) -> None:
             # Breaking out of the while True: ... loop means that we believe
             # that the socket doesn't work anymore.
             with self.protocol_mutex:
-                # Feed the end of the data stream to the connection.
+                # Feed the end of the data stream to the protocol.
                 self.protocol.receive_eof()
 
                 # This isn't expected to generate events.

src/websockets/asyncio/server.py

@@ -29,7 +29,7 @@
 
 class ServerConnection(Connection):
     """
-    Threaded implementation of a WebSocket server connection.
+    :mod:`threading` implementation of a WebSocket server connection.
 
     :class:`ServerConnection` provides :meth:`recv` and :meth:`send` methods for
     receiving and sending messages.