feat: support CORS, better implementation (#1217)

* feat: support CORS, better implementation

* chore: refine code
This commit is contained in:
Kevin Wan
2021-11-09 20:35:57 +08:00
committed by GitHub
parent c1abe87953
commit 28409791fa
3 changed files with 78 additions and 22 deletions

View File

@@ -99,7 +99,7 @@ func ToMiddleware(handler func(next http.Handler) http.Handler) Middleware {
// WithCors returns a func to enable CORS for given origin, or default to all origins (*).
func WithCors(origin ...string) RunOption {
return func(server *Server) {
server.router.SetNotAllowedHandler(cors.Handler(origin...))
server.router.SetNotAllowedHandler(cors.NotAllowedHandler(origin...))
server.Use(cors.Middleware(origin...))
}
}