Fiber框架编写自定义中间件拦截特定请求-Golang-PHP中文网

Fiber中间件不能多次读取ctx.Body()，因其底层依赖Go HTTP的单次可读io.ReadCloser；首次调用即消费流，后续读取返回空或panic，需通过io.ReadAll缓存并ResetBody重置。

fiber框架编写自定义中间件拦截特定请求

为什么 Fiber 的中间件不能直接读取 `ctx.Body()` 多次？

Fiber 默认的 ctx.Body() 底层调用的是 http.Request.Body，而 HTTP 请求体是单次可读流（io.ReadCloser）。一旦你调用一次 ctx.Body() 或 ctx.BodyParser()，底层的 Body 就被消费完毕；后续再调用会返回空字节或 panic（取决于是否已关闭）。

这不是 Fiber 的 bug，而是 Go HTTP 标准库的设计约束。常见错误现象包括：

中间件里解析了一次 JSON，Controller 里再解析就得到空结构体或 EOF 错误
日志中间件打印了 body，但下游 handler 拿不到原始数据
使用 ctx.FormValue() 后，ctx.Body() 返回空

解决办法是提前缓存请求体——用 ctx.Request().Body 读一次，存到 ctx.Locals，再用 bytes.NewReader 重置给后续使用。

如何写一个只拦截 `/api/v1/submit` 的中间件并校验 JSON 字段？

不要在中间件里写完整业务逻辑，聚焦「拦截 + 初筛」。下面这个中间件只对匹配路径且 Content-Type 是 application/json 的请求做字段检查，其他一律放行：

func ValidateSubmitMiddleware() fiber.Handler {
	return func(c *fiber.Ctx) error {
		// 只拦截指定路径和类型
		if c.Path() != "/api/v1/submit" || c.Get("Content-Type") != "application/json" {
			return c.Next()
		}

		// 缓存并读取 body
		body, err := io.ReadAll(c.Request().Body)
		if err != nil {
			return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{"error": "invalid request body"})
		}
		// 重置 Body，供后续 handler 使用
		c.Request().ResetBody()
		c.Request().SetBodyRaw(body)

		// 解析并校验关键字段
		var req struct {
			UserID  string `json:"user_id"`
			Payload string `json:"payload"`
		}
		if err := json.Unmarshal(body, &req); err != nil {
			return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{"error": "invalid JSON format"})
		}
		if req.UserID == "" || len(req.Payload) > 10240 {
			return c.Status(fiber.StatusBadRequest).JSON(fiber.Map{"error": "user_id required or payload too large"})
		}

		// 写入 locals，供 controller 复用解析结果
		c.Locals("parsed_submit", req)
		return c.Next()
	}
}

注册方式：

app.Post("/api/v1/submit", ValidateSubmitMiddleware(), submitHandler)

中间件里怎么安全地修改响应内容？

Fiber 不允许在中间件中直接调用 c.Send() 或 c.JSON() 后还执行 c.Next()，否则会 panic：「cannot set header after sent」。正确做法是用 c.Context().Response().SetBodyString() 配合状态码控制，且必须终止链路：

Fiber框架 3.0.0

Fiber框架 3.0.0 版本源码包下载，适合从 Fiber v2 迁移到 v3、体验新绑定系统、HTTP 客户端、状态管理和迁移工具的开发者。

下载

如果要拦截并返回自定义响应，必须显式 return，不调用 c.Next()
想加 Header（比如防爬标记），可以用 c.Set("X-Processed-By", "validator")，它不会触发发送
若需重写整个响应体（如统一错误格式），建议用 c.Context().Response().SetStatusCode() + SetBodyString()，然后 return

错误示范：c.JSON(...); c.Next() → 必然报错；正确写法：return c.JSON(...)。