百度360必应搜狗淘宝本站头条
当前位置:网站首页 > 热门文章 > 正文

Golang 操作 HTTP Header 的一个小细节

bigegpt 2024-08-24 23:11 2 浏览


在平时开发中发送 HTTP 请求是非常频繁的操作,对于 HTTP Header 的操作也是很常用的操作。今天在 Review 一段代码的时候,发现了一个小问题,事后想想其实是很小的问题,很可能都不会影响运行结果,但是都是如果没有遇到过很可能就不会注意到的一些问题,所以作为复盘记录,把这些小细节整理出来。

Header.Add 还是 Header.Set ?

这个问题起始于一段代码(仅保留相关部分):

http.HandleFunc("/forum/correlation", RegForumCorrelation)
func RegForumCorrelation(w http.ResponseWriter, r *http.Request) {
  // ... 省略业务代码 ...
  // 发送 HTTP 请求
  req, err := http.NewRequest("POST", bussinessURL, reqBody)
  if err != nil {
    // error handle
  }
  req.Header.Add("X-Authorization", authorizationSign)
  req.Header.Add("Content-Type", "application/json")
  // ...
}

在这里,我看到这个开发者使用了 Header.Add 来设置请求的 Header,我询问开发者这里为什么用 Add 而不用 Set,得到的回答是这两个方法得到的结果是一样的。那么,这里使用 Add 方法对么,或者说合适么?

如果说从结果来看,这里并不会有什么问题,因为这里只是一次性的设置,然后发送请求,目前的实现方法完全可以达到目的而且不会有问题。那么是否就意味着 Header.Add 和 Header.Set 没有什么区别呢,当然不是。所以这里我们本着要想彻底解决问题就要从源码探究的原则来看下这个问题。

其实源码中的注释已经介绍的很明显了,这里不再赘述,我们主要看代码。

type MIMEHeader map[string][]string

func (h MIMEHeader) Add(key, value string) {
  key = CanonicalMIMEHeaderKey(key)
  h[key] = append(h[key], value)
}

func (h MIMEHeader) Set(key, value string) {
  h[CanonicalMIMEHeaderKey(key)] = []string{value}
}

上面就是 Add 和 Set 两个方法的源码,一目了然。Header 是一个 key 为 string,value 为 []string 的 map 类型。其中 key 为 Header 的名称,value 为 Header 对应名称的值,是一个数组。到这里 Add 和 Set 的区别就显而易见了,Add 向 value 的数组中追加一个值,Set 则重写这个值,覆盖掉已有的。

然而这里其实还有个小细节,就是 Header.Get 方法,既然 Header 中 key 对应的 value 是一个 []string 类型,那么 Get 方法获取到的也是这个么?其实并不是:

func (h MIMEHeader) Get(key string) string {
  if h == nil {
    return ""
  }
  v := h[CanonicalMIMEHeaderKey(key)]
  if len(v) == 0 {
    return ""
  }
  return v[0]
}

通过源码可以看到,使用 Get 方法只能够获取到切片中的第一个值,如果想要获取其他的值,只能通过直接访问 Header 的 map 结构去获取了。当然这些也已经在 Get 方法的注释中提到了。所以这里也可以提一个习惯,使用标准库的时候,最好能够详细看看方法的注释,毕竟官方的注释还是很详细的。

在大多数 Get 可以满足需求的时候,我们还是更推荐使用 Get 方法。在源码中有一个登场率很高的函数 CanonicalMIMEHeaderKey,Header 名称在 Add,Set,Get 的时候都会经过这个函数,不知道大家有没有注意过,我们在通过 Add,Set 设置 Header 的时候,Header 的名称都是大小写不敏感的,而在我们接收到的 Header 中却都是规范的。其实这就是 CanonicalMIMEHeaderKey 的作用,把传入的字符串类型参数的首字母大写,然后其他连接词的首字母大写,其余的都小写。例如 Content-Type,User-Agent。

所以如果说在实际的开发中 Get 方法无法满足需求,需要直接访问 Header 的 map 结构,那么一定要注意这一点。

欢迎大家能够交流自己的一些问题和经验,持续学习,共同成长!

相关推荐

AI「自我复制」能力曝光,RepliBench警示:大模型正在学会伪造身份

科幻中AI自我复制失控场景,正成为现实世界严肃的研究课题。英国AISI推出RepliBench基准,分解并评估AI自主复制所需的四大核心能力。测试显示,当前AI尚不具备完全自主复制能力,但在获取资源...

