python thrift 實(shí)現(xiàn) 單端口多服務(wù)的過(guò)程

更新時(shí)間：2020年06月08日 10:44:56 作者：bug開(kāi)發(fā)工程師.

這篇文章主要介紹了python thrift 實(shí)現(xiàn) 單端口多服務(wù)的過(guò)程，本文通過(guò)實(shí)例代碼給大家介紹的非常詳細(xì)，對(duì)大家的學(xué)習(xí)或工作具有一定的參考借鑒價(jià)值,需要的朋友可以參考下

Thrift 是一種接口描述語(yǔ)言和二進(jìn)制通信協(xié)議。以前也沒(méi)接觸過(guò)，最近有個(gè)項(xiàng)目需要建立自動(dòng)化測(cè)試，這個(gè)項(xiàng)目之間的微服務(wù)都是通過(guò) Thrift 進(jìn)行通信的，然后寫(xiě)自動(dòng)化腳本之前研究了一下。

　　需要定義一個(gè)xxx.thrift的文件，來(lái)生成各種語(yǔ)言的代碼，生成之后我們的服務(wù)提供者和消費(fèi)者，都需要把代碼引入，服務(wù)端把代碼實(shí)現(xiàn)，消費(fèi)者直接使用API的存根，直接調(diào)用。

　　和 http 相比，同屬于應(yīng)用層，走 tcp 協(xié)議。Thrift 優(yōu)勢(shì)在于發(fā)送同樣的數(shù)據(jù)，request包和 response包要比 http 小很多，在整體性能上要優(yōu)于 http 。

前言

學(xué)習(xí)了兩天thrift 一直想實(shí)現(xiàn)單端口多服務(wù) 但是苦于網(wǎng)上的 thrift 實(shí)在太少而且大部分都是java實(shí)現(xiàn)的最后改了一個(gè)java的實(shí)現(xiàn)了單端口多服務(wù)

實(shí)現(xiàn)過(guò)程

1 創(chuàng)建 thrift 文件添加兩個(gè)服務(wù) Transmit Hello_test

service Transmit {
string invoke(1:i32 cmd 2:string token 3:string data)
}

service Hello_test {
string hello(1: string name)
}

2 運(yùn)行 thrift.exe -out gen-py --gen py test.thrift

生成對(duì)應(yīng)接口因?yàn)槲业?服務(wù)端和用戶端都是用 python寫(xiě)的所以只需要生成python 接口即可

3 編寫(xiě) server.py

# 服務(wù)類(lèi)1 TransmitHandler
class TransmitHandler:
 def __init__(self):
  self.log = {}

 def invoke(self, cmd, token, data):
  cmd = cmd
  token = token
  data = data
  if cmd == 1:
	  return data + 'and' + token
  else:
   return 'cmd不匹配'

# 服務(wù)類(lèi)2 HelloHandler
class HelloHandler:
	def hello(self, name):
		return 'hello'+name

4 編寫(xiě)服務(wù)端運(yùn)行代碼開(kāi)啟服務(wù)端

from test import Transmit
from test import Hello_test

from thrift.transport import TSocket
from thrift.transport import TTransport
from thrift.protocol import TBinaryProtocol
from thrift.server import TServer
# 導(dǎo)入
from thrift.TMultiplexedProcessor import TMultiplexedProcessor
from TransmitHandler_server import TransmitHandler
from Hello_server import HelloHandler


# open server
if __name__ == "__main__":
 # 實(shí)現(xiàn) 單端口 多服務(wù) 的方法

 transmit_handler = TransmitHandler()
 transmit_processor = Transmit.Processor(transmit_handler)

 hello_handler = HelloHandler()
 hello_processor = Hello_test.Processor(hello_handler)

 transport = TSocket.TServerSocket('127.0.0.1', 8000)
 tfactory = TTransport.TBufferedTransportFactory()
 pfactory = TBinaryProtocol.TBinaryProtocolFactory()
 # 多 processor
 processor = TMultiplexedProcessor()
 processor.registerProcessor('transmit', transmit_processor)
 processor.registerProcessor('hello', hello_processor)

 server = TServer.TSimpleServer(processor, transport, tfactory, pfactory)
 print("Starting python server...")
 server.serve()

值得注意的是要想實(shí)現(xiàn)單端口多服務(wù) 就必須得
引入processor = TMultiplexedProcessor()
用來(lái)注冊(cè)兩個(gè)服務(wù)類(lèi)
processor.registerProcessor(‘name', procress對(duì)象)
name 屬性將會(huì)在client 時(shí)用到

5運(yùn)行 runserver.py

如果出現(xiàn)Starting python server… 則運(yùn)行成功

6 編寫(xiě)client.py

from test import Transmit
from test import Hello_test
from thrift.transport import TSocket
from thrift.transport import TTransport
from thrift.protocol import TBinaryProtocol
from thrift.protocol.TMultiplexedProtocol import TMultiplexedProtocol


if __name__ == '__main__':
	# 啟動(dòng) 服務(wù)
	transport = TSocket.TSocket('127.0.0.1', 8000)
	transport = TTransport.TBufferedTransport(transport)
	protocol = TBinaryProtocol.TBinaryProtocol(transport)

	# 注冊(cè)兩個(gè)protocol 如果想要實(shí)現(xiàn)單端口 多服務(wù) 就必須使用 TMultiplexedProtocol
	transmit_protocol = TMultiplexedProtocol(protocol, 'transmit')
	hello_protocol = TMultiplexedProtocol(protocol, 'hello')

	# 注冊(cè)兩個(gè)客戶端
	transmit_client = Transmit.Client(transmit_protocol)
	hello_client = Hello_test.Client(hello_protocol)

	transport.open() # 打開(kāi)鏈接
	
	# 測(cè)試服務(wù)1
	cmd = 1
	token = '1111-2222-3333-4444'
	data = "kong_ge"
	msg = transmit_client.invoke(cmd, token, data)
	print(msg)
	
	# 測(cè)試服務(wù)2
	name = '孔格'
	msg2 = hello_client.hello(name)
	print(msg2)
	
	# 關(guān)閉
	transport.close()

7運(yùn)行client

觀察結(jié)果實(shí)現(xiàn)單端口多服務(wù)

總結(jié)

核心就是 TMultiplexedProcessor 類(lèi) 和 TMultiplexedProtocol
但是網(wǎng)上關(guān)于 thrift python的實(shí)例太少了導(dǎo)致浪費(fèi)了很長(zhǎng)時(shí)間
通過(guò)這篇文章的學(xué)習(xí)很快的明白thrift 中的一些概念

到此這篇關(guān)于python thrift 實(shí)現(xiàn) 單端口多服務(wù)的過(guò)程的文章就介紹到這了,更多相關(guān)python thrift單端口多服務(wù)內(nèi)容請(qǐng)搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！

您可能感興趣的文章: