/
service_resource.go
70 lines (59 loc) · 1.72 KB
/
service_resource.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
package k8s
import (
"context"
"fmt"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
"k8s.io/apimachinery/pkg/runtime"
"k8s.io/apimachinery/pkg/watch"
corev1 "k8s.io/client-go/kubernetes/typed/core/v1"
"k8s.io/client-go/tools/cache"
)
type ServiceResource struct {
Client *corev1.CoreV1Client
}
func NewServiceResource(client *corev1.CoreV1Client) *ServiceResource {
return &ServiceResource{
Client: client,
}
}
func (sr *ServiceResource) Watch(ctx context.Context, namespace string, name string, labelSelector string) (chan runtime.Object, error) {
serviceClient := sr.Client.Services(namespace)
services, err := serviceClient.List(ctx, metav1.ListOptions{
FieldSelector: fmt.Sprintf("metadata.name=%s", name),
LabelSelector: labelSelector,
})
if err != nil {
return nil, fmt.Errorf("failed to list services: %w", err)
}
serviceInformer := cache.NewSharedIndexInformer(
&cache.ListWatch{
ListFunc: func(options metav1.ListOptions) (runtime.Object, error) {
return serviceClient.List(ctx, options)
},
WatchFunc: func(options metav1.ListOptions) (watch.Interface, error) {
return serviceClient.Watch(ctx, options)
},
},
&corev1.Service{},
0,
cache.Indexers{},
)
serviceChan := make(chan runtime.Object)
serviceInformer.AddEventHandler(cache.ResourceEventHandlerFuncs{
AddFunc: func(obj interface{}) {
serviceChan <- obj.(runtime.Object)
},
UpdateFunc: func(oldObj, newObj interface{}) {
serviceChan <- newObj.(runtime.Object)
},
DeleteFunc: func(obj interface{}) {
serviceChan <- obj.(runtime.Object)
},
})
stop := make(chan struct{})
go serviceInformer.Run(stop)
return serviceChan, nil
}
func (sr *ServiceResource) IsReady(resource runtime.Object) bool {
return true
}