-
Notifications
You must be signed in to change notification settings - Fork 9
/
controller.go
83 lines (69 loc) · 2.52 KB
/
controller.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
80
81
82
83
package app
import (
"context"
"fmt"
"time"
agonesv1 "agones.dev/agones/pkg/apis/agones/v1"
"github.com/pkg/errors"
"github.com/sirupsen/logrus"
"github.com/Octops/gameserver-ingress-controller/pkg/controller"
"github.com/Octops/gameserver-ingress-controller/pkg/handlers"
"github.com/Octops/gameserver-ingress-controller/pkg/k8sutil"
"github.com/Octops/gameserver-ingress-controller/pkg/manager"
"github.com/Octops/gameserver-ingress-controller/pkg/record"
"github.com/Octops/gameserver-ingress-controller/pkg/stores"
)
type Config struct {
Kubeconfig string
SyncPeriod string
Port int
Verbose bool
HealthProbeBindAddress string
MetricsBindAddress string
MaxConcurrentReconciles int
}
func StartController(ctx context.Context, logger *logrus.Entry, config Config) error {
duration, err := time.ParseDuration(config.SyncPeriod)
if err != nil {
withFatal(logger, err, fmt.Sprintf("error parsing sync-period flag: %s", config.SyncPeriod))
}
mgr, err := manager.NewManager(config.Kubeconfig, manager.Options{
SyncPeriod: &duration,
Port: config.Port,
HealthProbeBindAddress: config.HealthProbeBindAddress,
MetricsBindAddress: config.MetricsBindAddress,
MaxConcurrentReconciles: config.MaxConcurrentReconciles,
})
if err != nil {
withFatal(logger, err, "failed to create controller manager")
}
client, err := k8sutil.NewClientSet(config.Kubeconfig)
if err != nil {
withFatal(logger, err, "failed to create kubernetes client")
}
store, err := stores.NewStore(ctx, client)
if err != nil {
withFatal(logger, err, "failed to create store")
}
clusterConfig, err := k8sutil.NewClusterConfig(config.Kubeconfig)
if err != nil {
withFatal(logger, err, "failed to create cluster config")
}
agones, err := stores.NewAgonesStore(ctx, clusterConfig, duration)
recorder := mgr.GetEventRecorderFor("octops-gameserver-controller")
handler := handlers.NewGameSeverEventHandler(store, agones, record.NewEventRecorder(recorder))
ctrl, err := controller.NewGameServerController(ctx, mgr, handler, controller.Options{
For: &agonesv1.GameServer{},
})
if err != nil {
withFatal(logger, err, "failed to create controller")
}
logger.WithField("component", "controller").Info("starting gameserver controller")
if err := ctrl.Start(ctx); err != nil {
withFatal(logger, err, "failed to start controller")
}
return nil
}
func withFatal(logger *logrus.Entry, err error, msg string) {
logger.Fatal(errors.Wrap(err, msg))
}