1、 設(shè)置WriteHeader的順序問題

之前遇到個問題，在一段代碼中這樣設(shè)置WriteHeader,最后在header中取Name時怎么也取不到。

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

用 golang 寫 http server 時，可以很方便可通過 w.Header.Set(k, v) 來設(shè)置 http response 中 header 的內(nèi)容。但是需要特別注意的是：某些時候不僅要修改 response的header ，還要修改 response的StatusCode。修改response的StatusCode 可以通過：w.WriteHeader(code) 來實(shí)現(xiàn)，例如：

w.WriteHeader(404)

如果這兩種修改一起做，就必須讓 w.WriteHeader 在所有的 w.Header.Set 之后，因?yàn)?w.WriteHeader 后 Set Header 是無效的。

而且必須是在 w.Write([]byte("HelloWorld")) 之前，否則會報 http: multiple response.WriteHeader calls 因?yàn)槠鋵?shí)調(diào)用w.Write的時候也會調(diào)用WriteHeader()方法，然后將w.wroteHeader置為true，再次調(diào)WriteHeader()則會判斷wroteHeader，如果是true則會報錯，而且本次調(diào)用不生效。

可以看以下源碼說明WriteHeader必須在Write之前調(diào)用。

func (w *response) WriteHeader(code int) {
 if w.conn.hijacked() {
 w.conn.server.logf("http: response.WriteHeader on hijacked connection")
 return
 }
//第二次WriteHeader()進(jìn)來滿足if條件就報錯直接return
 if w.wroteHeader {
 w.conn.server.logf("http: multiple response.WriteHeader calls")
 return
 }
//第一次write()進(jìn)來這里會將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進(jìn)行規(guī)范化處理

go會對Header中的key進(jìn)行規(guī)范化處理，所以在獲取response的Header中的K,V值時一定要小心。

reader.go中非導(dǎo)出方法canonicalMIMEHeaderKey中有這樣一段，會將header的key進(jìn)行規(guī)范化處理。

1）reader.go中定義了isTokenTable數(shù)組，如果key的長度大于127或者包含不在isTokenTable中的字符，則該key不會被處理。

2）將key的首字母大寫，字符 - 后的單詞的首字母也大寫。

分析如下源碼，可以解釋對key的大寫處理：

for i, c := range a {
 // 規(guī)范化:首字母大寫
 // - 之后單子的首字母大寫
 // 如:(Host, User-Agent, If-Modified-Since).
 if upper  'a' = c  c = 'z' {
 //大寫轉(zhuǎn)小寫
 c -= toLower
 } else if !upper  'A' = c  c = 'Z' {
 //小寫轉(zhuǎn)大寫
 c += toLower
 }
 //重新給key數(shù)組賦值
 a[i] = c
 //設(shè)置大小寫標(biāo)志位
 upper = c == '-' // for next time
}

正確的調(diào)用方式：

服務(wù)器：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.運(yùn)行服務(wù)器

go run myServer.go

2.運(yùn)行客戶端

go run myHttp.go

輸出如下：statuscode是我們設(shè)置的500，Name也取到了值。

golang處理http response碰到的問題和需要注意的點(diǎn)

在處理http response的時候，偶然發(fā)現(xiàn)，body讀取之后想再次讀取的時候，發(fā)現(xiàn)讀不到任何東西。見下方代碼：

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

之后如果想再次ioutil.ReadAll(resp.Body)的時候會發(fā)現(xiàn)讀到的是空。于是我決定去看一下這個resp.Body，發(fā)現(xiàn)它是一個io.ReadCloser接口，包含了Reader和Closer接口：

type ReadCloser interface {
Reader
Closer
}

于是我想到了文件，它也實(shí)現(xiàn)了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)
}

發(fā)現(xiàn)結(jié)果是一致的，fmt.Println(string(fd2))打印不出任何結(jié)果。我猜測應(yīng)該是ioutil.ReadAll()是有記錄偏移量，所以會出現(xiàn)第二次讀取讀不到的情況。作為client端處理response的時候會碰到這個問題，作為server端要處理request body的時候，一樣會遇到此問題，那么該如何解決這個問題呢？
有一個方法是再造一個io.ReadCloser，如下：

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

此外，作為client端處理response的時候，有一點(diǎn)要注意的是，body一定要close，否則會造成GC回收不到，繼而產(chǎn)生內(nèi)存泄露。其實(shí)在go的官方源碼注釋中，也明確注明了response body需要調(diào)用方進(jìn)行手動關(guān)閉：It is the caller's responsibility to close Body.

至于response body為什么需要進(jìn)行關(guān)閉，這篇文章進(jìn)行了解釋： https://www.jb51.net/article/146275.htm

那么作為client端生成的request body，需不需要手動關(guān)閉呢，答案是不需要的，net/http中的func (c *Client) Do(req *Request) (*Response, error)會調(diào)用Close() 。

同樣的，作為server端接收的request body，也是需要關(guān)閉，由Server自動進(jìn)行關(guān)閉，The Server will close the request body. The ServeHTTP Handler does not need to.

總結(jié)

以上就是這篇文章的全部內(nèi)容了，希望本文的內(nèi)容對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價值，如果有疑問大家可以留言交流，謝謝大家對腳本之家的支持。