forked from kubevirt/containerized-data-importer
-
Notifications
You must be signed in to change notification settings - Fork 2
/
services.go
101 lines (93 loc) · 3.21 KB
/
services.go
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
package utils
import (
"github.com/pkg/errors"
"k8s.io/api/core/v1"
apierrs "k8s.io/apimachinery/pkg/api/errors"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
"k8s.io/apimachinery/pkg/util/wait"
"k8s.io/client-go/kubernetes"
"k8s.io/klog"
)
// GetServiceInNamespaceOrDie attempts to get the service in namespace `ns` by `name`. Returns pointer to service on
// success. Panics on error.
func GetServiceInNamespaceOrDie(c *kubernetes.Clientset, ns, name string) *v1.Service {
svc, err := GetServiceInNamespace(c, ns, name)
if err != nil || svc == nil {
klog.Fatal(err)
}
return svc
}
// GetServiceInNamespace retries get on service `name` in namespace `ns` until timeout or IsNotFound error. Ignores
// api errors that may be intermittent. Returns pointer to service (nil on error) or an error (nil on success)
func GetServiceInNamespace(c *kubernetes.Clientset, ns, name string) (*v1.Service, error) {
var svc *v1.Service
err := wait.PollImmediate(defaultPollInterval, defaultPollPeriod, func() (done bool, err error) {
svc, err = c.CoreV1().Services(ns).Get(name, metav1.GetOptions{})
// success
if err == nil {
return true, nil
}
// fail if the service does not exist
if apierrs.IsNotFound(err) {
return false, errors.Wrap(err, "Service not found")
}
// log non-fatal errors
klog.Error(errors.Wrapf(err, "Encountered non-fatal error getting service \"%s/%s\". retrying", ns, name))
return false, nil
})
return svc, err
}
// GetServicesInNamespaceByLabel retries get of services in namespace `ns` by `labelSelector` until timeout. Ignores
// api errors that may be intermittent. Returns pointer to ServiceList (nil on error) and an error (nil on success)
func GetServicesInNamespaceByLabel(c *kubernetes.Clientset, ns, labelSelector string) (*v1.ServiceList, error) {
var svcList *v1.ServiceList
err := wait.PollImmediate(defaultPollInterval, defaultPollPeriod, func() (done bool, err error) {
svcList, err = c.CoreV1().Services(ns).List(metav1.ListOptions{
LabelSelector: labelSelector,
})
// success
if err == nil {
return true, nil
}
// log non-fatal errors
klog.Error(errors.Wrapf(err, "Encountered non-fatal error getting service list in namespace %s", ns))
return false, nil
})
return svcList, err
}
// GetServicePortByName scans a service's ports for names matching `name`. Returns integer port value (0 on error) and
// error (nil on success).
func GetServicePortByName(svc *v1.Service, name string) (int, error) {
if svc == nil {
return 0, errors.New("nil service")
}
var port int
for _, p := range svc.Spec.Ports {
if p.Name == name {
port = int(p.Port)
break
}
}
if port == 0 {
return 0, errors.Errorf("port %q not found", name)
}
return port, nil
}
// GetServiceNodePortByName scans a service's nodePorts for a name matching the `name` parameter and returns
// the associated port integer or an error if not match is found.
func GetServiceNodePortByName(svc *v1.Service, name string) (int, error) {
if svc == nil {
return 0, errors.New("nil service")
}
var port int
for _, p := range svc.Spec.Ports {
if p.Name == name {
port = int(p.NodePort)
break
}
}
if port == 0 {
return 0, errors.Errorf("port %q not found", name)
}
return port, nil
}