清风细雨楼

golang设置http response响应头与填坑记录
脚本专栏 2024/12/27 佚名
3 1
清风细雨楼 Design By www.eepep.com

1、 设置WriteHeader的顺序问题

之前遇到个问题，在一段代码中这样设置WriteHeader,最后在header中取Name时怎么也取不到。

w.WriteHeader(201)
w.Header().Set("Name", "my name is smallsoup")

用 golang 写 http server 时，可以很方便可通过 w.Header.Set(k, v) 来设置 http response 中 header 的内容。但是需要特别注意的是：某些时候不仅要修改 response的header ，还要修改 response的StatusCode。修改response的StatusCode 可以通过：w.WriteHeader(code) 来实现，例如：

w.WriteHeader(404)

如果这两种修改一起做，就必须让 w.WriteHeader 在所有的 w.Header.Set 之后，因为 w.WriteHeader 后 Set Header 是无效的。

而且必须是在 w.Write([]byte("HelloWorld")) 之前，否则会报 http: multiple response.WriteHeader calls 因为其实调用w.Write的时候也会调用WriteHeader()方法，然后将w.wroteHeader置为true，再次调WriteHeader()则会判断wroteHeader，如果是true则会报错，而且本次调用不生效。

可以看以下源码说明WriteHeader必须在Write之前调用。

func (w *response) WriteHeader(code int) {
 if w.conn.hijacked() {
 w.conn.server.logf("http: response.WriteHeader on hijacked connection")
 return
 }
//第二次WriteHeader()进来满足if条件就报错直接return
 if w.wroteHeader {
 w.conn.server.logf("http: multiple response.WriteHeader calls")
 return
 }
//第一次write()进来这里会将w.wroteHeader置为true
 w.wroteHeader = true
 w.status = code

 if w.calledHeader && w.cw.header == nil {
 w.cw.header = w.handlerHeader.clone()
 }

 if cl := w.handlerHeader.get("Content-Length"); cl != "" {
 v, err := strconv.ParseInt(cl, 10, 64)
 if err == nil && v >= 0 {
 w.contentLength = v
 } else {
 w.conn.server.logf("http: invalid Content-Length of %q", cl)
 w.handlerHeader.Del("Content-Length")
 }
 }
}

2、 go会对Header中的key进行规范化处理

go会对Header中的key进行规范化处理，所以在获取response的Header中的K,V值时一定要小心。

reader.go中非导出方法canonicalMIMEHeaderKey中有这样一段，会将header的key进行规范化处理。

1）reader.go中定义了isTokenTable数组，如果key的长度大于127或者包含不在isTokenTable中的字符，则该key不会被处理。

2）将key的首字母大写，字符 - 后的单词的首字母也大写。

分析如下源码，可以解释对key的大写处理：

for i, c := range a {
 // 规范化:首字母大写
 // - 之后单子的首字母大写
 // 如:(Host, User-Agent, If-Modified-Since).
 if upper && 'a' <= c && c <= 'z' {
 //大写转小写
 c -= toLower
 } else if !upper && 'A' <= c && c <= 'Z' {
 //小写转大写
 c += toLower
 }
 //重新给key数组赋值
 a[i] = c
 //设置大小写标志位
 upper = c == '-' // for next time
}

正确的调用方式：

服务器：myServer.go

package main

import (
 "net/http"
)

func main() {

 http.HandleFunc("/", func (w http.ResponseWriter, r *http.Request){


 w.Header().Set("name", "my name is smallsoup")
 w.WriteHeader(500)
 w.Write([]byte("hello world\n"))

 })

 http.ListenAndServe(":8080", nil)
}

客户端：

myHttp.go：

package main

import (
 "fmt"
 "io/ioutil"
 "net/http"
)

func main() {

 myHttpGet()

}

func myHttpGet() {

 rsp, err := http.Get("http://localhost:8080")
 if err != nil {
 fmt.Println("myHttpGet error is ", err)
 return
 }

 defer rsp.Body.Close()
 body, err := ioutil.ReadAll(rsp.Body)
 if err != nil {
 fmt.Println("myHttpGet error is ", err)
 return
 }

 fmt.Println("response statuscode is ", rsp.StatusCode, 
 "\nhead[name]=", rsp.Header["Name"], 
 "\nbody is ", string(body))
}

