Tornado websocket 回调 on_message

Question

我实现了在我的 python 应用程序中有两个线程工作。 现在我想在 on_message 中添加一个回调函数来处理另一个线程中的接收消息。

这是我的代码：

import tornado.websocket
import tornado.web
import message.MessageHandler

from message.messageConstructor import MessageConstructor


class WebSocketHandler(tornado.websocket.WebSocketHandler):
    def open(self):
        print 'new connection'
        self.send(message.MessageHandler.serialize(MessageConstructor('Engine1', '100')))
        self.send("CONNECT!")

    def on_message(self, rawMessage):
        print rawMessage
        obj = message.MessageHandler.deserialize(rawMessage)
        callback(message)

    def on_close(self):
        print 'connection closed'

    def check_origin(self, origin):
        return True

    def send(self, message):
        self.write_message(message)

和创造：

import tornado.httpserver
import tornado.websocket
import tornado.ioloop
import tornado.web
import socket

from websocket import connectionHandler


class WebSocketConnection():
    def __init__(self, port, callback):
        self.socketHandler = connectionHandler.WebSocketHandler

        application = tornado.web.Application([
            (r'/', self.socketHandler),
        ])

        callback("World")

        http_server = tornado.httpserver.HTTPServer(application)
        http_server.listen(7004)
        myIP = socket.gethostbyname(socket.gethostname())
        print '*** Websocket Server Started at %s***' % myIP
        tornado.ioloop.IOLoop.instance().start()

    def send(self, message):
        self.socketHandler.send(message)

如何为未实例化的 WebsocketHandler 实例提供回调？

Answer 1

您可以使用RequestHandler.initialize 将配置通过应用程序传递给处理程序：

class WebSocketHandler(tornado.websocket.WebSocketHandler):
    def initialize(self, callback):
        self.callback = callback

app = Application([('/', WebSocketHandler, dict(callback=callback))])