-
Notifications
You must be signed in to change notification settings - Fork 27
/
reconcile_triggerer.go
79 lines (68 loc) · 2.07 KB
/
reconcile_triggerer.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
package reconcile
import (
"sync"
"time"
vcapi "github.com/ibm/varnish-operator/api/v1alpha1"
"github.com/ibm/varnish-operator/pkg/logger"
"sigs.k8s.io/controller-runtime/pkg/event"
)
func NewReconcileTriggerer(logr *logger.Logger, reconcileChan chan<- event.GenericEvent) *ReconcileTriggerer {
return &ReconcileTriggerer{
logger: logr,
timetable: make(map[string]map[string]*time.Timer),
reconcileChan: reconcileChan,
}
}
type ReconcileTriggerer struct {
logger *logger.Logger
timetable map[string]map[string]*time.Timer
reconcileChan chan<- event.GenericEvent
sync.Mutex
}
func (q *ReconcileTriggerer) TriggerAfter(key string, triggerAfter time.Duration, instance *vcapi.VarnishCluster) {
namespacedName := instance.Namespace + "/" + instance.Name
q.logger.Debugf("Setting timer to trigger in %s", triggerAfter)
timer := time.AfterFunc(triggerAfter, func() {
q.Lock()
delete(q.timetable, namespacedName)
q.Unlock()
q.logger.Debugf("Triggering reconcile")
q.reconcileChan <- event.GenericEvent{
Object: instance,
}
})
q.Lock()
defer q.Unlock()
if timers, exists := q.timetable[namespacedName]; exists {
if timer, exists := timers[key]; exists {
timer.Stop()
delete(q.timetable, namespacedName)
}
}
if _, exists := q.timetable[namespacedName]; !exists {
q.timetable[namespacedName] = map[string]*time.Timer{}
}
q.timetable[namespacedName][key] = timer
}
func (q *ReconcileTriggerer) Stop(key string, instance *vcapi.VarnishCluster) {
namespacedName := instance.Namespace + "/" + instance.Name
if timers, exists := q.timetable[namespacedName]; exists {
if timer, exists := timers[key]; exists {
timer.Stop()
q.Lock()
delete(q.timetable, namespacedName)
q.Unlock()
}
}
}
func (q *ReconcileTriggerer) TimerExists(key string, service *vcapi.VarnishCluster) bool {
namespacedName := service.Namespace + "/" + service.Name
q.Lock()
defer q.Unlock()
if _, exists := q.timetable[namespacedName]; exists {
if _, timerExists := q.timetable[namespacedName][key]; timerExists {
return true
}
}
return false
}