求助:关于 Channels2.0 的使用问题

2018-07-23 16:27:04 +08:00
 vimiix

问题背景:

我需要在后台订阅 redis 的消息,一有消息就给前端推信息。所以我这么做了:

    async def connect(self):
        logger.info("websocket connected.")
        redis_conn = StrictRedis(**settings.REDIS_DB)
        self.ps = redis_conn.pubsub()
        await self.ps.subscribe(settings.REDIS_CHANNELS)

        await self.accept()

    async def disconnect(self, code):
        logger.info("websocket disconnected.")
        await self.ps.unsubscribe()
        await self.close()

    async def receive(self, text_data=None, bytes_data=None):
        logger.info("websocket received data.\n%s" % text_data)
        try:
            data = json.loads(text_data)
            self.user_id = data['payload'].get('user_id')
        except json.JSONDecodeError:
            await self.close()

        while True:
            # 获取 redis 的消息
            msg = await self.get_msg_from_queue()
            if not msg:
                continue
            logger.info("get msg %s" % msg)
            await self.send(text_data=json.dumps(msg))

打开前端页面的时候,链接成功 websocket 会立马发一个信息,来对接,后端确认以后开始等 redis 的消息,这里我使用的 while 循环,就是因为这个 while 循环,导致我前端断开 websocket 链接以后,后端不会触发 disconnect()函数,所有就没有关闭这个实例,导致后端不断报这个错误:

WARNING  Application instance <Task pending coro=<AsyncConsumer.__call__() 
running at /xxx/venv/lib/python3.6/site-packages/channels/consumer.py:54> 
wait_for=<Future pending cb=[<TaskWakeupMethWrapper object at 0x7fade64e0678>()]>> 
for connection <WebSocketProtocol client=['127.0.0.1', 49204] path=b'/listen'> 
took too long to shut down and was killed.

求助 V 站大佬,我该如何在 while 循环中来判断当前客户端是否断开呢?

2899 次点击
所在节点    Python
3 条回复
myyou
2018-07-23 18:22:14 +08:00
如果你用的是 channels-redis,看看源码,等待消息的逻辑已经实现了,你不用重复搞一遍。
你这样搞等于把接受消息端的给阻塞了。
virusdefender
2018-07-23 18:46:16 +08:00
不要在 receive 中阻塞,而是在一个单独的进程中 pull 消息,然后推送给响应的 client

http://channels.readthedocs.io/en/latest/introduction.html#cross-process-communication
vimiix
2018-07-23 18:55:25 +08:00
@myyou 好的,谢谢指点。

这是一个专为移动设备优化的页面(即为了让你能够在 Google 搜索结果里秒开这个页面),如果你希望参与 V2EX 社区的讨论,你可以继续到 V2EX 上打开本讨论主题的完整版本。

https://www.v2ex.com/t/473364

V2EX 是创意工作者们的社区,是一个分享自己正在做的有趣事物、交流想法,可以遇见新朋友甚至新机会的地方。

V2EX is a community of developers, designers and creative people.

© 2021 V2EX