使用Thrift實現(xiàn)跨語言RPC的調用

更新時間：2023年10月07日 10:02:19 作者：愛音斯坦牛

Thrift最大的優(yōu)勢就是可以實現(xiàn)跨語言RPC調用,尤其在一些大廠,微服務各模塊之間使用不同的語言是很常見的,本文就將使用java作為服務端,用python作為客戶端,實現(xiàn)不同語言之間的RPC調用,需要的可以參考下

前言

前面我們在Thrift入門里面實現(xiàn)了Thrift實現(xiàn)RPC調用的簡單案例，而Thrift最大的優(yōu)勢就是可以實現(xiàn)跨語言RPC調用，尤其在一些大廠，微服務各模塊之間使用不同的語言是很常見的，有用java的，有go的，有python的，因此，選用Thrift實現(xiàn)RPC遠程調用是很不錯的選擇，本節(jié)將在RPC 框架之Thrift入門（一）的案例基礎上，使用java作為服務端，用python作為客戶端，實現(xiàn)不同語言之間的RPC調用！

IDL代碼

namespace java com.aniu.service
namespace py thrift_demo
struct Person {  // 定義 Person 結構體
    1: required string name;   // 姓名，必選字段
    2: required i32 age;       // 年齡，必選字段
    3: optional string sex;    // 性別，可選字段
}
service PersonService {  // 定義 PersonService 服務接口
    Person getByName(1: string name);  // 根據(jù)姓名獲取 Person 信息
    bool save(1: Person person);       // 保存 Person 信息
}

分別使用以下代碼生成java和python的代碼

thrift --gen java person.thrift

thrift --gen py person.thrift

代碼

服務端與業(yè)務邏輯

服務端代碼和業(yè)務邏輯就是RPC 框架之Thrift入門（一）的案例里面的

package com.aniu.server;
import org.apache.thrift.protocol.TBinaryProtocol;
import org.apache.thrift.server.TServer;
import org.apache.thrift.server.TSimpleServer;
import org.apache.thrift.transport.TServerSocket;
import com.aniu.service.PersonService;
import com.aniu.service.impl.PersonServiceImpl;
public class Server {
    public static void main(String[] args) {
        try{
            // 創(chuàng)建一個新的 Thrift 服務端套接字，監(jiān)聽在端口 9000 上
            TServerSocket socket = new TServerSocket(9000);
            // 創(chuàng)建一個 PersonService 的 Processor。Processor 是 Thrift 中用于處理請求的接口，它需要一個實現(xiàn)了 PersonService 接口的對象作為參數(shù)。
            PersonService.Processor<PersonServiceImpl> processor = new PersonService.Processor<>(new PersonServiceImpl());
            // 創(chuàng)建一個二進制協(xié)議工廠對象。Thrift 支持多種協(xié)議，如 TBinaryProtocol、TCompactProtocol、TJSONProtocol 等，這里選擇的是二進制協(xié)議。
            TBinaryProtocol.Factory factory = new TBinaryProtocol.Factory();
            // 創(chuàng)建一個 TSimpleServer 的參數(shù)對象 args1，并將之前創(chuàng)建的套接字、Processor 和協(xié)議工廠設置為其屬性。
            TServer.Args args1 = new TSimpleServer.Args(socket);
            args1.processor(processor);
            args1.protocolFactory(factory);
            // 使用之前設置好的參數(shù)創(chuàng)建 TSimpleServer 對象
            TSimpleServer tSimpleServer = new TSimpleServer(args1);
            // 開始執(zhí)行 TSimpleServer，開始監(jiān)聽并處理客戶端的請求。
            tSimpleServer.serve();
        }catch (Exception e){
            System.out.println(e);
        }
    }
}

package com.aniu.service.impl;
import com.aniu.service.Person;
import com.aniu.service.PersonService;
import org.apache.thrift.TException;
public class PersonServiceImpl implements PersonService.Iface {
    @Override
    public Person getByName(String name) throws TException {
        return new Person(name,18);
    }
    @Override
    public boolean save(Person person) throws TException {
        return false;
    }
}

客戶端

接下來我們就用python編寫Thrift客戶端來實現(xiàn)RPC調用 java服務端首先安裝thrift包：

pip install thrift

將生成的python代碼拷貝到項目，然后編寫python客戶端！

from thrift.transport import TSocket
from thrift.transport import TTransport
from thrift.protocol import TBinaryProtocol
from thrift_demo import PersonService
# 創(chuàng)建Thrift服務端的地址和端口
host = 'localhost'
port = 9000
# 創(chuàng)建Thrift傳輸層和協(xié)議層
t_socket = TSocket.TSocket(host, port)
transport = TTransport.TBufferedTransport(t_socket)
protocol = TBinaryProtocol.TBinaryProtocol(transport)
# 創(chuàng)建Thrift客戶端
client = PersonService.Client(protocol)
# 打開Thrift傳輸層連接
transport.open()
# 調用Thrift客戶端提供的接口
try:
    # 調用getName方法
    person = client.getByName("aniu")
    print(person)
except Exception as e:
    print('Error:', str(e))
# 關閉Thrift傳輸層連接
transport.close()

首先啟動java服務端，然后啟動python客戶端，可以看到調用成功！