1.运行服务器

go run myServer.go

2.运行客户端

go run myHttp.go

输出如下：statuscode是我们设置的500，Name也取到了值。

golang设置http response响应头与填坑记录

golang处理http response碰到的问题和需要注意的点

在处理http response的时候，偶然发现，body读取之后想再次读取的时候，发现读不到任何东西。见下方代码：

response, err = ioutil.ReadAll(resp.Body)
if err != nil {
log.Println("ioutil ReadAll failed :", err.Error())
return
}

之后如果想再次ioutil.ReadAll(resp.Body)的时候会发现读到的是空。于是我决定去看一下这个resp.Body，发现它是一个io.ReadCloser接口，包含了Reader和Closer接口：

type ReadCloser interface {
Reader
Closer
}

于是我想到了文件，它也实现了io.Reader接口，所以用读文件试了下：

func readFile(path string)string{
fi,err := os.Open(path)
if err != nil{panic(err)}
defer fi.Close()

byte1,err := ioutil.ReadAll(fi)
fmt.Println(string(byte1))

byte2,err := ioutil.ReadAll(fi)
fmt.Println(string(byte2))

return string(fd)
}

发现结果是一致的，fmt.Println(string(fd2))打印不出任何结果。我猜测应该是ioutil.ReadAll()是有记录偏移量，所以会出现第二次读取读不到的情况。作为client端处理response的时候会碰到这个问题，作为server端要处理request body的时候，一样会遇到此问题，那么该如何解决这个问题呢？
有一个方法是再造一个io.ReadCloser，如下：

fi2:= ioutil.NopCloser(bytes.NewBuffer(byte1))
byte3,err := ioutil.ReadAll(fi2)
fmt.Println(string(byte3))

此外，作为client端处理response的时候，有一点要注意的是，body一定要close，否则会造成GC回收不到，继而产生内存泄露。其实在go的官方源码注释中，也明确注明了response body需要调用方进行手动关闭：It is the caller's responsibility to close Body.

至于response body为什么需要进行关闭，这篇文章进行了解释： https://www.jb51.net/article/146275.htm

那么作为client端生成的request body，需不需要手动关闭呢，答案是不需要的，net/http中的func (c *Client) Do(req *Request) (*Response, error)会调用Close() 。

同样的，作为server端接收的request body，也是需要关闭，由Server自动进行关闭，The Server will close the request body. The ServeHTTP Handler does not need to.

总结

以上就是这篇文章的全部内容了，希望本文的内容对大家的学习或者工作具有一定的参考学习价值，如果有疑问大家可以留言交流，谢谢大家对的支持。

清风细雨楼 Design By www.eepep.com
广告合作：本站广告合作请联系QQ：858582 申请时备注：广告合作（否则不回）
免责声明：本站资源来自互联网收集,仅供用于学习和交流,请遵循相关法律法规,本站一切资源不代表本站立场,如有侵权、后门、不妥请联系本站删除！
清风细雨楼 Design By www.eepep.com
www.eepep.com 清风细雨楼
120,135影音资源
344,641技术资源
22,817软件资源
435,032站长资源

更新日志

友情链接

杰晶网络 DDR爱好者之家 桃源资源网 杰网资源 富贵资源网 南强小屋 铁雪资源网 幽灵资源网 万梅资源网 狼山资源网 白云岛资源网 昆仑资源网 相思资源网 明霞山资源网 内蒙古资源网 黑松山资源网 茶园资源网 饿虎岗资源网 大旗谷资源网 常春岛资源网 岱庙资源网 兴国资源网 快活林资源网 蝙蝠岛资源网 帝王谷资源网 白云城资源网 伏龙阁资源网 清风细雨楼 天枫庄资源网 圆月山庄资源网 无争山庄资源网 神水资源网 移花宫资源网 神剑山庄资源网 无为清净楼资源网 金钱帮资源网 丐帮资源网 华山资源网 极乐门资源网 小李飞刀资源网 凤求凰客栈 风云阁资源网 金狮镖局 鸳鸯亭资源网 千金楼资源网 更多链接
清风细雨楼 Design By www.eepep.com