解決Golang中ResponseWriter的一個坑

在使用Context.ResponseWriter中的Set/WriteHeader/Write這三個方法時,使用順序必須如下所示,否則會出現某一設置不生效的情況。

ctx.ResponseWriter.Header().Set("Content-type", "application/text")
 ctx.ResponseWriter.WriteHeader(403)
 ctx.ResponseWriter.Write([]byte(resp))

如1:

ctx.ResponseWriter.Header().Set("Content-type", "application/text")
 ctx.ResponseWriter.Write([]byte(resp))
 ctx.ResponseWriter.WriteHeader(403)

會導致返回碼一直是200

補充:Go裡w http.ResponseWriter,調用w.Write()方法報錯

Go裡w http.ResponseWriter寫入報錯

http: request method or response status code does not allow

1. 下面是報錯截圖

2. 點進去Write方法

它首先是一個接口;

由於它是在HTTP web服務器的應用場景,所以它具體的實現方法在net/http/server.go裡:

func (w *response) Write(data []byte) (n int, err error) {
 return w.write(len(data), data, "")
}

再點進去,函數裡你會發現有一個關鍵的判斷

// 其中ErrBodyNotAllowed的
// 代碼內容
// ErrBodyNotAllowed = errors.New("http: request method or response status code does not allow body")
if !w.bodyAllowed() {
 return 0, ErrBodyNotAllowed
}
 

點進去,發現它在沒有設置Header時會panic,當然這跟我們當前要討論的問題關系不大,關鍵在bodyAllowedForStatus()方法

func (w *response) bodyAllowed() bool {
 if !w.wroteHeader {
  panic("")
 }
 return bodyAllowedForStatus(w.status)
}

再點,終於看到瞭,當設置狀態碼為【100,199】、204、304就會報這個錯,而我剛好設置的狀態碼就是204,我把它改成200重新試下,問題解決。

func bodyAllowedForStatus(status int) bool {
 switch {
 case status >= 100 && status <= 199:
  return false
 case status == 204:
  return false
 case status == 304:
  return false
 }
 return true
}

以上為個人經驗,希望能給大傢一個參考,也希望大傢多多支持WalkonNet。如有錯誤或未考慮完全的地方,望不吝賜教。