six_001

django_channels/SSE测试.py

@@ -0,0 +1,79 @@
+"""注释"""
+# consumers.py
+# 推荐解决方案：事件驱动架构
+# 修改实现为事件驱动模式：
+import asyncio
+import json
+import time
+
+from channels.generic.http import AsyncHttpConsumer
+
+# 全局消息队列 (生产环境用Redis等分布式队列)
+MESSAGE_QUEUE = asyncio.Queue()
+
+
+class SSEConsumer(AsyncHttpConsumer):
+    async def handle(self, body):
+        # 设置SSE响应头
+        await self.send_headers(
+            headers=[
+                (b"Content-Type", b"text/event-stream"),
+                (b"Cache-Control", b"no-cache"),
+                (b"Connection", b"keep-alive"),
+            ]
+        )
+
+        # 事件循环监听
+        while True:
+            # 异步等待新消息（挂起状态）
+            data = await MESSAGE_QUEUE.get()
+            # 推送消息给客户端
+            await self.send_body(f"event: message\ndata: {data}\n\n".encode())
+
+    async def disconnect(self):
+        # 客户端断开时清理资源
+        print(f"Client {self.scope['client']} disconnected")
+
+
+# 其他Django视图或后台任务中
+
+# 生产者触发更新示例
+async def trigger_update():
+    """模拟外部事件触发更新"""
+    # 创建新消息
+    message = {"time": time.time(), "event": "DataUpdated"}
+
+    # 将消息推送到全局队列
+    await MESSAGE_QUEUE.put(json.dumps(message))
+
+
+# 测试代码
+async def xiaozizi():
+    """从队列中获取消息并打印"""
+    print("等待消息...")
+    d = await MESSAGE_QUEUE.get()
+    print(f"收到消息: {d}")
+    return d
+
+
+# 在脚本末尾添加（如果需要测试）
+async def main():
+    """主测试函数"""
+    print("开始测试...")
+
+    # 启动一个任务来监听消息
+    listen_task = asyncio.create_task(xiaozizi())
+
+    # 稍等一下再发送消息，确保监听任务已经启动
+    await asyncio.sleep(0.1)
+
+    # 发送测试消息
+    # await trigger_update()
+
+    # 等待监听任务完成
+    result = await listen_task
+    print(f"测试完成，结果: {result}")
+
+
+if __name__ == "__main__":
+    asyncio.run(main())