【grpc】grpc入门二，基本使用方法

上篇介绍了使用protobuf生成grpc相关代码，并实现了服务端方法，本篇介绍下具体的使用。

一、服务端

// server.go

package main

import (
	"google.golang.org/grpc"
	"grpcDemo/calc"
	"net"
)

func startServer(addr string) {
	// 监听端口
	ls, err := net.Listen("tcp", addr)
	if err != nil {
		panic(err)
	}
	defer ls.Close()
	
	// 创建grpc服务
	gServer := grpc.NewServer()
	
	// 注册rpc
	calc.RegisterCalcServer(gServer, &calc.Server{})
	if err = gServer.Serve(ls); err != nil {
		panic(err)
	}
}

服务端开启方法主要包含三个步骤：

1. 开启监听端口；
2. 调用grpc.NewServer创建grpc服务；
3. 调用生成的包注册rpc服务。

二、客户端

// client.go
package main

import (
	"context"
	"fmt"
	"google.golang.org/grpc"
	"google.golang.org/grpc/credentials/insecure"
	"grpcDemo/calc"
)

func clientTest(addr string) {
	// 创建rpc连接
	cc, err := grpc.NewClient(addr,
		grpc.WithTransportCredentials(insecure.NewCredentials()))
	if err != nil {
		panic(err)
	}
	defer cc.Close()
	
	// 创建rpc调用客户端
	cli := calc.NewCalcClient(cc)
	
	// 调用具体的rpc方法
	versionRsp, err := cli.Version(context.Background(), &calc.Empty{})
	if err != nil {
		panic(err)
	}
	fmt.Println("server version:", versionRsp.GetStr())

	sumRsp, err := cli.Sum(context.Background(), &calc.CalcRequest{
		A: 1,
		B: 2,
	})
	fmt.Println("1+2=", sumRsp.GetNum())
}

客户端使用方法也分为三个步骤:

1. 调用grpc.NewClient创建grpc连接(这里有一点需要注意，grpc强制需要鉴权配置，否则会报错。但是目前没有做认证，传入的是grpc.WithTransportCredentials(insecure.NewCredentials())，后续会介绍生产环境中认证如何配置);
2. 调用生成的包创建rpc客户端；
3. 调用具体的rpc方法。

三、运行结果
目前代码结构如下

// main.go
package main

import (
	"time"
)

func main() {
	go startServer(":8080")
	// 确保服务开起来
	time.Sleep(time.Second)
	clientTest("localhost:8080")
}

这里有个需要注意的点，因为client.go、main.go、server.go 都是在 main 包下面的，在运行配置那里需要把这三个文件都加上去，如下所示:

最终运行效果如下: