当前位置:

首页 > 编程开发 > Golang实现简单的请求日志中间件

Golang实现简单的请求日志中间件

GoHTTP日志中间件需包装ResponseWriter捕获状态码,用time.Now和defer统计耗时,通过X-Forwarded-For等头获取真实IP并校验,日志异步化防阻塞,避免序列化大body。

本文目录

    Go HTTP日志中间件需包装ResponseWriter捕获状态码,用time.Now和defer统计耗时,通过X-Forwarded-For等头获取真实IP并校验,日志异步化防阻塞,避免序列化大body。

    Golang实现简单的请求日志中间件

    Go HTTP 中间件怎么记录请求基础信息

    直接用 http.Handler 包一层就能实现日志中间件,不需要第三方库。核心是拦截 http.ResponseWriter,在写入响应前记录状态码、耗时、路径等。

    注意:不能直接读取原始 ResponseWriter 的状态码,得用自定义结构体包装它,重写 WriteHeader 方法来捕获实际返回的状态码。

    • 必须包装 ResponseWriter,否则拿不到真实 StatusCode
    • 用 time.Now() 记开始时间,defer 算耗时,避免漏统计
    • 日志建议用 log.Printf 或结构化日志库(如 zerolog),别用 fmt.Println

    如何获取客户端真实 IP(尤其 Nginx 反向代理后)

    直接读 r.RemoteAddr 在有反向代理时会得到 127.0.0.1:xxxx,不可用。应优先检查 X-Forwarded-For 头,但要注意该头可被伪造,生产环境需配合 X-Real-IP 或信任的代理 IP 段做校验。

    • r.Header.Get("X-Forwarded-For") 返回逗号分隔的 IP 列表,取第一个(最左)通常是客户端真实 IP
    • 若使用 Nginx,确保配置了 proxy_set_header X-Forwarded-For $remote_addr;
    • 更安全的做法:只信任内网 IP 发来的 X-Forwarded-For,其他情况回落到 r.RemoteAddr 解析出 IP

    中间件里怎么避免阻塞主处理逻辑

    日志本身是 I/O 操作,但同步写文件或网络日志服务可能拖慢请求。简单场景下同步写 os.Stderr 或 log.Logger 是可接受的;高并发时建议异步提交日志(例如发到 channel 由 goroutine 消费)。

    • 不要在中间件里调用 time.Sleep、http.Get 等阻塞操作
    • 避免在日志中序列化大体积请求体(r.Body)或响应体——默认不读、不复制
    • 如果真要记录 body,需用 io.TeeReader / io.MultiWriter 做透传,且限制最大长度(防 OOM)
    func loggingMiddleware(next http.Handler) http.Handler {
    	return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
    		start := time.Now()
    		lw := &responseWriter{ResponseWriter: w, statusCode: http.StatusOK}
    		
    		next.ServeHTTP(lw, r)
    		
    		ip := getClientIP(r)
    		duration := time.Since(start).Microseconds()
    		
    		log.Printf(
    			"[%s] %s %s %d %dµs %s",
    			time.Now().Format("2006-01-02T15:04:05Z"),
    			r.Method,
    			r.URL.Path,
    			lw.statusCode,
    			duration,
    			ip,
    		)
    	})
    }
    
    type responseWriter struct {
    	http.ResponseWriter
    	statusCode int
    }
    
    func (rw *responseWriter) WriteHeader(code int) {
    	rw.statusCode = code
    	rw.ResponseWriter.WriteHeader(code)
    }
    
    func getClientIP(r *http.Request) string {
    	if ip := r.Header.Get("X-Forwarded-For"); ip != "" {
    		return strings.TrimSpace(strings.Split(ip, ",")[0])
    	}
    	if ip := r.Header.Get("X-Real-IP"); ip != "" {
    		return ip
    	}
    	return strings.Split(r.RemoteAddr, ":")[0]
    }

    注意 responseWriter 没有实现 Flush、Hijack 等方法,如果下游 handler 用到了流式响应(比如 SSE),需要补全这些接口,否则 panic。

    本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系bd@zhengruan.com
    作者最新文章
    编程开发
    相关文章 更多
    GitHub六个AI开源项目能做什么?软件理解、语音与求职场景
    GitHub六个AI开源项目能做什么?软件理解、语音与求职场景

    解析六个GitHub AI开源项目的具体用途,包括软件静态分析、智能体回答检查、多语言配音、求职材料生成、本地代码检索及动态歌词展示,帮助开发者根据实际需求筛选工具并正确验证结果。

    opencode安装skills教程及本地配置方法
    opencode安装skills教程及本地配置方法

    遇到OpenCode Skills无法加载或配置无效?本文手把手教你完成Skills的安装、环境变量设置及本地调试,避开常见坑点,让AI助手更懂你的代码习惯。

    如何开启Visual Studio2022实时代码分析警告提示
    如何开启Visual Studio2022实时代码分析警告提示

    解决 Visual Studio 2022 编辑时不显示代码分析警告的问题。指南涵盖 .NET 项目实时分析设置、Roslyn 分析器范围调整及错误列表来源筛选,确保开发者能高效利用实时代码诊断功能。

    如何在Visual Studio2022查看变量监视窗口调试程序
    如何在Visual Studio2022查看变量监视窗口调试程序

    详解 Visual Studio 2022 调试技巧:包括断点设置、监视窗口开启方法、变量添加步骤以及通过 F10 单步执行观察变量值变化的全过程。

    Visual Studio2017设置默认编码为UTF-8不带BOM
    Visual Studio2017设置默认编码为UTF-8不带BOM

    针对 Visual Studio 2017 缺乏全局默认编码选项的情况,详解如何利用 .editorconfig 设定项目级 UTF-8 无 BOM 规则,以及手动转换现有文件并校验字节的方法。

    Visual Studio2022F5调试提示启动项目未配置怎么修复
    Visual Studio2022F5调试提示启动项目未配置怎么修复

    解决 Visual Studio 2022 F5 调试报错问题,通过设置启动项目、检查解决方案属性及正确区分类库与应用项目,快速恢复调试功能。

    Visual Studio2022NuGet包还原失败离线处理方案
    Visual Studio2022NuGet包还原失败离线处理方案

    解决Visual Studio 2022离线环境中NuGet包还原失败的难题,涵盖依赖链分析、本地源配置、包集收集及隔离验证步骤,帮助开发者构建稳定的离线开发环境。

    Visual Studio2017大项目加载缓慢,优化加载性能技巧
    Visual Studio2017大项目加载缓慢,优化加载性能技巧

    解决Visual Studio 2017大型项目加载慢的实用指南。涵盖性能监测、扩展管理、文档恢复优化及版本兼容性检查,帮助开发者精准定位瓶颈并提升工作效率。

    Visual Studio2022设置多启动项目同时调试运行
    Visual Studio2022设置多启动项目同时调试运行

    详解 Visual Studio 2022 多启动项目配置方法,包括启动动作选择、顺序调整及调试技巧,帮助开发者高效调试分布式或前后端分离应用。

    fastapi下载文件接口实现教程
    fastapi下载文件接口实现教程

    深入解析在FastAPI框架中实现文件下载接口的完整方案,涵盖基于FileResponse的本地静态文件传输、基于StreamingResponse的分块流式传输、RFC 5987中文文件名编码规范、路径穿越安全防御及BackgroundTasks临时文件清理,并配备详细代码示例与架构解析图。

    查看更多
    精品专题 更多
    装机必备
    装机必备

    正软商城装机必备专区,精选办公、浏览器、安全防护、影音播放、压缩解压、设计创作和系统工具等电脑常用正版软件,帮助用户快速完成新电脑软件配置。

    Windows
    Windows

    正软商城Windows软件专区,汇集适用于Windows电脑的办公、设计、安全防护、影音播放、开发工具和系统优化软件,提供软件介绍、系统要求、正版授权及购买下载服务。

    PDF教程
    PDF教程

    正软商城PDF教程频道提供PDF编辑、转换、合并、拆分、压缩及格式处理方法,同时介绍常用PDF软件和工具的使用技巧。

    Mac软件 更多
    讯飞智作 网页版
    讯飞智作 网页版
    Windows/ macOS/ Linux

    讯飞智作是一款基于语音合成技术、虚拟人驱动技术与数字人形象声音构建能力的 AI 数字人音视频创作平台。

    火山引擎 网页版
    火山引擎 网页版
    Windows/ macOS/ Linux

    火山引擎是字节跳动旗下企业级云与AI服务平台。聚焦大模型核心能力,依托字节自研豆包多模态大模型,凭借火山方舟一站式平台,提供模型精调、推理部署、智能应用搭建等全链路服务,结合高性能AI算力,助力各行业快速落地大模型数字化与智能化应用。

    Aperty macOS版
    Aperty macOS版
    macOS

    Aperty 是 Skylum 推出的 AI 人像修图软件,支持自然肤质修饰、瑕疵处理、面部与身体调整、预设及批量同步。

    WINDOWS 更多
    讯飞智作 网页版
    讯飞智作 网页版
    Windows/ macOS/ Linux

    讯飞智作是一款基于语音合成技术、虚拟人驱动技术与数字人形象声音构建能力的 AI 数字人音视频创作平台。

    火山引擎 网页版
    火山引擎 网页版
    Windows/ macOS/ Linux

    火山引擎是字节跳动旗下企业级云与AI服务平台。聚焦大模型核心能力,依托字节自研豆包多模态大模型,凭借火山方舟一站式平台,提供模型精调、推理部署、智能应用搭建等全链路服务,结合高性能AI算力,助力各行业快速落地大模型数字化与智能化应用。

    PixTV 网页版
    PixTV 网页版
    Windows

    PixTV 是一款在浏览器中使用的 AI 视觉内容创作平台,提供图片、视频、音频、文字生成、AI 分镜、局部重绘、提示词反推和无限画布等工具,适合短剧、漫剧、广告及社交媒体内容制作。