| /* |
| Copyright 2016 The Kubernetes Authors. |
| |
| Licensed under the Apache License, Version 2.0 (the "License"); |
| you may not use this file except in compliance with the License. |
| You may obtain a copy of the License at |
| |
| http://www.apache.org/licenses/LICENSE-2.0 |
| |
| Unless required by applicable law or agreed to in writing, software |
| distributed under the License is distributed on an "AS IS" BASIS, |
| WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| See the License for the specific language governing permissions and |
| limitations under the License. |
| */ |
| |
| package filters |
| |
| import ( |
| "fmt" |
| "net/http" |
| |
| "k8s.io/apiserver/pkg/endpoints/handlers/responsewriters" |
| "k8s.io/apiserver/pkg/endpoints/request" |
| ) |
| |
| // WithRequestInfo attaches a RequestInfo to the context. |
| func WithRequestInfo(handler http.Handler, resolver request.RequestInfoResolver) http.Handler { |
| return http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) { |
| ctx := req.Context() |
| info, err := resolver.NewRequestInfo(req) |
| if err != nil { |
| responsewriters.InternalError(w, req, fmt.Errorf("failed to create RequestInfo: %v", err)) |
| return |
| } |
| |
| req = req.WithContext(request.WithRequestInfo(ctx, info)) |
| |
| handler.ServeHTTP(w, req) |
| }) |
| } |