gRPC使用

gRPC简介

grpc是google remote procedure call, 远程过程调用。多用于分布式服务。服务端定义服务的具体实现,客户端直接调用服务端的实现,得到对应的返回。gRPC以protocol buffer作为消息序列化工具,封装了客户端与服务器之间的消息通信,

服务定义

服务定义是直接写在proto文件中,如下所示:

// The greeting service definition.
service Greeter {
  // Sends a greeting
  rpc SayHello (HelloRequest) returns (HelloReply) {}
}

// The request message containing the user's name.
message HelloRequest {
  string name = 1;
}

// The response message containing the greetings
message HelloReply {
  string message = 1;
}

service定义服务名称,rpc定义远程函数调用名称。

将proto文件转化为go代码

protoc hello.proto --go_out=plugins=grpc:./

生成hellp.pb.go文件,可以被go程序调用。

编写服务器

实现rpc方法

type server struct{}
// SayHello implements helloworld.GreeterServer
func (s *server) SayHello(ctx context.Context, in *pb.HelloRequest) (*pb.HelloReply, error) {
    return &pb.HelloReply{Message: "Hello " + in.Name}, nil
}

监听端口&开启服务

func main() {
    lis, err := net.Listen("tcp", port)
    if err != nil {
        log.Fatalf("failed to listen: %v", err)
    }
    s := grpc.NewServer()
    pb.RegisterGreeterServer(s, &server{})
    // Register reflection service on gRPC server.
    reflection.Register(s)
    if err := s.Serve(lis); err != nil {
        log.Fatalf("failed to serve: %v", err)
    }
}
  1. 设置监听端口
  2. 注册服务
  3. 开始服务

编写客户端

func main() {
    // Set up a connection to the server.
    conn, err := grpc.Dial(address, grpc.WithInsecure())
    if err != nil {
        log.Fatalf("did not connect: %v", err)
    }
    defer conn.Close()

    c := pb.NewGreeterClient(conn)
    r, err := c.SayHello(context.Background(), &pb.HelloRequest{Name: "world"})
    if err != nil {
        log.Fatalf("could not greet: %v", err)
    }
    log.Printf("Greeting: %s", r.Message)
}
  1. 连接服务器端口
  2. 创建服务对应的客户端
  3. 调用服务的方法

相关链接

  1. What is gRPC?
  2. Go Quick Start
  3. Example