forked from voyagermesh/voyager
-
Notifications
You must be signed in to change notification settings - Fork 0
/
daemonsets.go
66 lines (60 loc) · 2.06 KB
/
daemonsets.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
package operator
import (
acrt "github.com/appscode/go/runtime"
"github.com/appscode/log"
"github.com/appscode/voyager/api"
metav1 "k8s.io/apimachinery/pkg/apis/meta/v1"
"k8s.io/apimachinery/pkg/runtime"
"k8s.io/apimachinery/pkg/util/wait"
"k8s.io/apimachinery/pkg/watch"
apiv1 "k8s.io/client-go/pkg/api/v1"
extensions "k8s.io/client-go/pkg/apis/extensions/v1beta1"
"k8s.io/client-go/tools/cache"
)
// Blocks caller. Intended to be called as a Go routine.
func (op *Operator) WatchDaemonSets() {
defer acrt.HandleCrash()
lw := &cache.ListWatch{
ListFunc: func(opts metav1.ListOptions) (runtime.Object, error) {
return op.KubeClient.ExtensionsV1beta1().DaemonSets(apiv1.NamespaceAll).List(metav1.ListOptions{})
},
WatchFunc: func(options metav1.ListOptions) (watch.Interface, error) {
return op.KubeClient.ExtensionsV1beta1().DaemonSets(apiv1.NamespaceAll).Watch(metav1.ListOptions{})
},
}
_, ctrl := cache.NewInformer(lw,
&extensions.DaemonSet{},
op.SyncPeriod,
cache.ResourceEventHandlerFuncs{
DeleteFunc: func(obj interface{}) {
if daemon, ok := obj.(*extensions.DaemonSet); ok {
log.Infof("DaemonSet %s@%s deleted", daemon.Name, daemon.Namespace)
op.restoreDaemonSetIfRequired(daemon)
}
},
},
)
ctrl.Run(wait.NeverStop)
}
func (op *Operator) restoreDaemonSetIfRequired(daemon *extensions.DaemonSet) error {
if daemon.Annotations == nil {
return nil
}
// deleted resource have source reference
engress, err := op.findOrigin(daemon.ObjectMeta)
if err != nil {
return err
}
// Ingress Still exists, restore resource
log.Infof("DaemonSet %s@%s requires restoration", daemon.Name, daemon.Namespace)
daemon.SelfLink = ""
daemon.ResourceVersion = ""
// Old resource and annotations are missing so we need to add the annotations
if daemon.Annotations == nil {
daemon.Annotations = make(map[string]string)
}
daemon.Annotations[api.OriginAPISchema] = engress.APISchema()
daemon.Annotations[api.OriginName] = engress.Name
_, err = op.KubeClient.ExtensionsV1beta1().DaemonSets(daemon.Namespace).Create(daemon)
return err
}