github-mirror/caddy
2
0
Fork 0
mirror of https://github.com/caddyserver/caddy.git synced 2025-05-24 19:49:56 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Refactored headers middleware to return errors

Browse Source
This commit is contained in:
Matthew Holt
2015-03-28 16:48:31 -06:00
parent 8f4e7f7fdc
commit a39e71ca26
2 changed files with 7 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
middleware/headers/headers.go
View File

@ -12,13 +12,13 @@ import (
// Headers is middleware that adds headers to the responses
// for requests matching a certain path.
type Headers struct {
Next http.HandlerFunc
Next middleware.HandlerFunc
Rules []HeaderRule
}
// ServeHTTP implements the http.Handler interface and serves the requests,
// ServeHTTP implements the middleware.Handler interface and serves requests,
// adding headers to the response according to the configured rules.
func (h Headers) ServeHTTP(w http.ResponseWriter, r *http.Request) {
func (h Headers) ServeHTTP(w http.ResponseWriter, r *http.Request) (int, error) {
for _, rule := range h.Rules {
if middleware.Path(r.URL.Path).Matches(rule.Url) {
for _, header := range rule.Headers {
@ -26,12 +26,12 @@ func (h Headers) ServeHTTP(w http.ResponseWriter, r *http.Request) {
}
}
}
h.Next(w, r)
return h.Next(w, r)
}
type (
// HeaderRule groups a slice of HTTP headers by a URL pattern.
// TODO: use http.Header type instead??
// TODO: use http.Header type instead?
HeaderRule struct {
Url string
Headers []Header