[grpc快速入门] 一 grpc生成与调用

下载通用编译器

地址：https://github.com/protocolbuffers/protobuf/releases
选择对应的版本，解压后将文件夹下bin目录配置到环境变量
安装go编译器

1. $ go install google.golang.org/protobuf/cmd/protoc-gen-go
2. $ go install google.golang.org/grpc/cmd/protoc-gen-go-grpc

复制代码

创建文件夹，创建两个项目

记得 go mod init 项目名
然后 go mod tidy
Demo.proto内容

1. // 这个就是protobuf的中间文件
3. // 指定的当前proto语法的版本，有2和3
4. syntax = "proto3";
5. //  输出路径
6. option go_package="../pdfile";
9. // 定义request model
10. message GrpcDemoRequest{
11.   int32 number1 = 1; // 1代表顺序
12.   int32 number2 = 2;
13. }
15. // 定义response model
16. message GrpcDemoResponse{
17.   int32 sum = 1; // 1代表顺序
18. }
20. // 定义服务主体
21. service ProdService{
22.   // 定义方法
23.   rpc GetSum(GrpcDemoRequest) returns(GrpcDemoResponse);
25. }

复制代码

这里我们实现GetSum的功能是,传过来的GrpcDemoRequest结构体内 n1+n2 然后通过GrpcDemoResponse返回。
proto文件定义了方法和方法的参数，方法的内容要通过编译后的文件以接口的方式来实现
编译文件,终端进入到pdfile文件夹下，运行命令

1. protoc --go_out=. --go-grpc_out=. Demo.proto

复制代码

会在pdfile下生成两个文件

生成后的go文件不要改动
如果导入包报红

鼠标悬浮点击Sync
Demo_grpc.pd.go
我们要实现的就是51 52 行的方法，下面红框是示例，到时候我们复制一下拿来用

在pdfile文件夹下创建Demo_imp.go

代码

1. package pdfile
3. import (
4.         "context"
5. )
7. type DemoServiceServer struct {
8. }
10. func (d *DemoServiceServer) GetSum(ctx context.Context,req *GrpcDemoRequest) (*GrpcDemoResponse, error) {
11.         resp := GrpcDemoResponse{
12.                 Sum:                  req.Number1+req.Number2,
13.         }
15.         return &resp,nil
16. }
18. func (d *DemoServiceServer) mustEmbedUnimplementedProdServiceServer() {}

复制代码

服务注册
回到我们srv下的main文件

1. package main
3. import (
4.         "google.golang.org/grpc"
5.         "net"
6.         "srv/pdfile"
7. )
9. func main() {
12.         server := grpc.NewServer()
13.         demoServer := &pdfile.DemoServiceServer{}
14.         pdfile.RegisterProdServiceServer(
15.                 server,
16.                 demoServer,
17.         )
19.         listenin ,err := net.Listen("tcp","0.0.0.0:8858")
20.         if err != nil {
21.                 panic(err)
22.         }
23.         _ = server.Serve(listenin)
24. }

复制代码

客户端调用
将pdfile文件夹拷贝到cli文件夹下

修改cli main文件

1. package main
3. import (
4.         "cli/pdfile"
5.         "context"
6.         "fmt"
7.         "google.golang.org/grpc"
8.         "google.golang.org/grpc/credentials/insecure"
9.         "log"
10. )
12. func main() {
13.         conn, err := grpc.Dial("127.0.0.1:8858",grpc.WithTransportCredentials(insecure.NewCredentials()))
14.         if err != nil {
15.                 log.Fatal(err)
16.         }
18.         demoServiceClient := pdfile.NewProdServiceClient(conn)
20.         // 直接像调用本地方法一样调用GetSum方法
21.         resp, err := demoServiceClient.GetSum(context.Background(), &pdfile.GrpcDemoRequest{
22.                 Number1:              10,
23.                 Number2:              5,
24.         })
25.         if err != nil {
26.                 log.Fatal("调用gRPC方法错误: ", err)
27.         }
28.         fmt.Println(resp)
29. }

复制代码

运行srv 然后运行 cli

完成
封装方便调用
新建文件

1. package grpc_conn
3. import (
4.         "cli/pdfile"
5.         "google.golang.org/grpc"
6.         "google.golang.org/grpc/credentials/insecure"
7.         "log"
8. )
10. var Grpc_conn pdfile.ProdServiceClient
12. func init() {
13.         conn, err := grpc.Dial("127.0.0.1:8858",grpc.WithTransportCredentials(insecure.NewCredentials()))
14.         if err != nil {
15.                 log.Fatal(err)
16.         }
17.         Grpc_conn = pdfile.NewProdServiceClient(conn)
18. }

复制代码

新建测试模块

1. package test
3. import (
4.         "cli/grpc_conn"
5.         "cli/pdfile"
6.         "context"
7.         "fmt"
8.         "log"
9.         "testing"
10. )
12. func TestDemo1(t *testing.T) {
13.         resp, err := grpc_conn.Grpc_conn.GetSum(context.Background(), &pdfile.GrpcDemoRequest{
14.                 Number1:              10,
15.                 Number2:              5,
16.         })
17.         if err != nil {
18.                 log.Fatal("调用gRPC方法错误: ", err)
19.         }
20.         fmt.Println(resp)
21. }

复制代码

运行test


免责声明：如果侵犯了您的权益，请联系站长，我们会及时删除侵权内容，谢谢合作！