Go語言實現(xiàn)開發(fā)一個簡單的gRPC Demo

更新時間：2023年07月16日 11:08:02 作者：不背鍋運維

這篇文章主要為大家詳細介紹了如何利用Go語言實現(xiàn)開發(fā)一個簡單的gRPC Demo，文中的示例代碼講解詳細，感興趣的小伙伴可以跟隨小編一起了解一下

Part1先決條件

獲得并安裝好Golang（https://go.dev/doc/install）

Protocol buffer compiler（protobuf編譯器）, protoc version 3（protoc工具版本建議為3版本），獲得并安裝：https://github.com/protocolbuffers/protobuf/releases

安裝protocol編譯器的Go插件

go?install?google.golang.org/protobuf/cmd/protoc-gen-go@v1.28
go?install?google.golang.org/grpc/cmd/protoc-gen-go-grpc@v1.2

Part2開發(fā)環(huán)境準備

準備go環(huán)境

wget?https://go.dev/dl/go1.20.5.linux-amd64.tar.gz
tar?-zxf?go1.20.5.linux-amd64.tar.gz
mv?go?/usr/local/

安裝protobuf編譯器protoc

wget?https://github.com/protocolbuffers/protobuf/releases/download/v23.4/protoc-23.4-linux-x86_64.zip
mkdir?protoc-23.4
mv?protoc-23.4-linux-x86_64.zip
cd?protoc-23.4
unzip?protoc-23.4-linux-x86_64.zip
rm?-rf?protoc-23.4-linux-x86_64.zip
cd?..
mv?protoc-23.4?/usr/local/

添加相關(guān)環(huán)境變量

export?GOROOT="/usr/local/go"
export?GOPATH="/home/tantianran/goCode"
export?GOPROXY="https://goproxy.cn,direct"
export?GO111MODULE="on"
export?PATH=$PATH:$GOROOT/bin:$GOPATH:$GOPATH/bin
export?PROTOC_HOME=/usr/local/protoc-23.4
export?PATH=$PATH:$PROTOC_HOME/bin

Part3實戰(zhàn)：開發(fā)一個簡單的gRPC Demo

1.創(chuàng)建user-service模塊

tantianran@go-dev:~/goCode/src$?mkdir?user-service
tantianran@go-dev:~/goCode/src$?cd?user-service/
tantianran@go-dev:~/goCode/src/user-service$?go?mod?init

2.使用protobuf idl語言定義服務接口

創(chuàng)建service包

tantianran@go-dev:~/goCode/src/user-service$?mkdir?service
tantianran@go-dev:~/goCode/src/user-service$?cd?service/

service/users.proto內(nèi)容：

syntax = "proto3";
option go_package = "user-service/service";
// 服務和方法
service Users {
    rpc GetUser (UserGetRequest) returns (UserGetReply) {}
}
// 請求消息
message UserGetRequest {
    string email = 1;
    int32 id = 2;
}
// 響應消息
message User {
    string id = 1;
    string first_name = 2;
    string last_name = 3;
    int32 age = 4;
}
message UserGetReply {
    User user = 1; // 嵌套
}

3.生成客戶端和服務端代碼

首先安裝用于編譯器的go語言插件protoc-gen-go

go?install?google.golang.org/protobuf/cmd/protoc-gen-go@latest

開始生成

tantianran@go-dev:~/goCode/src/user-service/service$?protoc?--go_out=.?--go_opt=paths=source_relative?--go-grpc_out=.?--go-grpc_opt=paths=source_relative?users.proto
tantianran@go-dev:~/goCode/src/user-service/service$?ls?-l
total?24
-rw-rw-r--?1?tantianran?tantianran???37?Jul?11?10:05?go.mod
-rw-rw-r--?1?tantianran?tantianran?3387?Jul?11?10:10?users_grpc.pb.go
-rw-rw-r--?1?tantianran?tantianran?8984?Jul?11?10:10?users.pb.go
-rw-rw-r--?1?tantianran?tantianran??362?Jul?11?10:09?users.proto

