| package middleware |
| |
| import ( |
| "net/http" |
| |
| "golang.org/x/net/context" |
| ) |
| |
| // ExperimentalMiddleware is a the middleware in charge of adding the |
| // 'Docker-Experimental' header to every outgoing request |
| type ExperimentalMiddleware struct { |
| experimental string |
| } |
| |
| // NewExperimentalMiddleware creates a new ExperimentalMiddleware |
| func NewExperimentalMiddleware(experimentalEnabled bool) ExperimentalMiddleware { |
| if experimentalEnabled { |
| return ExperimentalMiddleware{"true"} |
| } |
| return ExperimentalMiddleware{"false"} |
| } |
| |
| // WrapHandler returns a new handler function wrapping the previous one in the request chain. |
| func (e ExperimentalMiddleware) WrapHandler(handler func(ctx context.Context, w http.ResponseWriter, r *http.Request, vars map[string]string) error) func(ctx context.Context, w http.ResponseWriter, r *http.Request, vars map[string]string) error { |
| return func(ctx context.Context, w http.ResponseWriter, r *http.Request, vars map[string]string) error { |
| w.Header().Set("Docker-Experimental", e.experimental) |
| return handler(ctx, w, r, vars) |
| } |
| } |