如何使用Golang实现RPC调用_使用gRPC生成客户端与服务端代码

Go实现gRPC调用需四步:1.安装protoc、protoc-gen-go和protoc-gen-go-grpc;2.编写.proto文件并生成.pb.go和_grpc.pb.go;3.服务端实现UnimplementedGreeterServer并注册;4.客户端通过grpc.Dial连接并调用。

用 Go 实现 gRPC 调用,核心是定义 Protocol Buffer 接口、用 protoc 生成 Go 代码、再分别编写服务端和客户端逻辑。整个过程不复杂但容易忽略细节,比如插件安装、导入路径、TLS 配置等。

1. 安装必要工具

要生成 Go 代码,需提前装好:

  • Protocol Buffers 编译器(protoc):从 GitHub releases 下载对应系统的二进制,加入 $PATH
  • Go 的 protoc 插件:protoc-gen-go
    go install google.golang.org/protobuf/cmd/protoc-gen-go@latest
  • gRPC 插件:protoc-gen-go-grpc
    go install google.golang.org/grpc/cmd/protoc-gen-go-grpc@latest

确认安装成功:运行 protoc --versionprotoc-gen-go --version(部分版本可能无响应,只要不报错且能生成代码即可)。

2. 编写 .proto 文件并生成 Go 代码

在项目根目录创建 api/hello.proto

syntax = "proto3";

package api;

option go_package = "./api;api";

service Greeter {
  rpc SayHello (HelloRequest) returns (HelloReply) {}
}

message HelloRequest {
  string name = 1;
}

message HelloReply {
  string message = 1;
}

执行命令生成代码(注意当前目录和输出路径):

protoc --go_out=. --go-grpc_out=. --go-grpc_opt=paths=source_relative api/hello.proto

会生成两个文件:api/hello.pb.go(数据结构 + 序列化)和 api/hello_grpc.pb.go(客户端接口 + 服务端抽象)。

3. 实现 gRPC 服务端

新建 server/main.go

package main

import (
	"context"
	"log"
	"net"

	pb "your-module-name/api" // 替换为你的模块路径
	"google.golang.org/grpc"
)

type server struct {
	pb.UnimplementedGreeterServer
}

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

func main() {
	lis, err := net.Listen("tcp", ":50051")
	if err != nil {
		log.Fatal(err)
	}
	defer lis.Close()

	grpcServer := grpc.NewServer()
	pb.RegisterGreeterServer(grpcServer, &server{})

	log.Println("gRPC server listening on :50051")
	if err := grpcServer.Serve(lis); err != nil {
		log.Fatal(err)
	}
}

关键点:

  • 实现接口时嵌入 UnimplementedXXXServer,避免未来新增方法导致编译失败
  • RegisterGreeterServer 是生成代码里提供的注册函数,不能手写
  • 默认不启用 TLS;如需加密,用 grpc.Creds(credentials.NewServerTLSFromCert(...))

4. 实现 gRPC 客户端

新建 client/main.go

package main

import (
	"context"
	"log"
	"time"

	pb "your-module-name/api"
	"google.golang.org/grpc"
	"google.golang.org/grpc/credentials/insecure"
)

func main() {
	// 连接服务端(禁用 TLS 时用 insecure)
	conn, err := grpc.Dial("localhost:50051", grpc.WithTransportCredentials(insecure.NewCredentials()))
	if err != nil {
		log.Fatal(err)
	}
	defer conn.Close()

	client := pb.NewGreeterClient(conn)

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

	rsp, err := client.SayHello(ctx, &pb.HelloRequest{Name: "World"})
	if err != nil {
		log.Fatal(err)
	}

	log.Printf("Reply: %s", rsp.Message)
}

注意:

  • grpc.Dial 默认不阻塞,加 grpc.WithBlock() 可等待连接就绪(开发调试可用,生产慎用)
  • 使用 insecure.NewCredentials() 表示跳过 TLS 校验;正式环境应配置证书
  • 每次调用建议带 context 控制超时和取消,避免永久阻塞

基本上就这些。跑起来只需 go run server/main.go 和另一个终端 go run client/main.go。后续可扩展中间件(拦截器)、流式 RPC、认证鉴权等。