thrift之python客户端使用

| 2014年2月13日

  1. 写好了cpp的服务器后,测试是个问题,写cpp程序来测试好像不是很灵活,所以想用python脚本来写测试程序,记录一下过程。

  2. 首先是thrift的python程序库的安装,有些基础库需要安装后才能运行,在thrift的目录库目录中,选择py目录,使用root账户就可以直接安装了。

cd thrift-0.9.0/lib/py
python setup.py install

即可完成安装。当然安装完成之后也要记得测试一下,安装是否成功。

  1. 编译thrift脚本文件为py语言。
thrift --gen py meta_node.thrift

在但前目录下生成gen-py目录。

ls gen-py/
__init__.py  meta_node
helightxu@TengDa_10_6_209_151:meta_node>ls gen-py/meta_node/
constants.py   __init__.py   MetaNodeService.py   MetaNodeService-remote  ttypes.py

这里就是thrift文件中定义的服务器类,客户端类,和数据结构的定义了,很简单,看看就明白。

  1. 编写测试程序
#!/usr/bin/env python

import sys
sys.path.append('gen-py')
from meta_node import MetaNodeService     # 引入客户端类
from meta_node.ttypes import *            # 数据类型
from meta_node.constants import *         # 该文件是用来引入thrift的数据类型
# 下面这几个就不用多说了
from thrift import Thrift
from thrift.transport import TSocket
from thrift.transport import TTransport
from thrift.protocol import TBinaryProtocol

try:
# 建立socket
transport = TSocket.TSocket('127.0.0.1', 9300)
# 选择传输层,这块要和服务器的设置一样
transport = TTransport.TFramedTransport(transport)
#选择传输协议,这个也要和服务器保持一致,负责无法通信
protocol = TBinaryProtocol.TBinaryProtocol(transport)
# 创建客户端
client = MetaNodeService.Client(protocol)
transport.open()
#调用rpc方法
active_meta = client.add_dir("/asdfasd")
print active_meta
# 关闭传输
transport.close()
#捕获异常
except Thrift.TException, tx:
print "%s" % (tx.message)
  1. 直接运行即可
python py_client.py
  1. 总结

使用python写客户端测试相对c++快捷,调试和方便,但是还要在测试机器上安装thrift的python库,对于移动到其它机器上测试就不方便了,虽然也可以把python库抽取的单独的目录中也同时移动到测试机器,但是这种依赖还是比较麻烦,不过一般服务器部署后简单测试只要在固定几台机器上做也就相对方便一点。

看完本文有收获?请分享给更多人

关注「黑光技术」,关注大数据+微服务