Go函数中中间资源的defer关闭处理方案咨询

一、jsonBodyReader的资源处理 #

关于r.Body的关闭 #

http.Request的Body属于请求处理的上层责任，应该由调用jsonBodyReader的父函数（比如HTTP Handler）来关闭，比如在Handler中写defer r.Body.Close()。这个函数内部不要尝试关闭r.Body，否则会导致上层无法正常读取或触发重复关闭错误。

关于gzipReader的关闭 #

如果直接在函数里defer gzipReader.Close()，函数返回后defer会立即执行，关闭gzipReader，导致返回的io.LimitReader因为底层读取器已关闭而无法使用。解决办法是把gzipReader包装成一个带Close方法的自定义Reader，让调用者关闭返回的Reader时顺便清理gzipReader：

type gzipLimitReader struct {
    io.Reader
    gz *gzip.Reader
}

// 实现io.ReadCloser的Close方法，关闭内部的gzipReader
func (g *gzipLimitReader) Close() error {
    return g.gz.Close()
}

func jsonBodyReader(config Config, r *http.Request) (io.ReadCloser, error) {
    bodyReader := io.LimitReader(r.Body, config.Reports.Max.Body)

    gzipReader, err := gzip.NewReader(bodyReader)
    if err != nil {
        return nil, err
    }

    limitReader := io.LimitReader(gzipReader, config.Reports.Max.Json)
    // 返回自定义的ReadCloser，让调用者负责关闭
    return &gzipLimitReader{Reader: limitReader, gz: gzipReader}, nil
}

调用方使用时，只需要关闭返回的io.ReadCloser即可：

reader, err := jsonBodyReader(cfg, req)
if err != nil {
    // 处理错误
}
defer reader.Close()
// 读取reader内容

结论：不需要返回所有中间资源，通过包装成带Close方法的自定义类型，就能让调用方一次性完成资源清理。

二、save函数的文件关闭优化 #

当前方案需要返回*os.File让父函数关闭，其实可以通过两种方式避免：

方案1：使用回调函数限制资源作用域 #

把需要处理TeeReader的逻辑通过回调传入save函数，让save自己负责关闭文件：

func save(target string, reader io.Reader, fn func(io.Reader) error) error {
    file, err := os.Create(target)
    if err != nil {
        return err
    }
    // 函数退出时自动关闭文件，不管回调执行成功还是失败
    defer file.Close()

    teeReader := io.TeeReader(reader, file)
    // 执行回调逻辑，处理teeReader
    return fn(teeReader)
}

调用示例：

err := save("output.txt", inputReader, func(r io.Reader) error {
    // 这里写处理teeReader的逻辑，比如读取内容到其他地方
    _, err := io.Copy(os.Stdout, r)
    return err
})
if err != nil {
    // 处理错误
}

这种方式下，文件的创建和关闭完全由save函数管理，调用方不用关心文件资源。

方案2：自定义ReadCloser包装TeeReader和文件 #

和第一个函数的思路类似，把TeeReader和文件包装成一个实现io.ReadCloser的类型：

type teeFileReader struct {
    io.Reader
    file *os.File
}

func (t *teeFileReader) Close() error {
    return t.file.Close()
}

func save(target string, reader io.Reader) (io.ReadCloser, error) {
    file, err := os.Create(target)
    if err != nil {
        return nil, err
    }

    teeReader := io.TeeReader(reader, file)
    return &teeFileReader{Reader: teeReader, file: file}, nil
}

调用方只需要关闭返回的io.ReadCloser：

reader, err := save("output.txt", inputReader)
if err != nil {
    // 处理错误
}
defer reader.Close()
// 读取reader内容，同时内容会被写入文件

内容的提问来源于stack exchange，提问作者user2025261