083f6dee05
* Refactor provider init This moves provider init out of vkubelet setup, instead preferring to initialize vkubelet with a provider. * Split API server configuration from setup. This makes sure that configuration (which is done primarily through env vars) is separate from actually standing up the servers. This also makes sure to abort daemon initialization if the API servers are not able to start.
88 lines
3.1 KiB
Go
88 lines
3.1 KiB
Go
package vkubelet
|
|
|
|
import (
|
|
"context"
|
|
"net"
|
|
"net/http"
|
|
|
|
"github.com/Sirupsen/logrus"
|
|
"github.com/gorilla/mux"
|
|
"github.com/virtual-kubelet/virtual-kubelet/log"
|
|
"github.com/virtual-kubelet/virtual-kubelet/providers"
|
|
"github.com/virtual-kubelet/virtual-kubelet/vkubelet/api"
|
|
)
|
|
|
|
// PodHandler creates an http handler for interacting with pods/containers.
|
|
func PodHandler(p providers.Provider) http.Handler {
|
|
r := mux.NewRouter()
|
|
r.HandleFunc("/containerLogs/{namespace}/{pod}/{container}", api.PodLogsHandlerFunc(p)).Methods("GET")
|
|
r.HandleFunc("/exec/{namespace}/{pod}/{container}", api.PodExecHandlerFunc(p)).Methods("POST")
|
|
r.NotFoundHandler = http.HandlerFunc(NotFound)
|
|
return r
|
|
}
|
|
|
|
// MetricsSummaryHandler creates an http handler for serving pod metrics.
|
|
//
|
|
// If the passed in provider does not implement providers.PodMetricsProvider,
|
|
// it will create handlers that just serves http.StatusNotImplemented
|
|
func MetricsSummaryHandler(p providers.Provider) http.Handler {
|
|
r := mux.NewRouter()
|
|
|
|
mp, ok := p.(providers.PodMetricsProvider)
|
|
if !ok {
|
|
r.HandleFunc("/stats/summary", NotImplemented).Methods("GET")
|
|
r.HandleFunc("/stats/summary/", NotImplemented).Methods("GET")
|
|
} else {
|
|
r.HandleFunc("/stats/summary", api.PodMetricsHandlerFunc(mp)).Methods("GET")
|
|
r.HandleFunc("/stats/summary/", api.PodMetricsHandlerFunc(mp)).Methods("GET")
|
|
}
|
|
r.NotFoundHandler = http.HandlerFunc(NotFound)
|
|
return r
|
|
}
|
|
|
|
// KubeletServerStart starts the virtual kubelet HTTP server.
|
|
func KubeletServerStart(p providers.Provider, l net.Listener, cert, key string) {
|
|
if err := http.ServeTLS(l, InstrumentHandler(PodHandler(p)), cert, key); err != nil {
|
|
log.G(context.TODO()).WithError(err).Error("error setting up http server")
|
|
}
|
|
}
|
|
|
|
// MetricsServerStart starts an HTTP server on the provided addr for serving the kubelset summary stats API.
|
|
func MetricsServerStart(p providers.Provider, l net.Listener) {
|
|
if err := http.Serve(l, InstrumentHandler(MetricsSummaryHandler(p))); err != nil {
|
|
log.G(context.TODO()).WithError(err).Error("Error starting http server")
|
|
}
|
|
}
|
|
|
|
func instrumentRequest(r *http.Request) context.Context {
|
|
ctx := r.Context()
|
|
logger := log.G(ctx).WithFields(logrus.Fields{
|
|
"uri": r.RequestURI,
|
|
"vars": mux.Vars(r),
|
|
})
|
|
return log.WithLogger(ctx, logger)
|
|
}
|
|
|
|
// InstrumentHandler wraps an http.Handler and injects instrumentation into the request context.
|
|
func InstrumentHandler(h http.Handler) http.Handler {
|
|
return http.HandlerFunc(func(w http.ResponseWriter, req *http.Request) {
|
|
ctx := instrumentRequest(req)
|
|
req = req.WithContext(ctx)
|
|
h.ServeHTTP(w, req)
|
|
})
|
|
}
|
|
|
|
// NotFound provides a handler for cases where the requested endpoint doesn't exist
|
|
func NotFound(w http.ResponseWriter, r *http.Request) {
|
|
logger := log.G(r.Context())
|
|
log.Trace(logger, "404 request not found")
|
|
http.Error(w, "404 request not found", http.StatusNotFound)
|
|
}
|
|
|
|
// NotImplemented provides a handler for cases where a provider does not implement a given API
|
|
func NotImplemented(w http.ResponseWriter, r *http.Request) {
|
|
logger := log.G(r.Context())
|
|
log.Trace(logger, "501 not implemented")
|
|
http.Error(w, "501 not implemented", http.StatusNotImplemented)
|
|
}
|