Go 中 GRPC 元数据更新问题

知识点掌握了，还需要不断练习才能熟练运用。下面golang学习网给大家带来一个Golang开发实战，手把手教大家学习《Go 中 GRPC 元数据更新问题》，在实现功能的过程中也带大家重新温习相关知识点，温故而知新，回头看看说不定又有不一样的感悟！

我正在尝试编写一个一元拦截器来将请求 id 添加到传入请求：

func requestidinterceptor(ctx context.context, req interface{},
    info *grpc.unaryserverinfo, handler grpc.unaryhandler) (val interface{}, err error) {

    newctx := ctx
    newctx = createandsetmetadata(newctx, metadata.fromincomingcontext, metadata.newincomingcontext)
    newctx = createandsetmetadata(newctx, metadata.fromoutgoingcontext, metadata.newoutgoingcontext)

    md, ok := metadata.fromoutgoingcontext(newctx)
    fmt.printf("after update. metadata: %v, found: %t\n", md, ok)

    return handler(newctx, req)
}

func createandsetmetadata(ctx context.context, getter func(context.context) (metadata.md, bool),
    setter func(context.context, metadata.md) context.context) context.context {

    meta, ok := getter(ctx)
    fmt.printf("found? %t, metadata: %v\n", ok, meta)
    if !ok {
        meta = metadata.new(map[string]string{})
        ctx = setter(ctx, meta)
    }

    meta.set("x-request-id", "test_request_id")
    fmt.printf("metadata: %v\n", meta)
    return ctx
}

当请求不包含元数据 (requestidinterceptor(context.background(), nil, nil, handler)) 时，此代码按预期工作，但如果上下文包含元数据，则不会填充 id。本质上，输出如下所示：

WITHOUT METADATA
Found? false, Metadata: map[]
Metadata: map[x-request-id:[TEST_REQUEST_ID]]
Found? false, Metadata: map[]
Metadata: map[x-request-id:[TEST_REQUEST_ID]]
After update. Metadata: map[x-request-id:[TEST_REQUEST_ID]], Found: true

WITH METADATA
Found? true, Metadata: map[]
Metadata: map[x-request-id:[TEST_REQUEST_ID]]
Found? true, Metadata: map[]
Metadata: map[x-request-id:[TEST_REQUEST_ID]]
After update. Metadata: map[], Found: true

据我了解，发生这种情况的原因可能是 metadata.md 是按值传递而不是按引用传递，因此当 createandsetmetadata 返回时更新将被丢弃。我该如何修复此代码？

正确答案

问题在于 metadata.fromoutgoingcontext 实际上返回在上下文中找到的 metadata.md 的深层副本。您可以检查 the sources，截至 google.golang.org/[email protected] 仍然如此。

因此，像使用地图一样直接写入它（副本）不会影响存储在上下文中的 md。

您必须再次调用 setter 来传播新值：

func createAndSetMetadata(/* args */) context.Context {
    // ...
    return setter(ctx, meta)
}

以上就是《Go 中 GRPC 元数据更新问题》的详细内容，更多关于的资料请关注golang学习网公众号！