/
cache.go
69 lines (55 loc) · 1.94 KB
/
cache.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
package client
import (
"github.com/kubeservice-stack/basa/src/backend/client/api"
"k8s.io/client-go/informers"
"k8s.io/client-go/kubernetes"
appsv1 "k8s.io/client-go/listers/apps/v1"
autoscalingv1 "k8s.io/client-go/listers/autoscaling/v1"
"k8s.io/client-go/listers/core/v1"
)
type CacheFactory struct {
stopChan chan struct{}
sharedInformerFactory informers.SharedInformerFactory
}
func (c ClusterManager) Close() {
close(c.CacheFactory.stopChan)
}
func buildCacheController(client *kubernetes.Clientset) (*CacheFactory, error) {
stop := make(chan struct{})
sharedInformerFactory := informers.NewSharedInformerFactory(client, defaultResyncPeriod)
rangeMap, err := api.GetResourceMap(client)
if err != nil {
return nil, err
}
// Start all Resources defined in KindToResourceMap
for _, value := range rangeMap {
genericInformer, err := sharedInformerFactory.ForResource(value.GroupVersionResourceKind.GroupVersionResource)
if err != nil {
return nil, err
}
go genericInformer.Informer().Run(stop)
}
sharedInformerFactory.Start(stop)
return &CacheFactory{
stopChan: stop,
sharedInformerFactory: sharedInformerFactory,
}, nil
}
func (c *CacheFactory) PodLister() v1.PodLister {
return c.sharedInformerFactory.Core().V1().Pods().Lister()
}
func (c *CacheFactory) EventLister() v1.EventLister {
return c.sharedInformerFactory.Core().V1().Events().Lister()
}
func (c *CacheFactory) DeploymentLister() appsv1.DeploymentLister {
return c.sharedInformerFactory.Apps().V1().Deployments().Lister()
}
func (c *CacheFactory) NodeLister() v1.NodeLister {
return c.sharedInformerFactory.Core().V1().Nodes().Lister()
}
func (c *CacheFactory) EndpointLister() v1.EndpointsLister {
return c.sharedInformerFactory.Core().V1().Endpoints().Lister()
}
func (c *CacheFactory) HPALister() autoscalingv1.HorizontalPodAutoscalerLister {
return c.sharedInformerFactory.Autoscaling().V1().HorizontalPodAutoscalers().Lister()
}