利用go-zero實現分散式跨語言的RPC調用

隨著業務規模的成長，單體應用的存在已經無法滿足系統的需求，分散式架構逐漸成為了主流。而在分散式系統中，RPC成為了不可或缺的一部分，它提供了一種方便、高效、可靠的遠端呼叫服務方式，使得各個服務之間能夠快速、穩定地進行資料互動和呼叫。

而對於跨語言的RPC呼叫來說，要求通訊協定和序列化協定都需要相容於多種程式語言，因而實現起來相對困難。本文將介紹如何利用go-zero框架實現跨語言的分散式RPC調用，旨在為讀者提供實用的解決方案。

1. go-zero框架簡介

go-zero是一個輕量級的Web框架，它採用了go語言原生的net/http模組，提供一套簡潔、易用、高效能的API開發方式，可以方便地將HTTP服務與微服務結合。 go-zero可以幫助我們快速建立分散式的、高並發的服務端應用程序，可以免費地在GitHub上取得程式碼和文件。

2. 實作跨語言的RPC呼叫

2.1 定義服務

我們在go-zero中定義服務時需要先寫一個proto文件，定義服務端和客戶端之間的通訊介面。假設我們定義了一個名叫Example的服務，它包含兩個方法：

syntax = "proto3";

package rpc;

service Example {
    rpc SayHello (Request) returns (Response);
    rpc GetUser (UserRequest) returns (UserResponse);
}

message Request {
    string name = 1;
}

message Response {
    string message = 1;
}

message UserRequest {
    string id = 1;
}

message UserResponse {
    string name = 1;
    string email = 2;
}

定義完proto文件後，我們需要使用protobuf編譯器將其編譯成go語言的源文件，執行以下命令：

protoc --go_out=. --go-grpc_out=. rpc.proto

這將產生rpc.pb.go和rpc_grpc.pb.go兩個檔案。

2.2 實作服務端

在go-zero框架中，我們可以使用go-grpc模組來實作grpc服務。在實作服務端時，需要實作proto檔案中定義的接口，並且使用go-zero提供的server.NewServer並呼叫AddService方法新增服務，然後在Init方法中啟動grpc服務。

package server

import (
    "context"
    "rpc"

    "github.com/tal-tech/go-zero/core/logx"
    "github.com/tal-tech/go-zero/core/stores/sqlx"
    "github.com/tal-tech/go-zero/core/syncx"
    "github.com/tal-tech/go-zero/zrpc"
    "google.golang.org/grpc"
)

type ExampleContext struct {
    Logx      logx.Logger
    SqlConn   sqlx.SqlConn
    CacheConn syncx.SharedCalls
}

type ExampleServer struct {
    Example rpc.ExampleServer
}

func NewExampleServer(ctx ExampleContext) *ExampleServer {
    return &ExampleServer{
        Example: &exampleService{
            ctx: ctx,
        },
    }
}

func (s *ExampleServer) Init() {
    server := zrpc.MustNewServer(zrpc.RpcServerConf{
        BindAddress: "localhost:7777",
    })
    rpc.RegisterExampleServer(server, s.Example)
    server.Start()
}

type exampleService struct {
    ctx ExampleContext
}

func (s *exampleService) SayHello(ctx context.Context, req *rpc.Request) (*rpc.Response, error) {
    return &rpc.Response{
        Message: "Hello, " + req.Name,
    }, nil
}

func (s *exampleService) GetUser(ctx context.Context, req *rpc.UserRequest) (*rpc.UserResponse, error) {
    // 查询数据库
    return &rpc.UserResponse{
        Name:  "name",
        Email: "email",
    }, nil
}

在伺服器上，我們可以使用Init方法來啟動RPC伺服器，使用MustNewServer來建立RPC伺服器，必須傳入一個RpcServerConf結構，包含我們要綁定的位址。

2.3 實作客戶端

在go-zero框架中，我們可以使用zrpc模組來實作grpc客戶端。使用zrpc.Dial建立連線並實例化rpc客戶端。

package client

import (
    "context"
    "rpc"

    "google.golang.org/grpc"
)

type ExampleClient struct {
    client rpc.ExampleClient
}

func NewExampleClient(conn *grpc.ClientConn) *ExampleClient {
    return &ExampleClient{
        client: rpc.NewExampleClient(conn),
    }
}

func (c *ExampleClient) SayHello(name string) (string, error) {
    resp, err := c.client.SayHello(context.Background(), &rpc.Request{
        Name: name,
    })
    if err != nil {
        return "", err
    }
    return resp.Message, nil
}

func (c *ExampleClient) GetUser(id string) (*rpc.UserResponse, error) {
    return c.client.GetUser(context.Background(), &rpc.UserRequest{
        Id: id,
    })
}

在客戶端上，我們只需要使用NewExampleClient函數建立RPC客戶端即可。 SayHello方法的作用是在服務端上取得一個回應，並將其傳回。 GetUser方法向伺服器取得使用者資訊的回應，並將其以UserResponse的形式傳回。

2.4 測試

現在，我們已經實作了服務端和客戶端的程式碼，我們可以透過以下程式碼來測試：

package main

import (
    "fmt"
    "log"
    "rpc_example/client"
    "rpc_example/server"

    "google.golang.org/grpc"
)

func main() {
    ctx := server.ExampleContext{}
    conn, err := grpc.Dial("localhost:7777", grpc.WithInsecure())
    if err != nil {
        log.Fatalf("grpc.Dial err :%v", err)
    }

    defer conn.Close()

    client := client.NewExampleClient(conn)
    resp, err := client.SayHello("Alice")
    if err != nil {
        log.Fatalf("client.SayHello err : %v", err)
    }

    fmt.Println(resp)

    user, err := client.GetUser("123")
    if err != nil {
        log.Fatalf("client.GetUser err : %v", err)
    }

    fmt.Println(user)
}

在上面的程式碼中，我們創建了一個grpc連線並呼叫SayHello和GetUser方法來測試我們的RPC服務。我們可以順利地回應了正確的數據，確定了RPC服務已經正常運作了。

3. 總結

在本文中，我們介紹如何使用go-zero框架實現分散式跨語言的RPC調用，其中涉及go-zero的Def模組、grpc、protobuf和zrpc等技術。在實現RPC服務時我們先定義RPC接口，然後根據接口編寫服務端和客戶端程式碼。最後加入Init方法以啟動RPC服務。如果您正在尋找一個輕量、易用的分散式系統框架，go-zero絕對是不錯的選擇。

以上是利用go-zero實現分散式跨語言的RPC調用的詳細內容。更多資訊請關注PHP中文網其他相關文章！