k9s/internal/k8s/pdb.go

45 lines
1.1 KiB
Go

package k8s
import (
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
)
// PodDisruptionBudget represents a Kubernetes PodDisruptionBudget.
type PodDisruptionBudget struct {
*base
Connection
}
// NewPodDisruptionBudget returns a new PodDisruptionBudget.
func NewPodDisruptionBudget(c Connection) *PodDisruptionBudget {
return &PodDisruptionBudget{&base{}, c}
}
// Get a pdb.
func (p *PodDisruptionBudget) Get(ns, n string) (interface{}, error) {
return p.DialOrDie().PolicyV1beta1().PodDisruptionBudgets(ns).Get(n, metav1.GetOptions{})
}
// List all pdbs in a given namespace.
func (p *PodDisruptionBudget) List(ns string) (Collection, error) {
opts := metav1.ListOptions{
LabelSelector: p.labelSelector,
FieldSelector: p.fieldSelector,
}
rr, err := p.DialOrDie().PolicyV1beta1().PodDisruptionBudgets(ns).List(opts)
if err != nil {
return nil, err
}
cc := make(Collection, len(rr.Items))
for i, r := range rr.Items {
cc[i] = r
}
return cc, nil
}
// Delete a pdb.
func (p *PodDisruptionBudget) Delete(ns, n string) error {
return p.DialOrDie().PolicyV1beta1().PodDisruptionBudgets(ns).Delete(n, nil)
}