60 lines
1.4 KiB
Go
60 lines
1.4 KiB
Go
package view
|
|
|
|
import (
|
|
"github.com/derailed/k9s/internal/dao"
|
|
"github.com/derailed/k9s/internal/render"
|
|
"github.com/derailed/k9s/internal/ui"
|
|
appsv1 "k8s.io/api/apps/v1"
|
|
"k8s.io/apimachinery/pkg/apis/meta/v1/unstructured"
|
|
"k8s.io/apimachinery/pkg/labels"
|
|
"k8s.io/apimachinery/pkg/runtime"
|
|
)
|
|
|
|
// StatefulSet represents a statefulset viewer.
|
|
type StatefulSet struct {
|
|
ResourceViewer
|
|
}
|
|
|
|
// NewStatefulSet returns a new viewer.
|
|
func NewStatefulSet(gvr dao.GVR) ResourceViewer {
|
|
s := StatefulSet{
|
|
ResourceViewer: NewRestartExtender(
|
|
NewScaleExtender(
|
|
NewLogsExtender(
|
|
NewGeneric(gvr),
|
|
func() string { return "" },
|
|
),
|
|
),
|
|
),
|
|
}
|
|
s.BindKeys()
|
|
s.GetTable().SetEnterFn(s.showPods)
|
|
s.GetTable().SetColorerFn(render.StatefulSet{}.ColorerFunc())
|
|
|
|
return &s
|
|
}
|
|
|
|
func (s *StatefulSet) BindKeys() {
|
|
s.Actions().Add(ui.KeyActions{
|
|
ui.KeyShiftD: ui.NewKeyAction("Sort Desired", s.GetTable().SortColCmd(1, true), false),
|
|
ui.KeyShiftC: ui.NewKeyAction("Sort Current", s.GetTable().SortColCmd(2, true), false),
|
|
})
|
|
}
|
|
|
|
func (s *StatefulSet) showPods(app *App, _, res, sel string) {
|
|
ns, n := namespaced(sel)
|
|
o, err := app.factory.Get(ns, s.GVR(), n, labels.Everything())
|
|
if err != nil {
|
|
app.Flash().Err(err)
|
|
return
|
|
}
|
|
|
|
var sts appsv1.StatefulSet
|
|
err = runtime.DefaultUnstructuredConverter.FromUnstructured(o.(*unstructured.Unstructured).Object, &sts)
|
|
if err != nil {
|
|
app.Flash().Err(err)
|
|
}
|
|
|
|
showPodsFromSelector(app, ns, sts.Spec.Selector)
|
|
}
|