如何在go fasthttp中添加自定义http头部？

Question

我使用的是fasthttp服务器https://github.com/valyala/fasthttp

我需要为所有请求添加自定义标头

Access-Control-Allow-Origin: *

我该怎么做呢？

Answer 1

因为它是一个响应头，所以我假设你是这个意思：

ctx.Response.Header.Set("Access-Control-Allow-Origin", "*")

Answer 2

不使用Context时的另一种选择

func setResponseHeader(h http.HandlerFunc) http.HandlerFunc {
    return func(w http.ResponseWriter, r *http.Request) {
        w.Header().Set("Access-Control-Allow-Origin", "*")
        h.ServeHTTP(w, r)
    }
}

setResponseHeader本质上是参数HandlerFunc h的装饰者。当你组装你的路由时，你可以这样做：

http.HandleFunc("/api/endpoint", setResponseHeader(myHandlerFunc)) 
http.ListenAndServe(":8000", nil)       

Answer 3

要在fasthttp上启用CORS支持，最好使用fasthttpcors包。

import (
    ...
    cors "github.com/AdhityaRamadhanus/fasthttpcors"
    ...
)

func main() {
    ...

    withCors := cors.NewCorsHandler(cors.Options{
        AllowMaxAge: math.MaxInt32,
    })

    log.Fatal(fasthttp.ListenAndServe(":8080", withCors.CorsMiddleware(router.HandleRequest)))
}