gRPC 学习之三:初试 GO 版 gRPC 开发,Java 面试题中高级
}
[](
)根据 proto 生成 go 源码
在 helloworld.proto 所在的目录,执行以下命令:
protoc --go_out=plugins=grpc:. helloworld.proto
如果 helloworld.proto 没有语法错误,会在当前目录生成文件 helloworld.pb.go,这里面是工具 protoc-gen-go 自动生成的代码,里面生成的代码在开发服务端和客户端时都会用到;
下面是 helloworld.pb.go 的代码片段,作用是服务注册,入参是 GreeterServer 是个接口,由此可以推测:在服务端,由具体的业务代码来实现 GreeterServer 接口,并且调用 RegisterGreeterServer 方法注册,这样客户端远程调用的服务就可以实现业务功能了:
func RegisterGreeterServer(s *grpc.Server, srv GreeterServer) {
s.RegisterService(&_Greeter_serviceDesc, srv)
}
type GreeterServer interface {
// 具体的远程服务方法
SayHello(context.Context, *HelloRequest) (*HelloReply, error)
}
借助 GoLand 的 Structure 面板,可以进一步观察 helloworld.pb.go:
[](
)编写服务端代码 server.go 并启动
在 $GOPATH/src/helloworld 目录下新建文件夹 server,在此文件夹下新建 server.go,内容如下,已经添加详细注释:
package main
import (
"context"
"log"
"net"
"google.golang.org/grpc"
pb "helloworld"
)
const (
port = ":50051"
)
// 定义结构体,在调用注册 api 的时候作为入参,
// 该结构体会带上 SayHello 方法,里面是业务代码
// 这样远程调用时就执行了业务代码了
type server struct {
// pb.go 中自动生成的,是个空结构体
pb.UnimplementedGreeterServer
}
// 业务代码在此写,客户端远程调用 SayHello 时,
// 会执行这里的代码
func (s *server) SayHello(ctx context.Context, in *p
b.HelloRequest) (*pb.HelloReply, error) {
// 打印请求参数
log.Printf("Received: %v", in.GetName())
// 实例化结构体 HelloReply,作为返回值
return &pb.HelloReply{Message: "Hello " + in.GetName()}, nil
}
func main() {
// 要监听的协议和端口
lis, err := net.Listen("tcp", port)
if err != nil {
log.Fatalf("failed to listen: %v", err)
}
// 实例化 gRPC server 结构体
s := grpc.NewServer()
// 服务注册
pb.RegisterGreeterServer(s, &server{})
log.Println("开始监听,等待远程调用...")
if err := s.Serve(lis); err != nil {
log.Fatalf("failed to serve: %v", err)
}
}
在 server.go 所在目录执行 go run server.go,控制台提示如下:
[golang@centos7 server]$ go run server.go
2020/12/13 08:20:32 开始监听,等待远程调用...
此时 gRPC 的服务端已经启动,可以响应远程调用,接下来开发客户端代码;
[](
)编写客户端代码 client.go 并启动
再打开一个控制台;
在 $GOPATH/src/helloworld 目录下新建文件夹 client,在此文件夹下新建 client.go,内容如下,已经添加详细注释:
package main
import (
"context"
"log"
"os"
"time"
"google.golang.org/grpc"
pb "helloworld"
)
const (
address = "localhost:50051"
defaultName = "world"
)
func main() {
// 远程连接服务端
conn, err := grpc.Dial(address, grpc.WithInsecure(), grpc.WithBlock())
if err != nil {
log.Fatalf("did not connect: %v", err)
}
// main 方法执行完毕后关闭远程连接
defer conn.Close()
// 实例化数据结构
c := pb.NewGreeterClient(conn)
// 远程调用的请求参数,如果没有从命令行传入,就用默认值
name := defaultName
if len(os.Args) > 1 {
name = os.Args[1]
}
// 超时设置
ctx, cancel := context.WithTimeout(context.Background(), time.Second)
defer cancel()
// 远程调用
r, err := c.SayHello(ctx, &pb.HelloRequest{Name: name})
if err != nil {
log.Fatalf("could not greet: %v", err)
}
// 将服务端的返回信息打印出来
log.Printf("Greeting: %s", r.GetMessage())
}
在 client.go 所在目录执行 go run client.go,会立即向服务端发起远程调用,控制台提示如下,可见得到了服务端的返回信息 Hello world:
[golang@centos7 client]$ go run client.go
2020/12/13 08:38:05 Greeting: Hello world
再去服务端的控制台看一下,通过日志发现业务代码被执行,收到了远程请求的参数:
[golang@centos7 server]$ go run server.go
2020/12/13 08:20:32 开始监听,等待远程调用...
评论