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碰到的问题和需要注意的点
在处理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
.
总结
以上就是这篇文章的全部内容了,希望本文的内容对大家的学习或者工作具有一定的参考学习价值,如果有疑问大家可以留言交流,谢谢大家对的支持。
免责声明:本站资源来自互联网收集,仅供用于学习和交流,请遵循相关法律法规,本站一切资源不代表本站立场,如有侵权、后门、不妥请联系本站删除!
《魔兽世界》大逃杀!60人新游玩模式《强袭风暴》3月21日上线
暴雪近日发布了《魔兽世界》10.2.6 更新内容,新游玩模式《强袭风暴》即将于3月21 日在亚服上线,届时玩家将前往阿拉希高地展开一场 60 人大逃杀对战。
艾泽拉斯的冒险者已经征服了艾泽拉斯的大地及遥远的彼岸。他们在对抗世界上最致命的敌人时展现出过人的手腕,并且成功阻止终结宇宙等级的威胁。当他们在为即将于《魔兽世界》资料片《地心之战》中来袭的萨拉塔斯势力做战斗准备时,他们还需要在熟悉的阿拉希高地面对一个全新的敌人──那就是彼此。在《巨龙崛起》10.2.6 更新的《强袭风暴》中,玩家将会进入一个全新的海盗主题大逃杀式限时活动,其中包含极高的风险和史诗级的奖励。
《强袭风暴》不是普通的战场,作为一个独立于主游戏之外的活动,玩家可以用大逃杀的风格来体验《魔兽世界》,不分职业、不分装备(除了你在赛局中捡到的),光是技巧和战略的强弱之分就能决定出谁才是能坚持到最后的赢家。本次活动将会开放单人和双人模式,玩家在加入海盗主题的预赛大厅区域前,可以从强袭风暴角色画面新增好友。游玩游戏将可以累计名望轨迹,《巨龙崛起》和《魔兽世界:巫妖王之怒 经典版》的玩家都可以获得奖励。