golang 設置 http response 響應頭與坑

正文

一、 設置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")
   }
 }
}複製代碼

二、 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學習資源，若是你準備入IT坑，勵志成爲優秀的程序猿，那麼這些資源很適合你，包括但不限於java、go、python、springcloud、elk、嵌入式 、大數據、面試資料、前端 等資源。同時咱們組建了一個技術交流羣，裏面有不少大佬，會不定時分享技術文章，若是你想來一塊兒學習提升，能夠公衆號後臺回覆【2】，免費邀請加技術交流羣互相學習提升，會不按期分享編程IT相關資源。

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