错误场景下ctx.Response.Header.Set函数失效问题排查

我一眼就看出问题所在了——你遇到的是fasthttp里ctx.Error()方法的一个常见“陷阱”：这个方法会重置整个响应头，导致你之前设置的Access-Control-Allow-Origin被清空了。

先把你没写完的代码补全（方便大家理解场景）：

package main

import "fmt"
import "github.com/valyala/fasthttp"

func main() {
    requestHandler := func(ctx *fasthttp.RequestCtx) {
        ctx.Response.Header.Set("Access-Control-Allow-Origin", "*")
        switch string(ctx.Path()) {
        case "/foo":
            fmt.Fprintf(ctx, "Success result")
        case "/bar":
            ctx.Error("Unsupported path", fasthttp.StatusNotFound)
        }
    }
    fasthttp.ListenAndServe(":8080", requestHandler)
}

问题根源 #

当你调用ctx.Error(msg, statusCode)时，fasthttp内部会执行ctx.Response.Reset()，这个操作会清空所有已设置的响应头、响应体和状态码，然后再重新设置错误状态码和错误信息。所以你在调用ctx.Error()之前设置的CORS头就被完全覆盖了，自然不会出现在响应里。

三种解决方案 #

方案1：调用ctx.Error()后重新设置响应头 #

既然ctx.Error()会重置头，那我们在它之后再补一次头就行：

case "/bar":
    ctx.Error("Unsupported path", fasthttp.StatusNotFound)
    // 重新设置CORS头，覆盖重置后的空状态
    ctx.Response.Header.Set("Access-Control-Allow-Origin", "*")

方案2：手动构造错误响应（替代ctx.Error()） #

如果你不想依赖ctx.Error()的自动重置逻辑，可以自己手动设置状态码和响应内容，这样响应头不会被清空：

case "/bar":
    ctx.Response.SetStatusCode(fasthttp.StatusNotFound)
    // 这里的头不会被任何重置操作覆盖
    ctx.Response.Header.Set("Access-Control-Allow-Origin", "*")
    fmt.Fprintf(ctx, "Unsupported path")

方案3：用全局中间件统一处理CORS头（推荐） #

如果你的服务所有接口都需要CORS头，用中间件是最优雅的方式，避免在每个路由里重复写头的逻辑，同时确保错误场景也能正确返回头：

// 定义CORS中间件
func corsMiddleware(next fasthttp.RequestHandler) fasthttp.RequestHandler {
    return func(ctx *fasthttp.RequestCtx) {
        // 先执行请求处理逻辑（包括可能的ctx.Error调用）
        next(ctx)
        // 不管处理结果是成功还是错误，最后统一设置CORS头
        ctx.Response.Header.Set("Access-Control-Allow-Origin", "*")
    }
}

func main() {
    requestHandler := func(ctx *fasthttp.RequestCtx) {
        switch string(ctx.Path()) {
        case "/foo":
            fmt.Fprintf(ctx, "Success result")
        case "/bar":
            ctx.Error("Unsupported path", fasthttp.StatusNotFound)
        }
    }
    // 用中间件包裹原始handler
    fasthttp.ListenAndServe(":8080", corsMiddleware(requestHandler))
}

这个方案的好处是，不管你的请求处理逻辑里有没有调用ctx.Error()，中间件都会在最后帮你补上CORS头，完全不用在业务代码里操心。

验证效果 #

用curl测试/bar路径：

curl -I http://localhost:8080/bar

现在你会看到响应头里已经包含Access-Control-Allow-Origin: *了。

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