4.編寫服務器

創(chuàng)建server包

tantianran@go-dev:~/goCode/src$?cd?user-service/
tantianran@go-dev:~/goCode/src/user-service$?mkdir?server
tantianran@go-dev:~/goCode/src/user-service$?cd?server/

server/server.go

package?main
import?(
?"context"
?"log"
?"net"
?"os"
?users?"user-service/service"?//?導入之前生成的包
?"google.golang.org/grpc"
)
//?userService類型是Users服務的服務處理程序
type?userService?struct?{
?users.UnimplementedUsersServer?//?這個字段對于gRPC中的任何服務實現(xiàn)都是強制性的
}
func?(s?*userService)?GetUser(ctx?context.Context,?in?*users.UserGetRequest)?(*users.UserGetReply,?error)?{
?//?打印客戶端傳過來的數(shù)據(jù)
?log.Printf("已接收到郵件地址:?%s,?還有ID:?%d",?in.Email,?in.Id)
?//?自定義數(shù)據(jù)響應給客戶端
?u?:=?users.User{
??Id:????????"user-782935",
??FirstName:?"tan",
??LastName:??"tianran",
??Age:???????30,
?}
?return?&users.UserGetReply{User:?&u},?nil
}
//?向gRPC服務器注冊Users服務
func?registerServices(s?*grpc.Server)?{
?users.RegisterUsersServer(s,?&userService{})
}
//?啟動gRPC服務器
func?startServer(s?*grpc.Server,?l?net.Listener)?error?{
?return?s.Serve(l)
}
func?main()?{
?listenAddr?:=?os.Getenv("LISTEN_ADDR")
?if?len(listenAddr)?==?0?{
??listenAddr?=?":50051"
?}
?lis,?err?:=?net.Listen("tcp",?listenAddr)
?if?err?!=?nil?{
??log.Fatal(err)
?}
?s?:=?grpc.NewServer()
?registerServices(s)
?log.Fatal(startServer(s,?lis))
}

5.編寫客戶端

tantianran@go-dev:~/goCode/src$?cd?user-service/
tantianran@go-dev:~/goCode/src/user-service$?mkdir?client/
tantianran@go-dev:~/goCode/src/user-service$?cd?client/

client/main.go

package?main
import?(
?"context"
?"log"
?"os"
?users?"user-service/service"?//?導入之前生成的包
?"google.golang.org/grpc"
)
//?建立與服務器的連接（通道）
func?setupGrpcConnection(addr?string)?(*grpc.ClientConn,?error)?{
?return?grpc.DialContext(
??context.Background(),
??addr,
??grpc.WithInsecure(),
??grpc.WithBlock(),
?)
}
//?創(chuàng)建客戶端與Users服務通信
func?getUserServiceClient(conn?*grpc.ClientConn)?users.UsersClient?{
?return?users.NewUsersClient(conn)
}
//?調(diào)用Users服務中的GetUser()方法
func?getUser(client?users.UsersClient,?u?*users.UserGetRequest)?(*users.UserGetReply,?error)?{
?return?client.GetUser(context.Background(),?u)
}
func?main()?{
?if?len(os.Args)?!=?2?{
??log.Fatal("缺少gRPC服務器地址")
?}
?conn,?err?:=?setupGrpcConnection(os.Args[1])
?if?err?!=?nil?{
??log.Fatal(err)
?}
?defer?conn.Close()
?c?:=?getUserServiceClient(conn)
?result,?err?:=?getUser(c,?&users.UserGetRequest{
??Email:?"tantianran@qq.com",
??Id:????801896,
?})
?if?err?!=?nil?{
??log.Fatal(err)
?}
?//?打印響應
?log.Printf("收到響應:?%s?%s?%s?%d\n",?result.User.Id,?result.User.FirstName,?result.User.LastName,?result.User.Age)
}