7dd49516d8
This makes the concept of a `Provider` wholely implemented in the cli implementation in cmd/virtual-kubelet. It allows us to slim down the interfaces used in vkubelet (and vkubelet/api) to what is actually used there rather than a huge interface that is only there to serve the CLI's needs.
41 lines
1.1 KiB
Go
41 lines
1.1 KiB
Go
package api
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"net/http"
|
|
|
|
"github.com/cpuguy83/strongerrors"
|
|
"github.com/pkg/errors"
|
|
stats "k8s.io/kubernetes/pkg/kubelet/apis/stats/v1alpha1"
|
|
)
|
|
|
|
// PodStatsSummaryHandlerFunc defines the handler for getting pod stats summaries
|
|
type PodStatsSummaryHandlerFunc func(context.Context) (*stats.Summary, error)
|
|
|
|
// HandlePodStatsSummary makes an HTTP handler for implementing the kubelet summary stats endpoint
|
|
func HandlePodStatsSummary(h PodStatsSummaryHandlerFunc) http.HandlerFunc {
|
|
if h == nil {
|
|
return NotImplemented
|
|
}
|
|
return handleError(func(w http.ResponseWriter, req *http.Request) error {
|
|
stats, err := h(req.Context())
|
|
if err != nil {
|
|
if errors.Cause(err) == context.Canceled {
|
|
return strongerrors.Cancelled(err)
|
|
}
|
|
return errors.Wrap(err, "error getting status from provider")
|
|
}
|
|
|
|
b, err := json.Marshal(stats)
|
|
if err != nil {
|
|
return strongerrors.Unknown(errors.Wrap(err, "error marshalling stats"))
|
|
}
|
|
|
|
if _, err := w.Write(b); err != nil {
|
|
return strongerrors.Unknown(errors.Wrap(err, "could not write to client"))
|
|
}
|
|
return nil
|
|
})
|
|
}
|