79 lines
1.7 KiB
Go
79 lines
1.7 KiB
Go
package k8s
|
|
|
|
import (
|
|
v1 "k8s.io/api/core/v1"
|
|
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
|
|
restclient "k8s.io/client-go/rest"
|
|
)
|
|
|
|
const defaultKillGrace int64 = 5
|
|
|
|
// Pod represents a Kubernetes resource.
|
|
type Pod struct {
|
|
Connection
|
|
}
|
|
|
|
// NewPod returns a new Pod.
|
|
func NewPod(c Connection) Cruder {
|
|
return &Pod{c}
|
|
}
|
|
|
|
// Get a pod.
|
|
func (p *Pod) Get(ns, name string) (interface{}, error) {
|
|
return p.DialOrDie().CoreV1().Pods(ns).Get(name, metav1.GetOptions{})
|
|
}
|
|
|
|
// List all pods in a given namespace.
|
|
func (p *Pod) List(ns string) (Collection, error) {
|
|
rr, err := p.DialOrDie().CoreV1().Pods(ns).List(metav1.ListOptions{})
|
|
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 pod.
|
|
func (p *Pod) Delete(ns, n string) error {
|
|
// Make pods die faster?
|
|
grace := defaultKillGrace
|
|
return p.DialOrDie().CoreV1().Pods(ns).Delete(n, &metav1.DeleteOptions{
|
|
GracePeriodSeconds: &grace,
|
|
})
|
|
}
|
|
|
|
// Containers returns all container names on pod
|
|
func (p *Pod) Containers(ns, n string, includeInit bool) ([]string, error) {
|
|
po, err := p.DialOrDie().CoreV1().Pods(ns).Get(n, metav1.GetOptions{})
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
cc := []string{}
|
|
for _, c := range po.Spec.Containers {
|
|
cc = append(cc, c.Name)
|
|
}
|
|
|
|
if includeInit {
|
|
for _, c := range po.Spec.InitContainers {
|
|
cc = append(cc, c.Name)
|
|
}
|
|
}
|
|
|
|
return cc, nil
|
|
}
|
|
|
|
// Logs fetch container logs for a given pod and container.
|
|
func (p *Pod) Logs(ns, n, co string, lines int64, prev bool) *restclient.Request {
|
|
return p.DialOrDie().CoreV1().Pods(ns).GetLogs(n, &v1.PodLogOptions{
|
|
Container: co,
|
|
Follow: true,
|
|
TailLines: &lines,
|
|
Previous: prev,
|
|
})
|
|
}
|