解決Golang中ResponseWriter的一個坑

Posted on by

在使用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。如有錯誤或未考慮完全的地方，望不吝賜教。

推薦閱讀：