go语言 grpc简单示例

原创

mob64ca12daebd0 2024-08-31 09:47:02 ©著作权

©著作权归作者所有：来自51CTO博客作者mob64ca12daebd0的原创作品，请联系作者获取转载授权，否则将追究法律责任

Go语言 gRPC 简单示例

在现代应用程序中，gRPC是一种高效的远程过程调用框架，非常适合微服务架构。本文将指导你实现一个使用Go语言的简单gRPC示例，包括服务端和客户端的代码。

整体流程

以下是实现gRPC的步骤概述：

步骤	描述
1	安装Go和gRPC相关依赖
2	定义.proto文件
3	生成gRPC代码
4	实现服务端
5	实现客户端
6	启动服务和发送请求

gantt
    title gRPC 示例开发计划
    dateFormat  YYYY-MM-DD
    section 准备阶段
    安装Go和gRPC相关依赖      :a1, 2023-10-01, 1d
    section 开发阶段
    定义.proto文件            :a2, 2023-10-02, 1d
    生成gRPC代码              :a3, 2023-10-03, 1d
    实现服务端                :a4, 2023-10-04, 2d
    实现客户端                :a5, 2023-10-06, 2d
    启动服务和发送请求        :a6, 2023-10-08, 1d

步骤详解

1. 安装Go和gRPC相关依赖

首先，请确保你已经安装好Go语言环境。然后，可以使用以下命令安装gRPC：

go get -u google.golang.org/grpc
go get -u google.golang.org/protobuf/cmd/protoc-gen-go
go get -u google.golang.org/grpc/cmd/protoc-gen-go-grpc

这一步会将gRPC和protobuf的命令行工具安装到你的GOPATH中。

2. 定义.proto文件

创建一个新的目录，并在其中创建hello.proto文件，内容如下：

syntax = "proto3";

package hello;

service Greeter {
    rpc SayHello(HelloRequest) returns (HelloResponse);
}

message HelloRequest {
    string name = 1;
}

message HelloResponse {
    string message = 1;
}

这段代码定义了一个简单的gRPC服务Greeter，它有一个SayHello方法。

3. 生成gRPC代码

使用protoc工具生成Go代码：

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

这将生成两个文件：hello.pb.go和hello_grpc.pb.go。

4. 实现服务端

在同一目录下创建server.go文件，内容如下：

package main

import (
    "context"
    "log"
    "net"

    pb "path/to/your/hello"

    "google.golang.org/grpc"
)

type server struct {
    pb.UnimplementedGreeterServer
}

func (s *server) SayHello(ctx context.Context, req *pb.HelloRequest) (*pb.HelloResponse, error) {
    return &pb.HelloResponse{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.RegisterGreeterServer(s, &server{})
    log.Println("Server is running at :50051")
    if err := s.Serve(lis); err != nil {
        log.Fatalf("failed to serve: %v", err)
    }
}

这段代码创建了一个gRPC服务并监听在50051端口。

5. 实现客户端

创建client.go文件，内容如下：

package main

import (
    "context"
    "log"
    "time"
    
    pb "path/to/your/hello"
    
    "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.NewGreeterClient(conn)
    
    ctx, cancel := context.WithTimeout(context.Background(), time.Second)
    defer cancel()
    
    resp, err := client.SayHello(ctx, &pb.HelloRequest{Name: "World"})
    if err != nil {
        log.Fatalf("could not greet: %v", err)
    }
    log.Println(resp.Message)
}

客户端会连接到服务端，发送请求并打印返回的消息。