grpc搭配protobuf使用例子-Go服务器开发

下面是一个使用 gRPC 和 Protocol Buffers（protobuf）的基本示例，包括定义服务、生成代码以及实现服务器和客户端。

1. 安装必要的工具

确保你已安装以下工具：

Go（或其他语言，gRPC 支持多种语言）
Protobuf 编译器 (protoc)
gRPC 的相关库

对于 Go 用户，你可以通过以下命令安装 gRPC 和 protobuf 库：

go get google.golang.org/grpc
go get google.golang.org/protobuf

2. 定义 Proto 文件

首先，创建一个名为 example.proto 的文件，用于定义我们的服务和消息格式：

syntax = "proto3";

package example;

// 定义请求消息
message SayHelloRequest {
    string name = 1;
}

// 定义响应消息
message SayHelloResponse {
    string message = 1;
}

// 定义 HelloService 服务
service HelloService {
    rpc SayHello(SayHelloRequest) returns (SayHelloResponse);
}

3. 编译 Proto 文件

使用 protoc 命令编译 proto 文件，以生成相应的代码。对于 Go，可以运行以下命令：

protoc --go_out=. --go-grpc_out=. example.proto

这将生成 example.pb.go 和 example_grpc.pb.go 文件。

4. 实现服务器端

接下来，创建一个名为 server.go 的文件，实现 gRPC 服务器：

package main

import (
    "context"
    "log"
    "net"

    pb "path/to/your/generated/proto/package" // 导入生成的包

    "google.golang.org/grpc"
)

type server struct {
    pb.UnimplementedHelloServiceServer // 必须嵌入未实现的服务器类型，以便自动实现该方法。
}

func (s *server) SayHello(ctx context.Context, req *pb.SayHelloRequest) (*pb.SayHelloResponse, error) {
    return &pb.SayHelloResponse{
        Message: "Hello " + req.Name,
    }, nil
}

func main() {
    lis, err := net.Listen("tcp", ":50051")
    if err != nil {
        log.Fatalf("failed to listen: %v", err)
    }

    s := grpc.NewServer()
    pb.RegisterHelloServiceServer(s, &server{})

    log.Println("Server is running at :50051")
    
    if err := s.Serve(lis); err != nil {
        log.Fatalf("failed to serve: %v", err)
    }
}

5. 实现客户端

然后创建一个名为 client.go 的文件来实现客户端：

package main

import (
   "context"
   "log"
   "time"

   pb "path/to/your/generated/proto/package" // 导入生成的包
   
   "google.golang.org/grpc"
)

func main() {
   conn, err := grpc.Dial("localhost:50051", grpc.WithInsecure())
   if err != nil {
       log.Fatalf("did not connect: %v", err)
   }
   defer conn.Close()

   client := pb.NewHelloServiceClient(conn)

   ctx, cancel := context.WithTimeout(context.Background(), time.Second)
   defer cancel()

   resp, err := client.SayHello(ctx, &pb.SayHelloRequest{Name: "World"})
   if err != nil {
       log.Fatalf("could not greet: %v", err)
   }

   log.Printf("Greeting: %s", resp.Message)
}