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请求。

相关推荐
思则变2 小时前
[Pytest] [Part 2]增加 log功能
开发语言·python·pytest
漫谈网络2 小时前
WebSocket 在前后端的完整使用流程
javascript·python·websocket
try2find4 小时前
安装llama-cpp-python踩坑记
开发语言·python·llama
泡泡以安5 小时前
安卓高版本HTTPS抓包:终极解决方案
爬虫·https·安卓逆向·安卓抓包
博观而约取5 小时前
Django ORM 1. 创建模型(Model)
数据库·python·django
精灵vector6 小时前
构建专家级SQL Agent交互
python·aigc·ai编程
q567315236 小时前
Java Selenium反爬虫技术方案
java·爬虫·selenium
Zonda要好好学习7 小时前
Python入门Day2
开发语言·python
Vertira7 小时前
pdf 合并 python实现(已解决)
前端·python·pdf
太凉7 小时前
Python之 sorted() 函数的基本语法
python