@@ -22,23 +22,19 @@ import (
22
22
appsv1 "k8s.io/api/apps/v1"
23
23
corev1 "k8s.io/api/core/v1"
24
24
_ "k8s.io/client-go/plugin/pkg/client/auth/gcp"
25
- "sigs.k8s.io/controller-runtime/pkg/builder"
25
+
26
+ ctrl "sigs.k8s.io/controller-runtime"
26
27
"sigs.k8s.io/controller-runtime/pkg/client/config"
27
- "sigs.k8s.io/controller-runtime/pkg/controller"
28
28
"sigs.k8s.io/controller-runtime/pkg/handler"
29
- "sigs.k8s.io/controller-runtime/pkg/log"
30
29
"sigs.k8s.io/controller-runtime/pkg/log/zap"
31
30
"sigs.k8s.io/controller-runtime/pkg/manager"
32
31
"sigs.k8s.io/controller-runtime/pkg/manager/signals"
33
32
"sigs.k8s.io/controller-runtime/pkg/source"
34
33
)
35
34
36
- func init () {
37
- log .SetLogger (zap .New ())
38
- }
39
-
40
35
func main () {
41
- entryLog := log .Log .WithName ("entrypoint" )
36
+ ctrl .SetLogger (zap .New ())
37
+ entryLog := ctrl .Log .WithName ("entrypoint" )
42
38
43
39
// Setup a Manager
44
40
entryLog .Info ("setting up manager" )
@@ -50,28 +46,21 @@ func main() {
50
46
51
47
// Setup a new controller to reconcile ReplicaSets
52
48
entryLog .Info ("Setting up controller" )
53
- c , err := controller .New ("foo-controller" , mgr , controller.Options {
54
- Reconciler : & reconcileReplicaSet {client : mgr .GetClient ()},
55
- })
56
- if err != nil {
57
- entryLog .Error (err , "unable to set up individual controller" )
58
- os .Exit (1 )
59
- }
60
49
61
- // Watch ReplicaSets and enqueue ReplicaSet object key
62
- if err := c . Watch ( source . Kind ( mgr . GetCache (), & appsv1. ReplicaSet {}, & handler. TypedEnqueueRequestForObject [ * appsv1. ReplicaSet ]{})); err != nil {
63
- entryLog . Error ( err , "unable to watch ReplicaSets" )
64
- os . Exit ( 1 )
65
- }
66
-
67
- // Watch Pods and enqueue owning ReplicaSet key
68
- if err := c . Watch ( source . Kind ( mgr .GetCache (), & corev1. Pod {},
69
- handler . TypedEnqueueRequestForOwner [ * corev1. Pod ]( mgr . GetScheme (), mgr . GetRESTMapper (), & appsv1. ReplicaSet {}, handler . OnlyControllerOwner ()))); err != nil {
70
- entryLog .Error (err , "unable to watch Pods " )
50
+ err = ctrl .
51
+ NewControllerManagedBy ( mgr ).
52
+ Named ( "foo-controller" ).
53
+ WatchesRawSource ( source . Kind ( mgr . GetCache (), & appsv1. ReplicaSet {},
54
+ & handler. TypedEnqueueRequestForObject [ * appsv1. ReplicaSet ]{})).
55
+ WatchesRawSource ( source . Kind ( mgr . GetCache (), & corev1. Pod {},
56
+ handler . TypedEnqueueRequestForOwner [ * corev1. Pod ]( mgr . GetScheme (), mgr . GetRESTMapper (), & appsv1. ReplicaSet {}, handler . OnlyControllerOwner ()))).
57
+ Complete ( & reconcileReplicaSet { client : mgr .GetClient ()})
58
+ if err != nil {
59
+ entryLog .Error (err , "could not create controller " )
71
60
os .Exit (1 )
72
61
}
73
62
74
- if err := builder . WebhookManagedBy (mgr ).
63
+ if err := ctrl . NewWebhookManagedBy (mgr ).
75
64
For (& corev1.Pod {}).
76
65
WithDefaulter (& podAnnotator {}).
77
66
WithValidator (& podValidator {}).
0 commit comments