【Python第三方库安装】介绍8种情况,这里最全看这里就够了!

**本图文作品主要解决CMD或pycharm终端下载安装第三方库可能出错的问题**本作品介绍了8种安装方法,这里最全的python第三方库安装教程,简单易上手,满满干货!希望大家能愉快地写代码,而不要...

pyvips,一个神奇的 Python 库!(pythonvip视频)

大家好,今天为大家分享一个神奇的Python库-pyvips。在图像处理领域,高效和快速的图像处理工具对于开发者来说至关重要。pyvips是一个强大的Python库,基于libvips...

mac 安装tesseract、pytesseract以及简单使用

一.tesseract-OCR的介绍1.tesseract-OCR是一个开源的OCR引擎,能识别100多种语言,专门用于对图片文字进行识别,并获取文本。但是它的缺点是对手写的识别能力比较差。2.用te...

实测o3/o4-mini:3分钟解决欧拉问题,OpenAI最强模型名副其实!

号称“OpenAI迄今为止最强模型”,o3/o4-mini真实能力究竟如何?就在发布后的几小时内,网友们的第一波实测已新鲜出炉。最强推理模型o3,即使遇上首位全职提示词工程师RileyGoodsid...

使用Python将图片转换为字符画并保存到文件

字符画(ASCIIArt)是将图片转换为由字符组成的艺术作品。利用Python,我们可以轻松实现图片转字符画的功能。本教程将带你一步步实现这个功能,并详细解释每一步的代码和实现原理。环境准备首先,你...

5分钟-python包管理器pip安装(python pip安装包)

pip是一个现代的,通用、普遍的Python包管理工具。提供了对Python包的查找、下载、安装、卸载的功能,是Python开发的基础。第一步:PC端打开网址:选择gz后缀的文件下载第二步:...

网络问题快速排查,你也能当好自己家的网络攻城狮

前面写了一篇关于网络基础和常见故障排查的,只列举了工具。没具体排查方式。这篇重点把几个常用工具的组合讲解一下。先有请今天的主角:nslookup及dig,traceroute,httping,teln...

终于把TCP/IP 协议讲的明明白白了,再也不怕被问三次握手了

文:涤生_Woo下周就开始和大家成体系的讲hadoop了,里面的每一个模块的技术细节我都会涉及到,希望大家会喜欢。当然了你也可以评论或者留言自己喜欢的技术,还是那句话,希望咱们一起进步。今天周五,讲讲...

记一次工控触摸屏故障的处理(工控触摸屏维修)

先说明一下,虽然我是自动化专业毕业,但已经很多年不从事现场一线的工控工作了。但自己在单位做的工作也牵涉到信息化与自动化的整合,所以平时也略有关注。上一周一个朋友接到一个活,一家光伏企业用于启动机组的触...

19、90秒快速“读懂”路由、交换命令行基础

命令行视图VRP分层的命令结构定义了很多命令行视图,每条命令只能在特定的视图中执行。本例介绍了常见的命令行视图。每个命令都注册在一个或多个命令视图下,用户只有先进入这个命令所在的视图,才能运行相应的命...

摄像头没图像的几个检查方法(摄像头没图像怎么修复)

背景描述:安防监控项目上,用户的摄像头运行了一段时间有部分摄像头不能进行预览,需要针对不能预览的摄像头进行排查,下面列出几个常见的排查方法。问题解决:一般情况为网络、供电、设备配置等情况。一,网络检查...

小谈:必需脂肪酸(必需脂肪酸主要包括)

必需脂肪酸是指机体生命活动必不可少,但机体自身又不能合成,必需由食物供给的多不饱和脂肪酸(PUFA)。必需脂肪酸主要包括两种,一种是ω-3系列的α-亚麻酸(18:3),一种是ω-6系列的亚油酸(18:...

期刊推荐:15本sci四区易发表的机械类期刊

  虽然,Sci四区期刊相比收录在sci一区、二区、三区的期刊来说要求不是那么高,投稿起来也相对容易一些。但,sci四区所收录的期刊中每本期刊的投稿难易程度也是不一样的。为方便大家投稿,本文给大家推荐...

be sick of 用法考察(be in lack of的用法)

besick表示病了,做谓语.本身是形容词,有多种意思.最通常的是:生病,恶心,呕吐,不适,晕,厌烦,无法忍受asickchild生病的孩子Hermother'sverysi...