git init

Author: hacksman

requirements.txt

@@ -0,0 +1,10 @@
+# websocket_lab
+
+
+
+
+
+
+## 参考资料
+- https://medium.com/better-programming/how-to-create-a-websocket-in-python-b68d65dbd549
+- https://www.ruanyifeng.com/blog/2017/05/websocket.html

websoctet_lab/README.md

@@ -0,0 +1,28 @@
+# coding: utf-8
+# @Time : 2020/9/15 9:12 AM
+
+import asyncio
+import websockets
+from websockets import WebSocketClientProtocol
+from loguru import logger
+
+
+async def consumer_handler(websocket: WebSocketClientProtocol) -> None:
+    async for message in websocket:
+        log_message(message)
+
+
+async def cousume(hostname: str, port: int) -> None:
+    websocket_resource_url = f"ws://{hostname}:{port}"
+    async with websockets.connect(websocket_resource_url) as websocket:
+        await consumer_handler(websocket)
+
+
+def log_message(message: str) -> None:
+    logger.info(f"Message: {message}")
+
+
+if __name__ == '__main__':
+    loop = asyncio.get_event_loop()
+    loop.run_until_complete(cousume("127.0.0.1", 4000))
+    loop.run_forever()

websoctet_lab/__init__.py

@@ -0,0 +1,28 @@
+# coding: utf-8
+# @Time : 2020/9/15 9:17 AM
+
+import asyncio
+import websockets
+from websockets import WebSocketClientProtocol
+from loguru import logger
+
+
+async def consumer_handler(websocket: WebSocketClientProtocol) -> None:
+    async for message in websocket:
+        log_message(message)
+
+
+async def cousume(hostname: str, port: int) -> None:
+    websocket_resource_url = f"ws://{hostname}:{port}"
+    async with websockets.connect(websocket_resource_url) as websocket:
+        await consumer_handler(websocket)
+
+
+def log_message(message: str) -> None:
+    logger.info(f"Message: {message}")
+
+
+if __name__ == '__main__':
+    loop = asyncio.get_event_loop()
+    loop.run_until_complete(cousume("127.0.0.1", 4000))
+    loop.run_forever()

websoctet_lab/consumer.py

@@ -0,0 +1,15 @@
+# coding: utf-8
+# @Time : 2020/9/15 8:42 AM
+
+import asyncio
+import websockets
+
+
+async def produce(message: str, host: str, port: int) -> None:
+    async with websockets.connect(f"ws://{host}:{port}") as ws:
+        await ws.send(message)
+        await ws.recv()
+
+
+if __name__ == '__main__':
+    asyncio.run(produce("hi", "127.0.0.1", 4000))

websoctet_lab/cousumer_2.py

@@ -0,0 +1,43 @@
+# coding: utf-8
+# @Time : 2020/9/15 8:59 AM
+
+import asyncio
+import websockets
+from websockets import WebSocketServerProtocol
+
+from loguru import logger
+
+
+class Server:
+    clients = set()
+
+    async def register(self, ws: WebSocketServerProtocol) -> None:
+        self.clients.add(ws)
+        logger.info(f"{ws.remote_address} connects.")
+
+    async def unregister(self, ws: WebSocketServerProtocol) -> None:
+        self.clients.remove(ws)
+        logger.info(f"{ws.remote_address} disconnects.")
+
+    async def send_to_clients(self, message: str) -> None:
+        if self.clients:
+            await asyncio.wait([client.send(message) for client in self.clients])
+
+    async def ws_handler(self, ws: WebSocketServerProtocol, uri: str) -> None:
+        await self.register(ws)
+        try:
+            await self.distribute(ws)
+        finally:
+            await self.unregister(ws)
+
+    async def distribute(self, ws: WebSocketServerProtocol) -> None:
+        async for message in ws:
+            await self.send_to_clients(message)
+
+
+if __name__ == '__main__':
+    server = Server()
+    start_server = websockets.serve(server.ws_handler, "127.0.0.1", 4000)
+    loop = asyncio.get_event_loop()
+    loop.run_until_complete(start_server)
+    loop.run_forever()