**项目推荐:Django Channels Redis —— 高性能WebSocket和异步消息处理的优选方案**...
from channels.generic.websocket import AsyncWebsocketConsumer
from channels.routing import route
from asgiref.sync import async_to_sync
import redis
# 连接到Redis
redis_client = redis.StrictRedis(host='localhost', port=6379, db=0)
class ChatConsumer(AsyncWebsocketConsumer):
async def connect(self):
# 连接WebSocket时被调用
await self.accept()
await redis_client.hset('chat_consumers', self.channel_name, '')
async def disconnect(self, close_code):
# 断开WebSocket连接时被调用
await redis_client.hdel('chat_consumers', self.channel_name)
async def receive(self, text_data):
# 接收到消息时被调用
await redis_client.lpush('chat_messages', text_data)
await self.send_chat_message(text_data)
async def send_chat_message(self, message):
# 发送消息给所有连接的客户端
await self.send(text_data=message)
@staticmethod
async def broadcast_chat_message(message):
# 广播消息给所有连接的客户端
for consumer in async_to_sync(redis_client.hkeys)('chat_consumers'):
await consumer.send_chat_message(message)
# 路由定义
application = route('websocket.connect', ChatConsumer.connect) \
.route('websocket.disconnect', ChatConsumer.disconnect) \
.route('websocket.receive', ChatConsumer.receive)
这个示例代码展示了如何使用Django Channels和Redis来创建一个简单的WebSocket聊天服务。它包括连接处理、断开连接处理、接收消息处理以及广播消息给所有连接的客户端。这个例子是一个很好的起点,展示了如何将Redis用作存储活跃的WebSocket连接和消息队列。
评论已关闭