Custom Headers on Every Response With golangs httprouter

main-with-middleware.go

func main() {
  r := httprouter.New()
  // Health check
  r.GET(“/health”, HealthCheckHandler)
  // Authentication
  r.POST(“/token-auth”, HeaderMiddleware(TokenAuthHandler))
  r.POST(“/p”, HeaderMiddleware(SomeFormHandler))
  fmt.Println(“Starting server on :3030”)
  http.ListenAndServe(“:3030”, r)
}

func HeaderMiddleware(h httprouter.Handle) httprouter.Handle {
  return func(w http.ResponseWriter, r *http.Request, ps httprouter.Params) {
    w.Header().Set(“Access-Control-Allow-Origin”, “*”)
    h(w, r, ps)
  }
}