python thrift 樣本<轉>__python

轉自：http://tkang.blogspot.com/2010/07/thrift-server-client-in-python.html

在編寫python的thrift代碼時，需要先安裝thrift module

$ cd thrift-root/lib/py/$ sudo python setup.py install

下面是一個python的例子 helloworld.thrift

 

const string HELLO_IN_KOREAN = "an-nyoung-ha-se-yo"const string HELLO_IN_FRENCH = "bonjour!"const string HELLO_IN_JAPANESE = "konichiwa!"service HelloWorld { void ping(), string sayHello(), string sayMsg(1:string msg)}

生產代碼

$ thrift -r --gen py helloworld.thrift

編寫伺服器PythonServer.py

#!/usr/bin/env pythonimport syssys.path.append('./gen-py') from helloworld import HelloWorldfrom helloworld.ttypes import *from thrift.transport import TSocketfrom thrift.transport import TTransportfrom thrift.protocol import TBinaryProtocolfrom thrift.server import TServer import socketclass 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

#!/usr/bin/env pythonimport syssys.path.append('./gen-py')from helloworld import HelloWorldfrom helloworld.ttypes import *from helloworld.constants import *from thrift import Thriftfrom thrift.transport import TSocketfrom thrift.transport import TTransportfrom thrift.protocol import TBinaryProtocoltry:  # 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_IN_KOREAN)  print msg  transport.close()except Thrift.TException, tx:  print "%s" % (tx.message)

運行程式

$ python PythonServer.py$ python PythonClient.py