快捷導(dǎo)航

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

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

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

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

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

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

前言

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

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

1 創(chuàng)建 thrift 文件添加兩個服務(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

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

3 編寫 server.py

# 服務(wù)類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ù)類2 HelloHandler
class HelloHandler:
	def hello(self, name):
		return 'hello'+name

4 編寫服務(wù)端運(yùn)行代碼開啟服務(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()
用來注冊兩個服務(wù)類
processor.registerProcessor(‘name', procress對象)
name 屬性將會在client 時(shí)用到

5運(yùn)行 runserver.py

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

6 編寫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__':
	# 啟動 服務(wù)
	transport = TSocket.TSocket('127.0.0.1', 8000)
	transport = TTransport.TBufferedTransport(transport)
	protocol = TBinaryProtocol.TBinaryProtocol(transport)

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

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

	transport.open() # 打開鏈接
	
	# 測試服務(wù)1
	cmd = 1
	token = '1111-2222-3333-4444'
	data = "kong_ge"
	msg = transmit_client.invoke(cmd, token, data)
	print(msg)
	
	# 測試服務(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 類和 TMultiplexedProtocol
但是網(wǎng)上關(guān)于 thrift python的實(shí)例太少了導(dǎo)致浪費(fèi)了很長時(shí)間
通過這篇文章的學(xué)習(xí)很快的明白thrift 中的一些概念

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

您可能感興趣的文章: