基於微服務架構go-micro開發gRPC應用程式

2022-07-04 14:02:37

go-micro是golang的一個微服務架構。這篇文章將介紹使用go-micro最新版本v4開發gRPC服務的方式。

1、安裝protoc

這個工具也稱為proto編譯器，可以用來生成各種開發語言使用proto協定的程式碼。

下載地址：https://github.com/protocolbuffers/protobuf/releases

一般下載最新版本就行，注意要符合自己當前的作業系統。

解壓后里邊有個 protoc.exe ，拷貝到 GOPATH 的 bin 目錄下，我這裡就是 C:/Users/PC-001/go/bin ，PC-001 是登入當前作業系統的使用者名稱，需要換成你自己的。GOPATH 可以通過執行 go env 檢視到。

一般都是把 GOPATH 的 bin 目錄新增到環境變數的 PATH 變數中，如果沒有，請自行新增上。

2、安裝protoc的go-micro外掛

需要安裝兩個外掛，它們用來生成 Golang 版本的 proto 協定程式碼和 go-micro 的 gRPC 代理程式碼。

執行如下命令，會在 GOPATH 的 bin 目錄下生成兩個可執行檔案：protoc-gen-go.exe 和 protoc-gen-micro.exe。

go install google.golang.org/protobuf/cmd/protoc-gen-go
go install go-micro.dev/v4/cmd/protoc-gen-micro@v4

3、編寫proto檔案

proto檔案是符合Protocol Buffers語言規範的資料交換協定檔案，就像以前WebService定義服務時使用的XML檔案。現在一般都是用proto3了，這裡建立一個名為 hello.proto 的檔案，放到專案的proto目錄下：

syntax = "proto3";

option go_package="/proto";

package Business;

service Hello {
  rpc Say (SayRequest) returns (SayResponse);
}

message SayResponse {
  string Message = 1;
}

message SayRequest {
  string Name = 1;
}

這個協定很簡單，有個名字為Hello的服務，提供一個名字為Say的rpc方法，這個方法有輸入輸出，輸入資訊中有一個名為Name的引數，輸出資訊中有一個名為Message的返回值。

4、生成gRPC代理程式碼

在專案根目錄中執行如下命令，會在proto目錄下生成兩個檔案：hello.pb.go 和 hello.pb.micro.go。

protoc --go_out=. --go_opt=paths=source_relative --micro_out=. --micro_opt=paths=source_relative proto/hello.proto

hello.pb.go 中是輸入輸出資料型別的Golang定義，hello.pb.micro.go 中是go-micro使用者端和伺服器端的gRPC代理程式。

也可以將檔案生成到別的目錄中，比如修改 proto 檔案中 go_package 的設定為：

option go_package="/business";

然後執行下面的命令，會生成到專案下的business目錄中：

protoc --go_out=.  --micro_out=. proto/hello.proto

具體這個命令還有哪些引數，可以執行 protoc -help 去探索下，這裡就不羅嗦了。

5、編寫gRPC伺服器端程式

這裡先上程式碼：

package main

import (
	"context"
	"fmt"
	"grpchello/proto"
	"log"
	"time"

	"github.com/asim/go-micro/plugins/server/grpc/v4"
	"go-micro.dev/v4"
	"go-micro.dev/v4/server"
)

type Hello struct{}

func (s *Hello) Say(ctx context.Context, req *proto.SayRequest, rsp *proto.SayResponse) error {
	fmt.Println("request:", req.Name)
	rsp.Message = "Hello " + req.Name
	return nil
}

func main() {

	grpcServer := grpc.NewServer()

	service := micro.NewService(
		micro.Server(grpcServer),
		micro.Name("grpchello.service"),
		micro.Address("0.0.0.0:8001"),
		micro.RegisterTTL(time.Second*30),
		micro.RegisterInterval(time.Second*10),
	)

	// optionally setup command line usage
	service.Init()

	// Register Handlers
	proto.RegisterHelloHandler(service.Server(), &Hello{})

	// Run server
	if err := service.Run(); err != nil {
		log.Fatal(err)
	}
}

整個過程如下：

先定義一個名為 Hello 的型別，它去實現hello.pb.micro.go中生成的HelloHandler介面，也就是編寫具體的業務實現。

然後通過grpc.NewServer建立一個gRPC伺服器，這裡決定go-micro底層通訊使用gPRC；

然後通過micro.NewService建立一個micro服務，這裡需要傳入上面建立的grpc server，還要設定服務的名稱、監聽埠，以及保持服務健康用的TTL和上報週期。go-mico當前預設的服務發現機制是mdns，不用額外部署服務發現工具，不過一般只建議在開發環境使用，生產環境建議使用etcd；

通過上面的步驟對服務進行一些初始化之後，再將Hello型別的範例註冊到grpc server；

最後啟動服務，就可以開始接客了。

如果直接貼上這段程式碼的話，會有很多包沒有參照進來，可以通過 go get 獲取這些包，也可以執行命令：go mod tidy 來引入需要的包。

這裡邊還import了一個 grpchello/proto ，這個是我這邊專案的mod是 grpchello，gRPC代理程式的package是proto，如果你的不一樣，需要修改下。

6、編寫gRPC使用者端程式

存取gRPC服務也是通過 hello.pb.micro.go 中生成的方法，直接上程式碼：

package main

import (
	"bufio"
	"context"
	"fmt"
	"grpchello/proto"
	"os"

	"github.com/asim/go-micro/plugins/client/grpc/v4"
	"go-micro.dev/v4"
)

func main() {

	service := micro.NewService(
		micro.Client(grpc.NewClient()),
	)

	service.Init()
	client := proto.NewHelloService("grpchello.service", service.Client())

	rsp, err := client.Say(context.TODO(), &proto.SayRequest{Name: "BOSSMA"})
	if err != nil {
		fmt.Println(err)
	}

	fmt.Println(rsp)

	fmt.Println("按確認鍵退出程式...")
	in := bufio.NewReader(os.Stdin)
	_, _, _ = in.ReadLine()
}

上面的程式使用了 grpc.NewClient , 這決定了底層通訊採用gRPC；然後再使用生成的程式碼 proto.NewHelloService 建立使用者端代理，這裡指定的服務名字必須是伺服器端註冊的名字；最後使用這個使用者端代理呼叫相應的方法。