python thrift 实现 单端口多服务的过程

时间:2021-05-22

Thrift 是一种接口描述语言和二进制通信协议。以前也没接触过,最近有个项目需要建立自动化测试,这个项目之间的微服务都是通过 Thrift 进行通信的,然后写自动化脚本之前研究了一下。

  需要定义一个xxx.thrift的文件, 来生成各种语言的代码,生成之后我们的服务提供者和消费者,都需要把代码引入,服务端把代码实现,消费者直接使用API的存根,直接调用。

  和 http 相比,同属于应用层,走 tcp 协议。Thrift 优势在于发送同样的数据,request包 和 response包 要比 http 小很多,在整体性能上要优于 http 。

前言

学习了两天thrift 一直想实现单端口多服务 但是苦于网上的 thrift 实在太少 而且大部分都是java实现的 最后 改了一个java的 实现了 单端口多服务

实现过程

1 创建 thrift 文件 添加两个服务 Transmit Hello_test

service Transmit {string invoke(1:i32 cmd 2:string token 3:string data)}service Hello_test {string hello(1: string name)}

2 运行 thrift.exe -out gen-py --gen py test.thrift

生成对应接口 因为我的 服务端和 用户端 都是用 python写的 所以 只需要 生成python 接口即可

3 编写 server.py

# 服务类1 TransmitHandlerclass TransmitHandler: def __init__(self): self.log = {} def invoke(self, cmd, token, data): cmd = cmd token = token data = data if cmd == 1: return data + 'and' + token else: return 'cmd不匹配'# 服务类2 HelloHandlerclass HelloHandler: def hello(self, name): return 'hello'+name

4 编写服务端运行代码 开启服务端

from test import Transmitfrom test import Hello_testfrom thrift.transport import TSocketfrom thrift.transport import TTransportfrom thrift.protocol import TBinaryProtocolfrom thrift.server import TServer# 导入from thrift.TMultiplexedProcessor import TMultiplexedProcessorfrom TransmitHandler_server import TransmitHandlerfrom Hello_server import HelloHandler# open serverif __name__ == "__main__": # 实现 单端口 多服务 的方法 transmit_handler = TransmitHandler() transmit_processor = Transmit.Processor(transmit_handler) hello_handler = HelloHandler() hello_processor = Hello_test.Processor(hello_handler) transport = TSocket.TServerSocket('127.0.0.1', 8000) tfactory = TTransport.TBufferedTransportFactory() pfactory = TBinaryProtocol.TBinaryProtocolFactory() # 多 processor processor = TMultiplexedProcessor() processor.registerProcessor('transmit', transmit_processor) processor.registerProcessor('hello', hello_processor) server = TServer.TSimpleServer(processor, transport, tfactory, pfactory) print("Starting python server...") server.serve()

值得注意的是 要想实现单端口 多服务 就必须得
引入processor = TMultiplexedProcessor()
用来注册两个服务类
processor.registerProcessor(‘name', procress对象)
name 属性将会在client 时用到

5运行 runserver.py

如果出现Starting python server… 则运行成功

6 编写client.py

from test import Transmitfrom test import Hello_testfrom thrift.transport import TSocketfrom thrift.transport import TTransportfrom thrift.protocol import TBinaryProtocolfrom thrift.protocol.TMultiplexedProtocol import TMultiplexedProtocolif __name__ == '__main__': # 启动 服务 transport = TSocket.TSocket('127.0.0.1', 8000) transport = TTransport.TBufferedTransport(transport) protocol = TBinaryProtocol.TBinaryProtocol(transport) # 注册两个protocol 如果想要实现单端口 多服务 就必须使用 TMultiplexedProtocol transmit_protocol = TMultiplexedProtocol(protocol, 'transmit') hello_protocol = TMultiplexedProtocol(protocol, 'hello') # 注册两个客户端 transmit_client = Transmit.Client(transmit_protocol) hello_client = Hello_test.Client(hello_protocol) transport.open() # 打开链接 # 测试服务1 cmd = 1 token = '1111-2222-3333-4444' data = "kong_ge" msg = transmit_client.invoke(cmd, token, data) print(msg) # 测试服务2 name = '孔格' msg2 = hello_client.hello(name) print(msg2) # 关闭 transport.close()

7运行client

观察结果 实现单端口多服务

总结

核心就是 TMultiplexedProcessor 类 和 TMultiplexedProtocol
但是网上关于 thrift python的实例 太少了 导致浪费了很长时间
通过这篇文章的学习很快的明白thrift 中的一些概念

到此这篇关于python thrift 实现 单端口多服务的过程的文章就介绍到这了,更多相关python thrift单端口多服务内容请搜索以前的文章或继续浏览下面的相关文章希望大家以后多多支持!

声明:本页内容来源网络,仅供用户参考;我单位不保证亦不表示资料全面及准确无误,也不保证亦不表示这些资料为最新信息,如因任何原因,本网内容或者用户因倚赖本网内容造成任何损失或损害,我单位将不会负任何法律责任。如涉及版权问题,请提交至online#300.cn邮箱联系删除。

相关文章