使用不同的函数在单个高速公路 WebSocketProtocol 中处理消息答案

【问题标题】：Using different functions to process messages in a single autobahn WebSocketProtocol使用不同的函数在单个高速公路 WebSocketProtocol 中处理消息
【发布时间】：2013-09-04 23:24:37
【问题描述】：

我对 websockets 很陌生，一直在尝试将 python 与高速公路一起使用，并通过扭曲连接到多个套接字，然后对来自每个不同套接字的数据执行不同的操作。

最后，我将拥有 3 到 4 个不同的处理功能和多达 10-15 个 websocket 连接。

我想将一个函数传递给协议以用于分析返回的数据。我遇到的问题是我能弄清楚访问 WebSocketClientProtocol 的“onMessage()”方法中的函数的唯一方法是为类而不是实际对象设置的，所以我坚持使用一个数据处理函数适用于所有 websocket。

下面的代码示例。

from autobahn.websocket import WebSocketClientFactory, WebSocketClientProtocol
from twisted.internet import reactor, ssl

# message processing methods
def method1(self, data):
    print 'Using Method 1.'

def method2(self, data):
    print 'Using Method 2.'

# factory and protocol classes
class myFactory(WebSocketClientFactory):
    def clientConnectionLost(self, connector, reason):
        connector.connect()

class myWS(WebSocketClientProtocol):

    def initialize(self, url, data_fcn):
        self.factory = myFactory(url)
        self.factory.protocol = myWS

        # !!! set in class, not object !!!
        self.factory.protocol.data_fcn = data_fcn 

    def onMessage(self, msg, binary):
        self.data_fcn(msg)

    def onClose(self, a, b, c):
        reactor.stop()
        reactor.disconnectAll()

    def kill(self):
        self.transport.loseConnection()
        reactor.stop()
    reactor.disconnectAll()


if __name__ == '__main__':

    # websocket topics unique
    topic_ids = [ '3395',
                  '3563',
                  '3562' ]

    # data processing functions can repeat
    data_process_fcns = [ method1,
                          method2,
                          method1 ]

    for topic, func in zip(topic_ids, data_process_fcns):

        url = 'wss://mywebsocket.com/feed?topic_id[]=' + topic
        ws = myWS()
        ws.initialize(url, func)

        reactor.connectSSL(ws.factory.host, ws.factory.port, ws.factory, ssl.ClientContextFactory())

    reactor.run()

我目前的解决方案是为我想使用的每个数据处理函数创建一个类。所以，对于method1我有

class myWS_method1(WebSocketClientProtocol):

与

    def onMessage(self, msg, binary):
        method1(msg)

以及我想使用的每种数据处理方法的类似类。

希望找到一个更优雅的解决方案，让我可以重用单个类。

谢谢，

【问题讨论】：

标签： python twisted autobahn

【解决方案1】：

在 Twisted 中，您几乎不会自己构建协议，这是工厂的责任。我想要这样的东西：

class myFactory(WebSocketClientFactory):

    def __init__(self, url, data_processing):
        WebSocketClientFactory.__init__(self, url)
        # here you could have passed whatever parameters
        # later allow you to build up the protocol instance
        self._data_processing = data_processing

    def buildProtocol(self, addr):
        # Use this method to control the creation of the 
        # protocol instance.
        # You can either build a protocol here and inject
        # data_func, or use different protocol class type.

        protocol = myWs()
        protocol.data_func = self._data_processing
        return protocol

然后，带有连接的循环看起来像：

for topic, func in zip(topic_ids, data_process_fcns):
    url = 'wss://mywebsocket.com/feed?topic_id[]=' + topic
    factory = myFactory(url, func)
    reactor.connectSSL(factory.host, factory.port, factory, ssl.ClientContextFactory())

【讨论】：