今天我来讲一讲 gRPC 元数据的使用,以及如何获取元数据。 gRPC 应用程序通常会通过 gRPC 服务和消费者之间的 RPC 来共享信息。在大多数场景中,某些与服务端业务逻辑相关的信息会作为远程调用方法的参数,但在某些场景可能存在与服务端业务上下文无关的数据,这些数据不应该通过参数来传递,而应该通过 gRPC 元数据来处理传递。元数据的结构构造是K-V 形式,其中 k 为字符串,v 为任意类型。
接下来就介绍如何在 客户端和服务端之间使用元数据来传递数据。
创建和查询元数据
在 gRPC 应用程序中, 创建元数据非常简单,在 go 语言中有两种方式创建元数据:1) 通过 metadata.New(map[string]string{“key1”:“val1”}) 函数创建;2) 通过 metadata.Pairs 来创建元数据对,相同的 key 会被合并为切片数组。
md := metadata.New(map[string]string{"foo": "bar"})
md := metadata.Pairs(
"key1", "value1",
"key2", "value2",
)
二进制数据也可以设置为元数据值,以元数据值形式所设置的二进制数据在发送前都会进行 base4 编码,在传输过程中会被解码。
在客户端或服务端读取元数据,可以通过传入的 RPC 上下文和 metadata.FromIncomingContext 函数来获取元数据。
md, metadataAvailble := metadata.FromIncomingContext(ctx)
接下来讲解客户端和服务端如何发送和接受元数据。
客户端发送接收元数据
在客户端,要发送元数据,可以创建元数据并将其设置到 RPC 上下文中。在 go 语言中有两种方式实现, 可以使用 NewOutgoingContext 函数创建,也可以使用 AppendToOutgoingContext 函数来将元数据附加到 RPC 上下文中, 使用 NewOutgoingContext 会替换掉上下文中已有的元数据。 在创建完带有元数据的上下文后,就可以用于 RPC 中了。在上下文中设置的元数据会转换成 header 信息。
// 客户端发送元数据
// ****** Metadata : Creation *****
md := metadata.Pairs(
"timestamp", time.Now().Format(time.StampNano),
"kn", "vn",
)
mdCtx := metadata.NewOutgoingContext(context.Background(), md)
ctxA := metadata.AppendToOutgoingContext(mdCtx, "k1", "v1", "k1", "v2", "k2", "v3")
// Search Order
searchStream, _ := client.SearchOrders(ctxA, &wrapper.StringValue{Value: "Google"})
for {
searchOrder, err := searchStream.Recv()
if err == io.EOF {
log.Print("EOF")
break
}
if err == nil {
log.Print("Search Result : ", searchOrder)
}
}
// 在 gRPC 中接受元数据
var header,trailer metadata.MD
r,err := client.SomeRPC(
ctx,
someReq,
grpc.Header(&header),
grpc.Trailer(trailer),
)
stream, err := client.SomeStreamingRPC(ctx)
header, err := stream.Header()
trailer := stream.Trailer()
从对应的 RPC 获取到值时,就可以像处理map一样进行,对元数据进行相应处理。
服务端接收和发送元数据
在服务端接收元数据非常简单,只需在方法调用方法 metadata.FromIncomingContext(ctx),即可读取元数据。
要从服务端发送元数据,可以根据元数据发送头信息或设置 trailer,创建元数据的方法跟前面一样。在一元RPC 和流 RPC 都可以通过 grpc.SendHeader 发送元数据,如果想要将元数据作为 trailer 的一部分发送,则需要使用 grpc.SetTrailer方法设置。
小结
本完先是介绍了如何创建和查询元数据,后面分别简单介绍了 客户端和服务端如何发送和接收元数据。