上篇介绍了使用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)
}
}
服务端开启方法主要包含三个步骤:
- 开启监听端口;
- 调用
grpc.NewServer
创建grpc服务; - 调用生成的包注册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())
}
客户端使用方法也分为三个步骤:
- 调用
grpc.NewClient
创建grpc连接(这里有一点需要注意,grpc强制需要鉴权配置,否则会报错。但是目前没有做鉴权,传入的是grpc.WithTransportCredentials(insecure.NewCredentials())
,后续会介绍生产环境中鉴权如何配置); - 调用生成的包创建rpc客户端;
- 调用具体的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 包下面的,在运行配置那里需要把这三个文件都加上去,如下所示:
最终运行效果如下: