一尘不染

Golang context.WithValue:如何添加几个键值对

go

使用Go的context程序包,可以使用以下命令将特定于请求的数据传递到请求处理函数堆栈

func WithValue(parent Context, key, val interface{}) Context

这将创建一个新Context的副本,它是父副本,并包含可以用密钥访问的值val。

如果要在中存储多个键值对,该Context如何进行?我是否应该打WithValue()几次电话,每次将Context上次通话收到的信息传递给WithValue()?这看起来很麻烦。
还是我应该使用一个结构并将所有数据都放在那里,我只需要传递一个值(该结构),就可以从中访问所有其他值?

还是有办法将几个键值对传递给WithValue()


阅读 885

收藏
2020-07-02

共1个答案

一尘不染

您几乎列出了您的选择。您要寻找的答案取决于您要如何使用上下文中存储的值。

context.Context是一个不可变的对象,只有通过复制它并将其添加新的键值(通过context程序包在幕后完成),才能使用键值对“扩展”它。

您是否希望其他处理程序能够通过键以透明方式访问所有值?然后,始终使用最后一个操作的上下文将所有内容添加到循环中。

这里要注意的一件事是,context.Context不要map在幕后使用a 来存储键-
值对,乍一看可能令人惊讶,但如果您考虑到它必须是不变的并且可以安全地并发使用,则不是。

用一个map

因此,例如,如果您有很多键值对,并且需要通过键 快速
查找值,则分别添加每个键将导致ContextValue()方法变慢。在这种情况下,最好将所有键值对添加为单个map值,该值可以通过进行访问Context.Value(),并且可以通过关联的键O(1)及时查询其中的每个值。知道这对于并发使用并不安全,因为可以从并发goroutine修改映射。

用一个struct

如果要struct对所有要添加的键值对使用一个具有字段的大值,那么这也可能是一个可行的选择。使用访问该结构Context.Value()将返回该结构的一个副本,因此可以安全地并发使用(每个goroutine只能获得一个不同的副本),但是如果您有很多键值对,这将导致不必要的副本。每当有人需要一个单独的字段时,它都是一个大结构。

使用 混合 解决方案

一个 混合
的解决办法是把所有的键值对的map,并为此创建地图的包装结构,隐藏了map(不导出字段),并提供了对存储在地图的值的吸气剂。仅将此包装添加到上下文中,可以保持对多个goroutine
安全并发访问map未导出),但是 不需要复制大数据map值是没有键值数据的小描述符),而且仍然
(最终您将为地图编制索引)。

它看起来像这样:

type Values struct {
    m map[string]string
}

func (v Values) Get(key string) string {
    return v.m[key]
}

使用它:

v := Values{map[string]string{
    "1": "one",
    "2": "two",
}}

c := context.Background()
c2 := context.WithValue(c, "myvalues", v)

fmt.Println(c2.Value("myvalues").(Values).Get("2"))

输出(在Go Playground上尝试):

two

如果性能不是很关键(或者您有相对较少的键/值对),我将分别添加它们。

2020-07-02