我是 Go 的新手，使用 Mux 接受 HTTP POST 數(shù)據(jù)。我想使用MaxBytesReader來確?？蛻舳瞬粫蜎]我的服務(wù)器。根據(jù)代碼，有一個requestBodyLimit布爾值指示是否已達到該限制。我的問題是：在使用 MaxBytesReader 時，如何確定我在處理請求時是否真的達到了最大值？這是我的代碼：package mainimport ( "fmt" "log" "html/template" "net/http" "github.com/gorilla/mux")func main() { r := mux.NewRouter() r.HandleFunc("/handle", maxBytes(PostHandler)).Methods("POST") http.ListenAndServe(":8080", r)}// Middleware to enforce the maximum post body sizefunc maxBytes(f http.HandlerFunc) http.HandlerFunc { return func(w http.ResponseWriter, r *http.Request) { // As an example, limit post body to 10 bytes r.Body = http.MaxBytesReader(w, r.Body, 10) f(w, r) }}func PostHandler(w http.ResponseWriter, r *http.Request) { // How do I know if the form data has been truncated? book := r.FormValue("email") fmt.Fprintf(w, "You've requested the book: %s\n", book)}我怎么能夠：確定我已達到最大 POST 限制（或有權(quán)訪問requestBodyLimit我的代碼可以在這種情況下分支嗎？
1 回答
慕標5832272
TA貢獻1966條經(jīng)驗 獲得超4個贊
在處理程序的開頭調(diào)用ParseForm 。如果此方法返回錯誤，則表明超出了大小限制或請求正文在某種程度上無效。寫入錯誤狀態(tài)并從處理程序返回。
沒有一種簡單的方法可以檢測錯誤是由于違反大小限制還是其他一些錯誤造成的。
func PostHandler(w http.ResponseWriter, r *http.Request) {
? ? if err := r.ParseForm(); err != nil {
? ? ? ? http.Error(w, "Bad Request", http.StatusBadRequest)
? ? ? ? return
? ? }
? ? book := r.FormValue("email")
? ? fmt.Fprintf(w, "You've requested the book: %s\n", book)
}
根據(jù)您的需要，將檢查放在中間件中可能會更好：
func maxBytes(f http.HandlerFunc) http.HandlerFunc {
? ? return func(w http.ResponseWriter, r *http.Request) {
? ? ? ? ? ? r.Body = http.MaxBytesReader(w, r.Body, 10)
? ? ? ? ? ? if err := r.ParseForm(); err != nil {
? ? ? ? ? ? ? ? http.Error(w, "Bad Request", http.StatusBadRequest)
? ? ? ? ? ? ? ? return
? ? ? ? ? ? }
? ? ? ? ? ? f(w, r)
? ? }
}
- 1 回答
- 0 關(guān)注
- 171 瀏覽
添加回答
舉報
0/150
提交
取消