1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
|
package kube
import (
"fmt"
"github.com/sirupsen/logrus"
kapi "k8s.io/api/core/v1"
kapisnetworking "k8s.io/api/networking/v1"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
"k8s.io/apimachinery/pkg/labels"
"k8s.io/apimachinery/pkg/types"
"k8s.io/client-go/kubernetes"
)
// Interface represents the exported methods for dealing with getting/setting
// kubernetes resources
type Interface interface {
SetAnnotationOnPod(pod *kapi.Pod, key, value string) error
SetAnnotationOnNode(node *kapi.Node, key, value string) error
SetAnnotationOnNamespace(ns *kapi.Namespace, key, value string) error
GetAnnotationsOnPod(namespace, name string) (map[string]string, error)
GetPod(namespace, name string) (*kapi.Pod, error)
GetPods(namespace string) (*kapi.PodList, error)
GetPodsByLabels(namespace string, selector labels.Selector) (*kapi.PodList, error)
GetNodes() (*kapi.NodeList, error)
GetNode(name string) (*kapi.Node, error)
GetService(namespace, name string) (*kapi.Service, error)
GetEndpoints(namespace string) (*kapi.EndpointsList, error)
GetNamespace(name string) (*kapi.Namespace, error)
GetNamespaces() (*kapi.NamespaceList, error)
GetNetworkPolicies(namespace string) (*kapisnetworking.NetworkPolicyList, error)
}
// Kube is the structure object upon which the Interface is implemented
type Kube struct {
KClient kubernetes.Interface
}
// SetAnnotationOnPod takes the pod object and key/value string pair to set it as an annotation
func (k *Kube) SetAnnotationOnPod(pod *kapi.Pod, key, value string) error {
logrus.Infof("Setting annotations %s=%s on pod %s", key, value, pod.Name)
patchData := fmt.Sprintf(`{"metadata":{"annotations":{"%s":"%s"}}}`, key, value)
_, err := k.KClient.CoreV1().Pods(pod.Namespace).Patch(pod.Name, types.MergePatchType, []byte(patchData))
if err != nil {
logrus.Errorf("Error in setting annotation on pod %s/%s: %v", pod.Name, pod.Namespace, err)
}
return err
}
// SetAnnotationOnNode takes the node object and key/value string pair to set it as an annotation
func (k *Kube) SetAnnotationOnNode(node *kapi.Node, key, value string) error {
logrus.Infof("Setting annotations %s=%s on node %s", key, value, node.Name)
patchData := fmt.Sprintf(`{"metadata":{"annotations":{"%s":"%s"}}}`, key, value)
_, err := k.KClient.CoreV1().Nodes().Patch(node.Name, types.MergePatchType, []byte(patchData))
if err != nil {
logrus.Errorf("Error in setting annotation on node %s: %v", node.Name, err)
}
return err
}
// SetAnnotationOnNamespace takes the Namespace object and key/value pair
// to set it as an annotation
func (k *Kube) SetAnnotationOnNamespace(ns *kapi.Namespace, key,
value string) error {
logrus.Infof("Setting annotations %s=%s on namespace %s", key, value,
ns.Name)
patchData := fmt.Sprintf(`{"metadata":{"annotations":{"%s":"%s"}}}`, key,
value)
_, err := k.KClient.CoreV1().Namespaces().Patch(ns.Name,
types.MergePatchType, []byte(patchData))
if err != nil {
logrus.Errorf("Error in setting annotation on namespace %s: %v",
ns.Name, err)
}
return err
}
// GetAnnotationsOnPod obtains the pod annotations from kubernetes apiserver, given the name and namespace
func (k *Kube) GetAnnotationsOnPod(namespace, name string) (map[string]string, error) {
pod, err := k.KClient.CoreV1().Pods(namespace).Get(name, metav1.GetOptions{})
if err != nil {
return nil, err
}
return pod.ObjectMeta.Annotations, nil
}
// GetPod obtains the Pod resource from kubernetes apiserver, given the name and namespace
func (k *Kube) GetPod(namespace, name string) (*kapi.Pod, error) {
return k.KClient.CoreV1().Pods(namespace).Get(name, metav1.GetOptions{})
}
// GetPods obtains the Pod resource from kubernetes apiserver, given the name and namespace
func (k *Kube) GetPods(namespace string) (*kapi.PodList, error) {
return k.KClient.CoreV1().Pods(namespace).List(metav1.ListOptions{})
}
// GetPodsByLabels obtains the Pod resources from kubernetes apiserver,
// given the namespace and label
func (k *Kube) GetPodsByLabels(namespace string, selector labels.Selector) (*kapi.PodList, error) {
options := metav1.ListOptions{}
options.LabelSelector = selector.String()
return k.KClient.CoreV1().Pods(namespace).List(options)
}
// GetNodes returns the list of all Node objects from kubernetes
func (k *Kube) GetNodes() (*kapi.NodeList, error) {
return k.KClient.CoreV1().Nodes().List(metav1.ListOptions{})
}
// GetNode returns the Node resource from kubernetes apiserver, given its name
func (k *Kube) GetNode(name string) (*kapi.Node, error) {
return k.KClient.CoreV1().Nodes().Get(name, metav1.GetOptions{})
}
// GetService returns the Service resource from kubernetes apiserver, given its name and namespace
func (k *Kube) GetService(namespace, name string) (*kapi.Service, error) {
return k.KClient.CoreV1().Services(namespace).Get(name, metav1.GetOptions{})
}
// GetEndpoints returns all the Endpoint resources from kubernetes
// apiserver, given namespace
func (k *Kube) GetEndpoints(namespace string) (*kapi.EndpointsList, error) {
return k.KClient.CoreV1().Endpoints(namespace).List(metav1.ListOptions{})
}
// GetNamespace returns the Namespace resource from kubernetes apiserver,
// given its name
func (k *Kube) GetNamespace(name string) (*kapi.Namespace, error) {
return k.KClient.CoreV1().Namespaces().Get(name, metav1.GetOptions{})
}
// GetNamespaces returns all Namespace resource from kubernetes apiserver
func (k *Kube) GetNamespaces() (*kapi.NamespaceList, error) {
return k.KClient.CoreV1().Namespaces().List(metav1.ListOptions{})
}
// GetNetworkPolicies returns all network policy objects from kubernetes
func (k *Kube) GetNetworkPolicies(namespace string) (*kapisnetworking.NetworkPolicyList, error) {
return k.KClient.NetworkingV1().NetworkPolicies(namespace).List(metav1.ListOptions{})
}
|
