如何通过Go的MaxBytesReader确定我是否达到了大小限制

ajsxfq5m  于 2023-11-14  发布在  Go
关注(0)|答案(3)|浏览(142)

我是Go的新手,使用Mux来接受HTTP POST数据。我想使用MaxBytesReader来确保客户端不会压倒我的服务器。根据代码,有一个requestBodyLimit布尔值表示是否达到了限制。
我的问题是:当使用MaxBytesReader时,我如何确定在处理请求时是否真的达到了最大值?
下面是我的代码:

package main

import (
        "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 size
func 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限制(或可以访问requestBodyLimit
  • 在这种情况下,我的代码是否能够进行分支?
ljsrvy3e

ljsrvy3e1#

在处理程序的开头调用ParseForm。如果此方法返回错误,则违反了大小限制或请求主体以某种方式无效。写入错误状态并从处理程序返回。
没有一种简单的方法来检测错误是由于违反大小限制还是其他错误造成的。

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)
}

字符串
根据您的需要,最好将检查放在中间件中:

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)
    }
}

i2byvkas

i2byvkas2#

可以这样检查返回的错误是否是http.MaxBytesError

r.Body = http.MaxBytesReader(w, r.Body, 10)
if err := r.ParseForm(); err != nil {
    if _, ok := err.(*http.MaxBytesError); ok {
        // handle http.StatusRequestEntityTooLarge error
    }
      // handle http.StatusBadRequest error
}

字符串
errors.Is()errors.As()在这种情况下不起作用。

8yparm6h

8yparm6h3#

您可以通过检查读取数据的长度是否大于(或等于)MaxBytesSize来确定是否超过了限制:

maxBytesSize := 10
r.Body = http.MaxBytesReader(w, r.Body, maxBytesSize)

// check if request body is not too large
data, err := ioutil.ReadAll(r.Body)
if err != nil {
    if len(data) >= maxBytesSize {
         //exceeded
    }
    // some other error
}

字符串

相关问题