使用thrift.protocol.TCompactProtocol()实现Python中的消息传递机制

在Python中使用Thrift实现消息传递机制，需要先定义消息的数据结构和协议，然后使用Thrift编译器生成相应的代码，最后通过生成的代码实现消息的发送和接收。

下面是一个使用Thrift进行消息传递的例子，具体步骤如下：

1. 安装thrift

pip install thrift

2. 定义消息的数据结构

新建一个thrift文件（如message.thrift），定义消息的数据结构和字段类型：

namespace py message
struct Message {
    1: required string content
    2: required i32 timestamp
}

3. 生成代码

使用Thrift编译器生成相应的代码（如gen-py目录下）：

thrift --gen py message.thrift

4. 客户端代码

创建一个socket连接，并使用TTransport和TCompactProtocol封装socket对象，用于发送和接收消息：

from thrift.transport import TSocket, TTransport
from thrift.protocol import TCompactProtocol
from message import Message

# 创建消息对象
message = Message()
message.content = "Hello, World!"
message.timestamp = 123456789

# 创建socket连接
socket = TSocket.TSocket('localhost', 9090)
transport = TTransport.TFramedTransport(socket)
protocol = TCompactProtocol.TCompactProtocol(transport)

# 打开连接
transport.open()

# 发送消息
protocol.writeMessageBegin("sendMessage", TMessageType.CALL, 0)
message.write(protocol)
protocol.writeMessageEnd()
transport.flush()

# 关闭连接
transport.close()

5. 服务器端代码

创建一个服务器监听指定端口，并接收消息，使用TTransport和TCompactProtocol封装socket对象，用于发送和接收消息：

from thrift.transport import TSocket, TTransport
from thrift.protocol import TCompactProtocol
from thrift.server import TServer
from message import Message

# 实现具体的消息处理逻辑
class MessageHandler:
    def sendMessage(self, message):
        print("Received message:")
        print("Content: ", message.content)
        print("Timestamp: ", message.timestamp)

# 创建处理器
handler = MessageHandler()

# 创建消息传输通道
transport = TTransport.TServerSocket(port=9090)
tfactory = TTransport.TFramedTransportFactory()
pfactory = TCompactProtocol.TCompactProtocolFactory()

# 创建服务端
server = TServer.TSimpleServer(processor, transport, tfactory, pfactory)

# 启动服务端
print("Starting the server...")
server.serve()
print("done.")

6. 运行代码

先运行服务器端代码，然后再运行客户端代码，可以看到服务器端输出接收到的消息：

Starting the server...
Received message:
Content:  Hello, World!
Timestamp:  123456789
done.

通过以上步骤，我们成功实现了基于Thrift的消息传递机制。可以根据实际需求，添加更多的消息类型和字段，通过定义不同的方法来处理不同类型的消息。通过Thrift提供的TCompactProtocol，可以实现较小的数据序列化和更高效的传输机制，从而提高消息传递的性能。