23 lines
520 B
Go
23 lines
520 B
Go
|
package server
|
||
|
|
|
||
|
|
import "net/http"
|
||
|
|
|
||
|
|
type Directive func(args ...string) Middleware
|
||
|
|
|
||
|
|
var Add_Header Directive = func(args ...string) Middleware {
|
||
|
|
return func(w http.ResponseWriter, r *http.Request, next func()) {
|
||
|
|
w.Header().Add(args[0], args[1])
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
var Set_Header Directive = func(args ...string) Middleware {
|
||
|
|
return func(w http.ResponseWriter, r *http.Request, next func()) {
|
||
|
|
w.Header().Set(args[0], args[1])
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
var DirectiveMap = map[string]Directive{
|
||
|
|
"Set-Header": Set_Header,
|
||
|
|
"Add-Header": Add_Header,
|
||
|
|
}
|