python怎么實現Thrift服務端

這篇文章將為大家詳細講解有關python怎么實現Thrift服務端，小編覺得挺實用的，因此分享給大家做個參考，希望大家閱讀完這篇文章后可以有所收獲。

python可以做什么

Python是一種編程語言，內置了許多有效的工具，Python幾乎無所不能，該語言通俗易懂、容易入門、功能強大，在許多領域中都有廣泛的應用，例如最熱門的大數據分析，人工智能，Web開發等。

環境準備

• 根據自身實際情況下載對應的Thrift編譯器，比如我在Windows系統上使用的是thrift-0.9.3.exe 。下載地址：http://archive.apache.org/dist/thrift/

• python安裝thrift庫：pip install thrift
  

編寫.thrift文件

.thrift文件定義了Thrift服務端和Thrift客戶端的通信接口，在該文件中定義的接口需由服務端實現，并可被客戶端調用。Thrift編譯器會調用.thrift文件生成不同語言的thrift代碼，用于之后實現thrift服務端或thrift客戶端。

.thrift文件的編寫規則可參考Thrift白皮書。下面將以demo.thrift文件舉例

service DemoService{
    string ping(1:string param)
    map<i32,string> get_int_string_mapping_result(1:i32 key, 2:string value)
    bool get_bool_result()
}

生成python對應的thrift代碼

使用以下命令可以生成不同語言的thrift代碼：

thrift --gen <language> <Thrift filename>

 通過thrift-0.9.3.exe --gen py demo.thrift 命令生成python版本的thrift文件，文件夾為gen-py，如下所示：

編寫服務端

編寫服務端server.py，用于實現在demo.thrift文件中定義的接口功能。

from thrift.transport import TSocket
from thrift.transport import TTransport
from thrift.protocol import TBinaryProtocol
from thrift.server import TServer
import sys

sys.path.append("./gen-py/")
from demo import DemoService
import random


class DemoServer:
    def __init__(self):
        self.log = {}

    def ping(self, param):
        return "echo:" + param

    def get_int_string_mapping_result(self, key, value):
        return {key: value}

    def get_bool_result(self):
        return random.choice([True, False])


if __name__ == '__main__':
    # 創建處理器
    handler = DemoServer()
    processor = DemoService.Processor(handler)

    # 監聽端口
    transport = TSocket.TServerSocket(host="0.0.0.0", port=9999)

    # 選擇傳輸層
    tfactory = TTransport.TBufferedTransportFactory()

    # 選擇傳輸協議
    pfactory = TBinaryProtocol.TBinaryProtocolFactory()

    # 創建服務端
    server = TServer.TThreadPoolServer(processor, transport, tfactory, pfactory)

    # 設置連接線程池數量
    server.setNumThreads(5)

    # 啟動服務
    server.serve()

編寫客戶端用于測試

編寫客戶端client.py，用于測試服務端功能是否可用。

from thrift import Thrift
from thrift.transport import TSocket
from thrift.transport import TTransport
from thrift.protocol import TBinaryProtocol
import sys
sys.path.append("./gen-py/")

from demo import DemoService


if __name__ == '__main__':
    transport = TSocket.TSocket('127.0.0.1', 9999)
    transport = TTransport.TBufferedTransport(transport)
    protocol = TBinaryProtocol.TBinaryProtocol(transport)
    client = DemoService.Client(protocol)

    # 連接服務端
    transport.open()

    recv = client.ping("test")
    print(recv)

    recv = client.get_int_string_mapping_result(10, "MyThrift")
    print(recv)

    recv = client.get_bool_result()
    print(recv)

    # 斷連服務端
    transport.close()

編寫完成后，整個項目結構如下圖所示：

測試服務端

運行服務端server.py后，運行客戶端client.py，打印的內容如下：

echo:test
{10: 'MyThrift'}
True

此時客戶端能夠正常調用服務端所提供的接口。（PS：在調試過程中，也許需要修改gen-py文件夾中Thrift編譯器生成的python代碼）

關于“python怎么實現Thrift服務端”這篇文章就分享到這里了，希望以上內容可以對大家有一定的幫助，使各位可以學到更多知識，如果覺得文章不錯，請把它分享出去讓更多的人看到。