我是 Go 的新手，使用 Mux 接受 HTTP POST 數(shù)據(jù)。我想使用MaxBytesReader來確保客戶端不會(huì)淹沒我的服務(wù)器。根據(jù)代碼，有一個(gè)requestBodyLimit布爾值指示是否已達(dá)到該限制。我的問題是：在使用 MaxBytesReader 時(shí)，如何確定我在處理請(qǐng)求時(shí)是否真的達(dá)到了最大值？這是我的代碼：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)}我怎么能夠：確定我已達(dá)到最大 POST 限制（或有權(quán)訪問requestBodyLimit我的代碼可以在這種情況下分支嗎？
1 回答
慕標(biāo)5832272
TA貢獻(xiàn)1966條經(jīng)驗(yàn) 獲得超4個(gè)贊
在處理程序的開頭調(diào)用ParseForm 。如果此方法返回錯(cuò)誤，則表明超出了大小限制或請(qǐng)求正文在某種程度上無效。寫入錯(cuò)誤狀態(tài)并從處理程序返回。
沒有一種簡(jiǎn)單的方法可以檢測(cè)錯(cuò)誤是由于違反大小限制還是其他一些錯(cuò)誤造成的。
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ù)您的需要，將檢查放在中間件中可能會(huì)更好：
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)注
- 152 瀏覽
添加回答
舉報(bào)
0/150
提交
取消