1、 設(shè)置WriteHeader的順序問(wèn)題
之前遇到個(gè)問(wèn)題，在一段代碼中這樣設(shè)置WriteHeader,最后在header中取Name時(shí)怎么也取不到。
|
1
2
|
w.WriteHeader(201)w.Header().Set("Name", "my name is smallsoup") |
用 golang 寫(xiě) http server 時(shí)，可以很方便可通過(guò) w.Header.Set(k, v) 來(lái)設(shè)置 http response 中 header 的內(nèi)容。但是需要特別注意的是：某些時(shí)候不僅要修改 response的header ，還要修改 response的StatusCode。修改response的StatusCode 可以通過(guò)：w.WriteHeader(code) 來(lái)實(shí)現(xiàn)，例如：
|
1
|
w.WriteHeader(404) |
如果這兩種修改一起做，就必須讓 w.WriteHeader 在所有的 w.Header.Set 之后，因?yàn)?w.WriteHeader 后 Set Header 是無(wú)效的。
而且必須是在 w.Write([]byte("HelloWorld")) 之前，否則會(huì)報(bào) http: multiple response.WriteHeader calls 因?yàn)槠鋵?shí)調(diào)用w.Write的時(shí)候也會(huì)調(diào)用WriteHeader()方法，然后將w.wroteHeader置為true，再次調(diào)WriteHeader()則會(huì)判斷wroteHeader，如果是true則會(huì)報(bào)錯(cuò)，而且本次調(diào)用不生效。
可以看以下源碼說(shuō)明WriteHeader必須在Write之前調(diào)用。
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
|
func (w *response) WriteHeader(code int) { if w.conn.hijacked() { w.conn.server.logf("http: response.WriteHeader on hijacked connection") return }//第二次WriteHeader()進(jìn)來(lái)滿足if條件就報(bào)錯(cuò)直接return if w.wroteHeader { w.conn.server.logf("http: multiple response.WriteHeader calls") return }//第一次write()進(jìn)來(lái)這里會(huì)將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會(huì)對(duì)Header中的key進(jìn)行規(guī)范化處理
go會(huì)對(duì)Header中的key進(jìn)行規(guī)范化處理，所以在獲取response的Header中的K,V值時(shí)一定要小心。
reader.go中非導(dǎo)出方法canonicalMIMEHeaderKey中有這樣一段，會(huì)將header的key進(jìn)行規(guī)范化處理。
1）reader.go中定義了isTokenTable數(shù)組，如果key的長(zhǎng)度大于127或者包含不在isTokenTable中的字符，則該key不會(huì)被處理。
2）將key的首字母大寫(xiě)，字符 - 后的單詞的首字母也大寫(xiě)。
分析如下源碼，可以解釋對(duì)key的大寫(xiě)處理：
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
|
for i, c := range a { // 規(guī)范化:首字母大寫(xiě) // - 之后單子的首字母大寫(xiě) // 如:(Host, User-Agent, If-Modified-Since). if upper && 'a' <= c && c <= 'z' { //大寫(xiě)轉(zhuǎn)小寫(xiě) c -= toLower } else if !upper && 'A' <= c && c <= 'Z' { //小寫(xiě)轉(zhuǎn)大寫(xiě) c += toLower } //重新給key數(shù)組賦值 a[i] = c //設(shè)置大小寫(xiě)標(biāo)志位 upper = c == '-' // for next time} |
正確的調(diào)用方式：
服務(wù)器：myServer.go
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
|
package mainimport ( "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：
|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
|
package mainimport ( "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ù)器
|
1
|
go run myServer.go |
2.運(yùn)行客戶端
|
1
|
go run myHttp.go |
輸出如下：statuscode是我們?cè)O(shè)置的500，Name也取到了值。
golang處理http response碰到的問(wèn)題和需要注意的點(diǎn)
在處理http response的時(shí)候，偶然發(fā)現(xiàn)，body讀取之后想再次讀取的時(shí)候，發(fā)現(xiàn)讀不到任何東西。見(jiàn)下方代碼：
|
1
2
3
4
5
|
response, err = ioutil.ReadAll(resp.Body)if err != nil {log.Println("ioutil ReadAll failed :", err.Error())return} |
之后如果想再次ioutil.ReadAll(resp.Body)的時(shí)候會(huì)發(fā)現(xiàn)讀到的是空。于是我決定去看一下這個(gè)resp.Body，發(fā)現(xiàn)它是一個(gè)io.ReadCloser接口，包含了Reader和Closer接口：
|
1
2
3
4
|
type ReadCloser interface {ReaderCloser} |
于是我想到了文件，它也實(shí)現(xiàn)了io.Reader接口，所以用讀文件試了下：
|
1
2
3
4
5
6
7
8
9
10
11
12
13
|
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é)果。我猜測(cè)應(yīng)該是ioutil.ReadAll()是有記錄偏移量，所以會(huì)出現(xiàn)第二次讀取讀不到的情況。作為client端處理response的時(shí)候會(huì)碰到這個(gè)問(wèn)題，作為server端要處理request body的時(shí)候，一樣會(huì)遇到此問(wèn)題，那么該如何解決這個(gè)問(wèn)題呢？
有一個(gè)方法是再造一個(gè)io.ReadCloser，如下：
|
1
2
3
|
fi2:= ioutil.NopCloser(bytes.NewBuffer(byte1))byte3,err := ioutil.ReadAll(fi2)fmt.Println(string(byte3)) |
此外，作為client端處理response的時(shí)候，有一點(diǎn)要注意的是，body一定要close，否則會(huì)造成GC回收不到，繼而產(chǎn)生內(nèi)存泄露。其實(shí)在go的官方源碼注釋中，也明確注明了response body需要調(diào)用方進(jìn)行手動(dòng)關(guān)閉：It is the caller's responsibility to close Body.
至于response body為什么需要進(jìn)行關(guān)閉，這篇文章進(jìn)行了解釋： http://m.ythuaji.com.cn/article/67352.html
那么作為client端生成的request body，需不需要手動(dòng)關(guān)閉呢，答案是不需要的，net/http中的func (c *Client) Do(req *Request) (*Response, error)會(huì)調(diào)用Close() 。
同樣的，作為server端接收的request body，也是需要關(guān)閉，由Server自動(dòng)進(jìn)行關(guān)閉，The Server will close the request body. The ServeHTTP Handler does not need to.
總結(jié)
以上就是這篇文章的全部?jī)?nèi)容了，希望本文的內(nèi)容對(duì)大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值，如果有疑問(wèn)大家可以留言交流，謝謝大家對(duì)服務(wù)器之家的支持。
原文鏈接：https://studygolang.com/articles/14431
