基于Golang+Echo的文件下载状态追踪与连接中断处理方案咨询

针对你在Echo框架中同时处理静态文件下载和反向代理时的字节统计需求，以下是具体实现方案：

一、核心思路：自定义响应包装器（Counting Response Writer） #

不管是静态文件还是反向代理，核心都是通过包装echo.ResponseWriter，在响应数据流向客户端的过程中实时统计字节数，并捕获连接中断等异常。

1. 实现计数响应包装器 #

import (
    "io"
    "net/http"
    "sync/atomic"

    "github.com/labstack/echo/v4"
)

type CountingResponseWriter struct {
    echo.ResponseWriter
    bytesWritten uint64
    clientClosed bool
}

func (w *CountingResponseWriter) Write(b []byte) (int, error) {
    n, err := w.ResponseWriter.Write(b)
    atomic.AddUint64(&w.bytesWritten, uint64(n))
    // 捕获客户端断开连接类错误
    if err != nil && isClientDisconnectError(err) {
        w.clientClosed = true
    }
    return n, err
}

// 判断是否为客户端连接中断错误，可根据运行环境补充匹配规则
func isClientDisconnectError(err error) bool {
    if err == http.ErrHandlerTimeout {
        return true
    }
    // 适配底层连接重置等错误，比如syscall.ECONNRESET
    if err != nil && err != io.EOF {
        errStr := err.Error()
        return errStr == "connection reset by peer" || errStr == "client disconnected"
    }
    return false
}

func (w *CountingResponseWriter) BytesWritten() uint64 {
    return atomic.LoadUint64(&w.bytesWritten)
}

func (w *CountingResponseWriter) ClientClosed() bool {
    return w.clientClosed
}

二、静态文件下载场景的集成 #

在静态文件路由前添加中间件，用包装器替换默认响应写入器：

func CountDownloadBytesMiddleware(next echo.HandlerFunc) echo.HandlerFunc {
    return func(c echo.Context) error {
        crw := &CountingResponseWriter{ResponseWriter: c.Response()}
        c.Response().Writer = crw

        err := next(c)
        
        // 请求结束后判断：客户端中断或下载失败时通知微服务
        if crw.ClientClosed() || (c.Response().Status >= 400 && crw.BytesWritten() > 0) {
            // 异步调用避免阻塞当前请求
            go func(bytes uint64) {
                // 这里实现调用微服务的逻辑，比如发送POST请求
                // notifyMicroService(bytes)
            }(crw.BytesWritten())
        }
        return err
    }
}

// 注册路由时应用中间件
e := echo.New()
e.Use(CountDownloadBytesMiddleware)
e.Static("/downloads", "./static/files")

三、反向代理场景的集成 #

反向代理的响应最终会通过Echo的ResponseWriter输出，因此直接复用上面的中间件即可，无需额外修改代理逻辑：

import "net/http/httputil"
import "net/url"

// 注册代理路由
e.Any("/proxy/*", func(c echo.Context) error {
    targetURL, _ := url.Parse("http://upstream-service:8080")
    proxy := httputil.NewSingleHostReverseProxy(targetURL)
    // 直接使用Echo的响应对象，中间件会自动统计字节数
    proxy.ServeHTTP(c.Response(), c.Request())
    return nil
})

这种方式无需修改代理内部逻辑，所有响应字节都会经过计数包装器，同时能捕获客户端中断错误。

四、关键注意事项 #

• 异步通知：调用微服务的逻辑必须异步执行，避免因微服务响应慢阻塞当前请求上下文。
• 大文件适配：所有统计逻辑都是流式处理，不会将文件加载到内存，完全支持GB级文件下载。
• 并发安全：使用atomic包处理字节计数，避免多协程写入时的计数错误。
• 错误适配：isClientDisconnectError函数需要根据实际部署环境补充错误匹配规则，比如不同操作系统的连接重置错误码。

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