57 lines
1.1 KiB
Go
57 lines
1.1 KiB
Go
package watch
|
|
|
|
import (
|
|
"fmt"
|
|
|
|
"github.com/derailed/k9s/internal/k8s"
|
|
v1 "k8s.io/api/core/v1"
|
|
wv1 "k8s.io/client-go/informers/core/v1"
|
|
"k8s.io/client-go/tools/cache"
|
|
)
|
|
|
|
const (
|
|
// PodIndex marker for stored pods.
|
|
PodIndex string = "po"
|
|
podCols = 11
|
|
)
|
|
|
|
// Connection represents an client api server connection.
|
|
type Connection k8s.Connection
|
|
|
|
// Pod tracks pod activities.
|
|
type Pod struct {
|
|
cache.SharedIndexInformer
|
|
}
|
|
|
|
// NewPod returns a new pod.
|
|
func NewPod(c Connection, ns string) *Pod {
|
|
return &Pod{
|
|
SharedIndexInformer: wv1.NewPodInformer(c.DialOrDie(), ns, 0, cache.Indexers{}),
|
|
}
|
|
}
|
|
|
|
// List all pods from store in the given namespace.
|
|
func (p *Pod) List(ns string) k8s.Collection {
|
|
var res k8s.Collection
|
|
for _, o := range p.GetStore().List() {
|
|
pod := o.(*v1.Pod)
|
|
if ns == "" || pod.Namespace == ns {
|
|
res = append(res, pod)
|
|
}
|
|
}
|
|
return res
|
|
}
|
|
|
|
// Get retrieves a given pod from store.
|
|
func (p *Pod) Get(fqn string) (interface{}, error) {
|
|
o, ok, err := p.GetStore().GetByKey(fqn)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
if !ok {
|
|
return nil, fmt.Errorf("Pod %s not found", fqn)
|
|
}
|
|
|
|
return o, nil
|
|
}
|