/
bootstrapper.go
79 lines (68 loc) · 2.14 KB
/
bootstrapper.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 operatorbootstrap
import (
"context"
"github.com/Noksa/operator-home/internal/operatorbootstrapinternal"
"github.com/Noksa/operator-home/internal/operatorconfiginternal"
"github.com/Noksa/operator-home/pkg/operatorconfig"
"github.com/samber/lo"
"os"
"os/signal"
ctrl "sigs.k8s.io/controller-runtime"
"sigs.k8s.io/controller-runtime/pkg/healthz"
"sigs.k8s.io/controller-runtime/pkg/log"
"sigs.k8s.io/controller-runtime/pkg/manager"
"syscall"
)
type Bootstrapper struct {
mgr manager.Manager
ctx context.Context
}
var cancelled = false
func Cancelled() bool {
return cancelled
}
func NewBootstrapper(ctx context.Context, operatorCfg operatorconfig.OperatorConfig, newOpts func() ctrl.Options, mgrFunc operatorbootstrapinternal.ManagerFunc) *Bootstrapper {
operatorconfiginternal.InstantiateConfiguration(operatorCfg)
mgr := operatorbootstrapinternal.NewManager(newOpts(), mgrFunc)
b := &Bootstrapper{mgr: mgr, ctx: ctx}
return b
}
func (b *Bootstrapper) GetMgr() manager.Manager {
return b.mgr
}
func (b *Bootstrapper) Context() context.Context {
return b.ctx
}
func (b *Bootstrapper) WithControllers(controllers ...KubernetesOperator) *Bootstrapper {
for _, controller := range controllers {
lo.Must0(controller.SetupWithManager(b.mgr))
}
return b
}
func (b *Bootstrapper) Run() {
lo.Must0(b.GetMgr().AddHealthzCheck("healthz", healthz.Ping), "unable to setup healthz")
lo.Must0(b.GetMgr().AddReadyzCheck("readyz", healthz.Ping), "unable to setup readyz")
lo.Must0(b.mgr.Start(b.ctx))
}
func CustomSignalsHandler(additionalActionBeforeCancel func()) context.Context {
ctx, cancel := context.WithCancel(context.Background())
c := make(chan os.Signal, 2)
signal.Notify(c, os.Interrupt, syscall.SIGTERM)
go func() {
s := <-c
logger := log.Log.WithName("CustomSignalsHandler")
if s != nil {
logger.WithValues("Signal", s.String()).Info("Received the signal")
}
cancelled = true
if additionalActionBeforeCancel != nil {
logger.Info("Running additional actions before exit")
}
additionalActionBeforeCancel()
logger.Info("Cancelling context")
cancel()
<-c
os.Exit(999) // second signal. Exit right now.
}()
return ctx
}