Golang 中的字符串长度

本篇文章主要是结合我之前面试的各种经历和实战开发中遇到的问题解决经验整理的，希望这篇《Golang 中的字符串长度》对你有很大帮助！欢迎收藏，分享给更多的需要的朋友学习~

我在网上找不到任何关于此问题的有用信息。

我正在编写一个 REST API，我想记录请求正文的大小（以字节为单位）以获取指标。 Go net/http API 不直接提供这一点。 http.Request 确实有 Content-Length 字段，但该字段可以为空，否则客户端可能会发送错误数据。

有没有办法在中间件级别实现这一点？暴力方法是读取全文并检查大小。但如果我在中间件中这样做，处理程序将无法访问主体，因为它会被读取并关闭。

解决方案

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

您可以编写一个自定义 readcloser 来代理现有的 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
}

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

本篇关于《Golang 中的字符串长度》的介绍就到此结束啦，但是学无止境，想要了解学习更多关于Golang的相关知识，请关注golang学习网公众号！