golang 設定 http response 回應標頭與坑

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 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：

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.運行伺服器

go run myServer.go

2.運行用戶端

go run myHttp.go

輸出如下：statuscode是我們設定的500，Name也取到了值。

image

關於go、docker、k8s相關學習資源關注文末公眾號後回複【1】擷取

最後，csdn資源，收集了海量學習資料，如果你準備入IT坑，勵志成為優秀的程式猿，那麼這些資源很適合你，包括java、go、python、springcloud、elk、嵌入式 、大資料、面試資料、前端等資源。同時我們組建了一個技術交流群，裡面有很多大佬，會不定時分享技術文章，如果你想來一起學習提高，可以關注以下公眾號後回複【2】，擷取。

我是小碗湯，我們一起學習，掃碼關注，精彩內容第一時間推給你

歡迎掃碼關注