技術標籤：response header設定

正文

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 worldn"))

 })

 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學習資源如果你準備入IT坑，勵志成為優秀的程式猿，那麼這些資源很適合你，包括但不限於java、go、python、springcloud、elk、嵌入式 、大資料、面試資料、前端 等資源。同時我們組建了一個技術交流群，裡面有很多大佬，會不定時分享技術文章，如果你想來一起學習提高，可以公眾號後臺回覆【2】，免費邀請加技術交流群互相學習提高，會不定期分享程式設計IT相關資源。

掃碼關注，精彩內容第一時間推給你