
需求是用 tornado 实现一个 websocket 接口,来实现 socket 消息的透传。 由于 sock.recv 是阻塞的,协程实现的话,总是会卡在第二次循环并且超时。 大致的代码如下:
import socket import tornado.websocket from tornado import gen class WebSocketHandler(tornado.websocket.WebSocketHandler): sock = socket @gen.coroutine def _read_from_sock(self, sock: socket, buffer_size: int): while True: try: chunk = sock.recv(buffer_size) if not chunk: return self.write_message(chunk) except socket.timeout: raise tornado.websocket.WebSocketError('timeout!') def open(self, container_id): self._read_from_sock(self.sock, 4096) def on_message(self, message): self.sock.sendall(message.encode(encoding='utf-8')) def on_close(self): pass 我想在主线程中实现 socket 的监听并返回 write_message ,但是这样的协程似乎不可以,子线程试过也是会阻塞。不知道问题出在哪里?求大佬帮帮忙。谢谢了!
1 neoblackcap 2022-05-12 18:20:56 +08:00 问题就是你把 socket 跟 tornado 混在一起了,tornado 已经帮你处理了底层 socket 的逻辑,包括 IO 复用。 你需要做的是在 on_message 里面写你的逻辑,并放弃使用标准库里面的 socket ,从而使用 WebSocketHandler.write_message 等方法 |
2 fgwmlhdkkkw 2022-05-12 18:23:01 +08:00 sock.setblocking(False); |
3 GYGYG OP @neoblackcap 这里的 socket 是 open 时 需要与另外的服务端完成的 socket 链接。也就是我要实现一个 WebSocketHandler ,内部逻辑是与中转与另一个 websocket 客户端的通信。 |
4 fgwmlhdkkkw 2022-05-12 18:24:13 +08:00 @fgwmlhdkkkw #2 我不了解 tornado ,1 楼应该没问题。 |
5 neoblackcap 2022-05-12 18:47:26 +08:00 @GYGYG 最简单的方法就是,你使用 tornado 自带的 tornado.tcpclient.TCPClient 当然你像二楼说的,你自己创建一个连接也是完全可以的,前提是你要把该连接设置成非堵塞,并把它注册到 tornado 的 IOLoop 上。 |
6 zhuangzhuang1988 2022-05-12 21:40:11 +08:00 用 IOStream. |
7 Kobayashi 2022-05-13 21:55:29 +08:00 > 来实现 socket 消息的透传 你要实现一层 websocket 代理,直接在 Nginx 上做不好吗? 鉴于示例代码中你对 tornado 的理解,不建议在代码层用 Python 来做这个事情。利用 tornado 做 websocket 代理,不仅需要理解异步中利用轮询处理 socket 的方式,还要完全掌握 websocket 协议。即便我翻过 tornado 源码我也不会选择用这个方式解决问题。 |
8 lolizeppelin 2022-05-15 12:46:39 +08:00 socket 代理用 websockify 这个库就好 openstack 就用这个来转发 vnc 别自己吃力不讨好的折腾其他的 |