Question

我有简单的websocket服务器，它将所有消息回送给客户端。

import gevent
from geventwebsocket.resource import WebSocketApplication
from geventwebsocket.server import WebSocketServer
from geventwebsocket.resource import Resource
import ams_pb2

class AMSWebSocketServer(WebSocketApplication):
    def __init__(self, ws):
        super(AMSWebSocketServer, self).__init__(ws)
        pass

    def on_open(self):
        pass

    def on_message(self, message):
        print 'received message'
        print message
        if message is None:
            print 'message none'
            return

        print 'echo message back'
        self.ws.send(message)

    def on_close(self, reason):
        print "connection closed"
        gevent.sleep(0)


resource = Resource({'/': AMSWebSocketServer})

使用gunicorn命令生成服务器

gunicorn -k "geventwebsocket.gunicorn.workers.GeventWebSocketWorker" -b 127.0.0.1:9000 gunicorn_test:resource

我有一个测试客户端，它发送websocket消息以回显

from ws4py.client.threadedclient import WebSocketClient
import ams_pb2


class DummySwitch(WebSocketClient):
    def closed(self, code, reason=None):
        pass

    def received_message(self, msg):
        if msg is None:
            print 'none'
            return
        print 'received message'
        ams_message = ams_pb2.AMSConfig()
        ams_message.ParseFromString(msg)
        print ams_message
        print msg


if __name__ == '__main__':
    end_point = 'ws://127.0.0.1:9000'

    client = DummySwitch(
        end_point,
        headers=[
        ]
    )
    client.connect()

    print 'sending message'

    AMSConfig = ams_pb2.AMSConfig()
    AMSConfig.CliConfig = True
    print AMSConfig
    msg = AMSConfig.SerializeToString()

    #msg = 'Hello'
    print msg

    client.send(msg)

    client.run_forever()

我的protobuff文件是：包ams;

message AMSConfig {
  optional bool CliConfig = 1;
}

每当我的客户端向服务器发送一个protobuff消息时，我能够看到它在服务器中被解析，但是当服务器向客户端回送相同的消息时，客户端由于以下原因而失败：

文件“client_test.py”，第15行，在received_message中 ams_message.ParseFromString（MSG） ParseFromString中的文件“/usr/lib/python2.6/site-packages/google/protobuf/message.py”，第186行 self.MergeFromString（连载） MergeFromString中的文件“/usr/lib/python2.6/site-packages/google/protobuf/internal/python_message.py”，第847行提出message_mod.DecodeError（'截断的消息。'） DecodeError：截断的消息。

所以，我修改了代码以发送一个简单的字符串，我看到发送到服务器的'Hello'字符串正在回显，客户端能够打印消息。但是，客户端无法解析回传的protobuff消息。

我无法理解为什么字符串的回显有效，但对于协议缓冲区，它在我的示例中不起作用。

感谢您的帮助。

Answer 1

我在客户端遇到同样的问题，但这可以通过扩大客户端代码中的接收缓冲区大小来解决。例如：

maxpkglen = 1024 * 1024 // original maxpkglen = 1024
s = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
s.sendto(req, (server, port))
rsp, server = s.recvfrom(maxpkglen)
s.close()

Answer 2

我在客户端程序中也遇到这样的问题，因为我将接收缓冲区设置为1024的硬编码。当我将接收缓冲区大小放大到10 * 1024时，它就解决了。还要感谢@lulyon的小费。

您可能需要一个循环来读取对等代码中的所有数据！

Answer 3

我遇到了这个问题，但对我来说，答案是不直接使用received_message(self, msg)中的msg.data变量，而是使用vagrant来实际获取与消息相关联的数据，解析。

python gevent websocket客户端获取错误“DecodeError：Truncated message”用于回显协议缓冲区消息

3 个答案: