服务器

Thrift提供的常见服务端类型有一下几种：

• thrift.server.TServer.TSimpleServer 单线程服务器
• thrift.server.TServer.TThreadedServer 多线程服务器
• thrift.server.TServer.TThreadPoolServer 线程池服务器
• thrift.server.TServer.TForkingServer 多进程服务器

我们以线程池服务器为例，其他方式相同。

      import sys
      sys.path.append('gen-py')  # 增加生成代码的查找包路径
      from calculate import Calculate
      from base.ttypes import InvalidOperation, Operation
      from thrift.transport import TSocket
      from thrift.transport import TTransport
      from thrift.protocol import TCompactProtocol
      from thrift.server import TServer
      class CalculateHandler(Calculate.Iface):
         """被调用方法的具体实现"""
         def ping(self):
              print('ping()')
         def divide(self, num1, num2):
             if num2 == 0:
                 raise InvalidOperation(0, 'Cannot divide by 0')
             return num1 / num2
         def calculate(self, work):
             if work.op == Operation.ADD:
                  val = work.num1 + work.num2
             elif work.op == Operation.SUBTRACT:
                  val = work.num1 - work.num2
             elif work.op == Operation.MULTIPLY:
                  val = work.num1 * work.num2
             else:
                 raise InvalidOperation(work.op, 'Invalid operation')
             return val
      if __name__ == '__main__':
          handler = CalculateHandler()
          processor = Calculate.Processor(handler)
          transport = TSocket.TServerSocket(host='127.0.0.1', port=8888)
          tfactory = TTransport.TBufferedTransportFactory()
          pfactory = TCompactProtocol.TCompactProtocolFactory()
          server = TServer.TThreadPoolServer(processor, transport, tfactory, pfactory)
          server.serve()
          print('服务器已启动')
  
 

客户端

      import sys
      sys.path.append('gen-py')
      from calculate import Calculate
      from base.ttypes import InvalidOperation, Operation, Work
      from thrift.transport import TSocket
      from thrift.transport import TTransport
      from thrift.protocol import TCompactProtocol
      def main():
          transport = TSocket.TSocket('127.0.0.1', 8888)
         # 使用缓存传输
          transport = TTransport.TBufferedTransport(transport)
         # 使用压缩的二进制消息协议
          protocol = TCompactProtocol.TCompactProtocol(transport)
          client = Calculate.Client(protocol)
         # 连接
          transport.open()
          client.ping()
          print('ping()')
          result = client.divide(100, 50)
          print('100/50={}'.format(result))
         try:
              result = client.divide(100, 0)
         except InvalidOperation as e:
              print(e.why)
          work = Work(1, 2, Operation.ADD)
          result = client.calculate(work)
          print('1+2={}'.format(result))
          work = Work(15, 10, Operation.SUBTRACT)
          result = client.calculate(work)
          print('15-10={}'.format(result))
         # 关闭
          transport.close()
      if __name__ == '__main__':
          main()
  
 

文章来源: lansonli.blog.csdn.net，作者：Lansonli，版权归原作者所有，如需转载，请联系作者。

原文链接：lansonli.blog.csdn.net/article/details/105304671