上篇介绍了使用protobuf生成grpc相关代码,并实现了服务端方法,本篇介绍下具体的使用。
一、服务端
// server.gopackage mainimport ("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()// 注册rpccalc.RegisterCalcServer(gServer, &calc.Server{})if err = gServer.Serve(ls); err != nil {panic(err)}
}
服务端开启方法主要包含三个步骤:
- 开启监听端口;
- 调用
grpc.NewServer
创建grpc服务; - 调用生成的包注册rpc服务。
二、客户端
// client.go
package mainimport ("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 mainimport ("time"
)func main() {go startServer(":8080")// 确保服务开起来time.Sleep(time.Second)clientTest("localhost:8080")
}
这里有个需要注意的点,因为
client.go
、main.go
、server.go
都是在 main 包下面的,在运行配置那里需要把这三个文件都加上去,如下所示:
最终运行效果如下: