golang gin自定义中间件：自定义context.Request.Header

最新推荐文章于 2025-01-20 19:28:30 发布

学亮编程手记

最新推荐文章于 2025-01-20 19:28:30 发布

阅读量1.1k

点赞数

CC 4.0 BY-SA版权

分类专栏： go 文章标签： golang gin 开发语言

本文链接：https://blog.csdn.net/a772304419/article/details/126509048

go 专栏收录该内容

485 篇文章

订阅专栏

本文介绍了一个使用Gin框架实现的自定义中间件，该中间件为HTTP请求添加了一个唯一的请求ID，如果请求头中不存在该ID，则自动生成并记录。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

package main

import (
	"log"

	"github.com/gin-gonic/gin"
	"github.com/gofrs/uuid"
)

func RequestIdMiddleware() gin.HandlerFunc {
	return func(c *gin.Context) {
		id := c.Request.Header.Get("zxl")
		if id == "" {
			u, _ := uuid.NewV4()
			id = u.String()
			log.Println(id)
		}
		c.Writer.Header().Set("zxl", id)
		c.Next()
	}
}

func main() {
	r := gin.Default()
	r.Use(RequestIdMiddleware())
	r.GET("ping", func(c *gin.Context) {
		c.JSON(200, gin.H{
			"code": 0,
		})
	})
	r.Run()
}