需求是用 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 ,但是这样的协程似乎不可以,子线程试过也是会阻塞。不知道问题出在哪里?求大佬帮帮忙。谢谢了!