Go服务端如何向客户端传达消息？

后端

2022-12-17 10:35:53

使用 gRPC 中的 Metadata 提升远程过程调用效率

在当今快节奏的数字世界中，开发人员经常面临着处理大量复杂业务逻辑和数据操作的挑战。

gRPC（远程过程调用）是一种先进的框架，旨在简化跨服务通信，它提供了一系列实用功能，其中 Metadata 尤为突出。

什么是 Metadata？

Metadata 是附加在 gRPC 请求中的额外信息，可由服务端和客户端利用，以实现各种目的。它类似于 HTTP 标头，但专用于 gRPC 通信。

Metadata 的用途

Metadata 可用于以下方面：

身份验证： 服务端可使用 Metadata 验证客户端身份，防止未经授权的访问。
日志记录： 服务端可记录 Metadata 中的客户端请求信息，以进行调试和故障排除。
请求定制： 客户端可使用 Metadata 向服务端传递参数，以便服务端根据需要定制响应。

在 Go 中使用 Metadata

在 Go 语言中，Metadata 与 context.Context 密切相关。可以通过 context.Context 获取和设置 Metadata。以下示例演示如何在 Go 中使用 Metadata：

// 服务端代码

package main

import (
	"context"
	"fmt"
	"log"
	"net"

	"github.com/golang/protobuf/ptypes/timestamp"

	pb "github.com/example/helloworld/helloworld"
	"google.golang.org/grpc"
	"google.golang.org/grpc/metadata"
)

const (
	port = ":50051"
)

type server struct{}

func (s *server) SayHello(ctx context.Context, in *pb.HelloRequest) (*pb.HelloReply, error) {
	md, ok := metadata.FromIncomingContext(ctx)
	if !ok {
		log.Fatalf("failed to get metadata from context")
	}

	// 获取 metadata 中的 "timestamp" 字段
	timestampString, ok := md["timestamp"]
	if !ok {
		log.Fatalf("failed to get timestamp from metadata")
	}

	// 将 timestampString 转换为 timestamp.Timestamp 类型
	timestamp, err := timestamp.Parse(timestampString[0])
	if err != nil {
		log.Fatalf("failed to parse timestamp: %v", err)
	}

	// 使用 timestamp 生成回复消息
	return &pb.HelloReply{Message: fmt.Sprintf("Hello, %s! You sent a timestamp of %s.", in.Name, timestamp)}, 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{})
	if err := s.Serve(lis); err != nil {
		log.Fatalf("failed to serve: %v", err)
	}
}

// 客户端代码

package main

import (
	"context"
	"fmt"
	"log"
	"time"

	pb "github.com/example/helloworld/helloworld"
	"google.golang.org/grpc"
	"google.golang.org/grpc/metadata"
)

const (
	address = "localhost:50051"
)

func main() {
	// 创建一个新的 gRPC 客户端连接
	conn, err := grpc.Dial(address, grpc.WithInsecure())
	if err != nil {
		log.Fatalf("failed to dial: %v", err)
	}
	defer conn.Close()

	// 创建一个新的 gRPC 客户端
	c := pb.NewGreeterClient(conn)

	// 创建一个新的 context 对象，并向其中添加 metadata
	ctx := context.Background()
	md := metadata.Pairs("timestamp", time.Now().Format(time.RFC3339))
	ctx = metadata.NewOutgoingContext(ctx, md)

	// 使用客户端向服务端发送请求
	r, err := c.SayHello(ctx, &pb.HelloRequest{Name: "World"})
	if err != nil {
		log.Fatalf("failed to greet: %v", err)
	}

	// 打印服务端返回的回复消息
	fmt.Println("Greeting:", r.Message)
}