Twisted 与 Tornado 中的 WebSocket 连接问题及解决方案

1、问题背景

项目中我们需要通过 Tornado HTTP 处理程序建立WebSocket连接,该连接需要处理多个用户请求,并且将从外部服务器获取的数据存储到数据库中。我们尝试了以下实现:

python 复制代码
from twisted.internet import reactor
from autobahn.websocket import WebSocketClientFactory, WebSocketClientProtocol, connectWS
from tornado.options import define, options, parse_command_line

class IndexHandler(tornado.web.RequestHandler):
    @tornado.web.asynchronous
    def get(self):
        self.write("This is your response")
        factory = WebSocketClientFactory("ws://localhost:7096")
        factory.protocol = BridgeSocket
        connectWS(factory)
        self.finish()
        reactor.run()

class BridgeSocket(WebSocketClientProtocol):

    def sendHello(self):
        self.sendMessage("rails")

    def onOpen(self):
        self.sendHello()

    def onMessage(self, msg, binary):
        print "Got echo: " + msg

    def onClose(wasClean,code,reason):
        print "GETTING CLOSE CONNECTION"
        print str(wasClean)+" ---"+str(code)+"---"+str(reason)
        reactor.stop()

但是,reactor.run()会阻止对Tornado Web服务器的进一步HTTP请求,如果我们在WebSocket工作完成后立即调用reactor.stop(),又会发现无法重新启动reactor。

2、解决方案

为了在Tornado中运行AutobahnPython的WebSocket客户端,我们需要使用Twisted-Tornado集成("Twisted on Tornado")。这个库允许我们在Tornado中运行Twisted reactor循环。

以下是如何使用Twisted on Tornado来解决问题的步骤:

  1. 安装Twisted on Tornado:

    pip install twisted-tornado

  2. 在你的Tornado应用程序中导入Twisted on Tornado:

javascript 复制代码
from twisted.internet import reactor
from twisted.internet.defer import Deferred
from tornado.ioloop import IOLoop
  1. 在你的Tornado HTTP处理程序中,使用Twisted on Tornado的reactor来运行WebSocket客户端:
css 复制代码
def main():
    reactor.suggestThreadPoolSize(1)  # Increase threadpool size for multiple persistent connections
    IOLoop.instance().run(True)

reactor.callWhenRunning(main)
  1. 修改你的WebSocket客户端协议类,使其继承自Twisted的WebSocketClientFactory:
ruby 复制代码
class BridgeSocket(WebSocketClientFactory):
    def buildProtocol(self, addr):
        return BridgeSocketProtocol()

class BridgeSocketProtocol(WebSocketClientProtocol):

    def sendHello(self):
        self.sendMessage("rails")

    def onOpen(self):
        self.sendHello()

    def onMessage(self, msg, binary):
        print "Got echo: " + msg

    def onClose(self, wasClean, code, reason):
        print "GETTING CLOSE CONNECTION"
        print str(wasClean)+" ---"+str(code)+"---"+str(reason)
        reactor.stop()
  1. 在你的Tornado应用程序中,使用Twisted on Tornado的reactor来连接到WebSocket服务器:
scss 复制代码
from twisted.internet import reactor

def connect_to_websocket():
    factory = BridgeSocket()
    reactor.connectTCP("localhost", 7096, factory)

reactor.callWhenRunning(connect_to_websocket)

这样,我们就可以在Tornado中使用AutobahnPython的WebSocket客户端,而不会阻止对Tornado Web服务器的进一步HTTP请求。

相关推荐
Tony_long74833 小时前
Python学习——猜拳小游戏
开发语言·python·学习
陈苏同学4 小时前
机器翻译 & 数据集 (NLP基础 - 预处理 → tokenize → 词表 → 截断/填充 → 迭代器) + 代码实现 —— 笔记3.9《动手学深度学习》
人工智能·pytorch·笔记·python·深度学习·自然语言处理·机器翻译
股票GPT分析4 小时前
《Python 股票交易分析:开启智能投资新时代》(二)
大数据·服务器·python·c#·fastapi
大熊程序猿4 小时前
python Flask指定IP和端口
开发语言·python·flask
我叫白小猿4 小时前
【大模型-智能体】AutoGen Studio测试和导出工作流程
人工智能·python·workflow·工作流·智能体·autogen
K2SO4钾4 小时前
16. 清理Python包管理工具(pip 和 conda)的缓存和冗余文件
python·conda·pip
YONG823_API5 小时前
1688商品数据采集API的测试对接步骤分享(提供免费测试key)
开发语言·数据库·爬虫·python·数据挖掘
好看资源平台5 小时前
网络爬虫总结与未来方向
爬虫·python
YiSLWLL5 小时前
Django+Nginx+uwsgi网站使用Channels+redis+daphne实现简单的多人在线聊天及消息存储功能
服务器·数据库·redis·python·nginx·django
陈王卜5 小时前
django+boostrap实现注册
python·django