2019年07月30日 14:59:29 Shower稻草人阅读数 25更多

分类专栏： Python

本文链接： https://blog.csdn.net/u013474436/article/details/97782226

0x00

首先需要安装python的thrift包

sudo pip install thrift

0x01

编写一个简单接IDL文件helloworld.thrift

const string HELLO_WORLD = "world"

service HelloWorld {
    void ping(), string sayHello(), string sayMsg(1:string msg) }

thrift脚本通过Thrift编辑器生成所要求的python开发语言代码。即：

thrift -r --gen py helloworld.thrift

生成gen-py目录：

├── gen-py
│   ├── helloworld
│   │   ├── constants.py
│   │   ├── HelloWorld.py
│   │   ├── HelloWorld-remote
│   │   ├── __init__.py
│   │   └── ttypes.py
│   └── __init__.py
└── helloworld.thrift

0x02

Thrift是一个典型的CS结构，客户端和服务端可以使用不同的语言开发。本文以python为例：
PythonServer.py

import sys
sys.path.append('./gen-py') from helloworld import HelloWorld from helloworld.ttypes import * from thrift.transport import TSocket from thrift.transport import TTransport from thrift.protocol import TBinaryProtocol from thrift.server import TServer import socket class HelloWorldHandler: def __init__(self): self.log = {} def ping(self): print "ping()" def sayHello(self): print "sayHello()" return "say hello from " + socket.gethostbyname(socket.gethostname()) def sayMsg(self, msg): print "sayMsg(" + msg + ")" return "say " + msg + " from " + socket.gethostbyname(socket.gethostname()) handler = HelloWorldHandler() processor = HelloWorld.Processor(handler) transport = TSocket.TServerSocket('127.0.0.1',30303) tfactory = TTransport.TBufferedTransportFactory() pfactory = TBinaryProtocol.TBinaryProtocolFactory() server = TServer.TSimpleServer(processor, transport, tfactory, pfactory) print "Starting python server..." server.serve() print "done!"

PythonClient.py

import sys
sys.path.append('./gen-py') from helloworld import HelloWorld from helloworld.ttypes import * from helloworld.constants import * from thrift import Thrift from thrift.transport import TSocket from thrift.transport import TTransport from thrift.protocol import TBinaryProtocol try: # Make socket transport = TSocket.TSocket('127.0.0.1', 30303) # Buffering is critical. Raw sockets are very slow transport = TTransport.TBufferedTransport(transport) # Wrap in a protocol protocol = TBinaryProtocol.TBinaryProtocol(transport) # Create a client to use the protocol encoder client = HelloWorld.Client(protocol) # Connect! transport.open() client.ping() print "ping()" msg = client.sayHello() print msg msg = client.sayMsg(HELLO_WORLD) print msg transport.close() except Thrift.TException, tx: print "%s" % (tx.message)

当前目录结构：

├── gen-py
├── helloworld.thrift
├── PythonClient.py
└── PythonServer.py

0x03

运行server端：

$ python PythonServer.py 
Starting python server...

运行client端：

$ python PythonClient.py 
ping()
say hello from 10.27.73.176
say world from 10.27.73.176

server端输出：

ping()
sayHello()
sayMsg(world)

0x04 使用thriftpy

thriftpy（现已更新为thriftpy2）对thrift进行了封装，可以动态解析thrift接口文件。项目地址：https://github.com/Thriftpy/thriftpy2

扫描二维码关注公众号，回复： 7240266 查看本文章

安装thriftpy2

sudo pip install thriftpy2

编写IDL

pingpong.thrift

service PingService {
    string ping(), } service AargsPingService { string ping(1:string ping); } service Sleep { oneway void sleep(1: i32 seconds) }

编写代码

server.py

# coding=utf-8
import thriftpy2
from thriftpy2.rpc import make_server pp_thrift = thriftpy2.load("pingpong.thrift", module_name="pp_thrift") # 实现.thrift文件定义的接口 class Dispatcher(object): def ping(self): print("ping pong!") return 'pong' def main(): # 定义监听的端口和服务 server = make_server(pp_thrift.PingService, Dispatcher(), '127.0.0.1', 6000) print("serving...") server.serve() if __name__ == '__main__': main()

client.py

# coding=utf-8
import thriftpy2
#from thriftpy2.rpc import client_context
from thriftpy2.rpc import make_client # 读入thrift文件，module_name最好与server端保持一致，也可以不保持一致 pp_thrift = thriftpy2.load("pingpong.thrift", module_name="pp_thrift") def main(): #with client_context(pp_thrift.PingService, '127.0.0.1', 6000) as c: # pong = c.ping() # print(pong) client = make_client(pp_thrift.PingService, '127.0.0.1', 6000) print(client.ping()) if __name__ == '__main__': main()

注释里是另一种创建链接的方式。

目录结构：

├── client.py
├── pingpong.thrift
└── server.py

运行