Golang 中的内容长度

Question

我在网上找不到任何有用的东西。我正在编写一个 REST API，我想以字节为单位记录请求正文的大小以获取指标。Go net/http API 不直接提供。http.Request 确实有 Content-Length 字段，但该字段可以为空，否则客户端可能会发送虚假数据。有没有办法在中间件级别获得它？蛮力方法是阅读全身并检查大小。但是如果我在中间件中这样做，处理程序将无法访问主体，因为它已经被读取和关闭。

Answer 1

你为什么要在这里中间？
简单的方法是 根据需要使用请求正文b, err = io.Copy(anyWriterOrMultiwriter, r.Body)
b时请求的总内容长度。err == nil
还b, err = io.Copy(ioutil.Discard, r.Body)

Answer 2

您可以编写ReadCloser一个自定义代理现有的并计算字节数。就像是：

type LengthReader struct {

    Source io.ReadCloser

    Length int

}

func (r *LengthReader) Read(b []byte) (int, error) {

    n, err := r.Source.Read(b)

    r.Length += n

    return n, err

}

func (r *LengthReader) Close() error {

    var buf [32]byte

    var n int

    var err error

    for err == nil {

        n, err = r.Source.Read(buf[:])

        r.Length += n

    }

    closeerr := r.Source.Close()

    if err != nil && err != io.EOF {

        return err

    }

    return closeerr

}

这将在您从流中读取字节数时计算字节数，关闭时它将首先消耗并计算所有剩余的未读字节数。完成流后，您可以访问长度。

Answer 3

选项1

使用TeeReader，这是可扩展的。它将阅读器分成两部分，其中之一使用分配的内存计算大小。另外，在第一种情况下

 maxmem := 4096

  var buf bytes.Buffer

  // comment this line out if you want to disable gathering metrics

  resp.Body = io.TeeReader(resp.Body, &buf) 

  readsize := func(r io.Reader) int {

    bytes := make([]byte, maxmem)

    var size int

      for {

        read, err := r.Read(bytes)

        if err == io.EOF {

        break

      }

      size += read

    }

    return size

  }

  log.Printf("Size is %d", readsize(&buf))

选项2不可扩展的方式（原始答案）

您可以只读取正文，计算大小，然后解组为结构，这样它就变成了：

    b, _ := ioutil.ReadAll(r.Body)

    size := len(b) // can be nil so check err in your app

    

    if err := json.Unmarshal(b, &input); err != nil {

        s.BadReq(w, errors.New("error reading body"))

        return

    }