前言
本文首發於公眾號【我的小碗湯】本公眾號免費提供csdn下載服務，海量IT學習資源，如果你准備入IT坑，勵志成為優秀的程序猿，那么這些資源很適合你，包括但不限於java、go、python、springcloud、elk、嵌入式 、大數據、面試資料、前端 等資源。掃碼關注：
正文
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也取到了值。
關於go、docker、k8s相關學習資源關注文末公眾號后回復【1】獲取
最后，csdn資源，收集了海量學習資料，如果你准備入IT坑，勵志成為優秀的程序猿，那么這些資源很適合你，包括java、go、python、springcloud、elk、嵌入式 、大數據、面試資料、前端等資源。同時我們組建了一個技術交流群，里面有很多大佬，會不定時分享技術文章，如果你想來一起學習提高，可以關注以下公眾號后回復【2】，獲取。
我是小碗湯，我們一起學習，掃碼關注，精彩內容第一時間推給你
