扩展 ResponseWriter 的 Header() 方法返回的映射

扩展 ResponseWriter 的 Header() 方法返回的映射，这是 HTTP 响应头操作中的一个常见需求。然而，由于 ResponseWriter 类型的限制，直接将一个响应头映射赋值给另一个是不可能的。为了解决这个问题，Go 1.18 引入了 golang.org/x/exp/maps 包，其中提供了 Copy 函数，可以方便地复制键/值对。使用 Copy 函数，可以将一个响应头映射中的所有键/值对添加到另一个映射中，从而有效地扩展了 Header() 方法返回的映射。

所以我确信我想在这里作弊，但是 responsewriter 文档有一个方法“header()”，它返回它正在使用的 header 对象。 https://pkg.go.dev/net/http#responsewriter.header

现在我从其他地方获得了 http.response，我想将其中的所有标头复制到我的 responsewriter 中。

现在，我可以使用这样的 for 循环：

for k := range resp.header 
    w.header().add(k, resp.header.get(k))
}

从逻辑上讲，将引用从 responsewriter 的标头更改为 response 的标头对我来说也是有意义的，但是 responsewriter 类型似乎正在积极尝试阻止我这样做。

想到了这样愚蠢的事情

w.header() = resp.header

或者

rwHeader := w.Header()
rwHeader = resp.Header

显然，这两种方法都没有意义，根本不起作用，但希望能传达我想要做的事情的想法。

谁能解释一下为什么我想做的事情不起作用？或者也许确实如此，但我只是没有找到方法？

正确答案

您不必自己编写该循环。 go 1.18 添加了包 golang.org/x/exp/maps，它提供了方便的 Copy function：

func copy[m ~map[k]v, k comparable, v any](dst, src m)

copy 复制 src 中的所有键/值对，并将它们添加到 dst。什么时候 src 中的键已存在于 dst 中，dst 中的值将是 被与 src 中的键关联的值覆盖。

import "golang.org/x/exp/maps"
// ...
maps.copy(w.header(), resp.header)

但是，请注意，由于响应可以包含 duplicate headers，因此 maps.copy 的使用并不完全等同于您的循环：

for k := range resp.Header 
    w.Header().Add(k, resp.Header.Get(k))
}

与 maps.copy 相反，您的循环仅检索与 resp.header 中存在的每个标头名称相对应的第一个标头值。

您不能。

w 是一个 http.ResponseWriter，它是一个接口类型 - 因此只有方法，没有直接访问的字段。如您所知，它有一个获取底层 header 映射的方法。

但是，它没有“setter”方法来替换地图。因此，复制标头值的唯一方法是像您引用的循环一样手动进行。

理论要掌握，实操不能落！以上关于《扩展 ResponseWriter 的 Header() 方法返回的映射》的详细介绍，大家都掌握了吧！如果想要继续提升自己的能力，那么就来关注golang学习网公众号吧！