implement InjectRecorder

Author: lichuqiang

pkg/client/config/config.go

@@ -102,12 +102,12 @@ func GetKubernetesClientSet() (*kubernetes.Clientset, error) {
 // in cluster and use the cluster provided kubeconfig.
 //
 // Will log.Fatal if KubernetesInformers cannot be created
-func GetKubernetesClientSetOrDie() (*kubernetes.Clientset, error) {
+func GetKubernetesClientSetOrDie() *kubernetes.Clientset {
 	cs, err := GetKubernetesClientSet()
 	if err != nil {
 		log.Fatalf("%v", err)
 	}
-	return cs, nil
+	return cs
 }
 
 // GetKubernetesInformers creates a informers.SharedInformerFactory for talking to a Kubernetes apiserver.

pkg/controller/controller.go

@@ -85,6 +85,11 @@ func New(name string, mrg manager.Manager, options Options) (Controller, error)
 		Name: name,
 	}
 
+	// Inject dependencies into the controller.
+	if err := mrg.SetFields(c); err != nil {
+		return nil, err
+	}
+
 	// Add the controller as a Manager components
 	return c, mrg.Add(c)
 }

pkg/internal/controller/controller.go

@@ -26,6 +26,7 @@ import (
 	"github.com/kubernetes-sigs/controller-runtime/pkg/handler"
 	"github.com/kubernetes-sigs/controller-runtime/pkg/predicate"
 	"github.com/kubernetes-sigs/controller-runtime/pkg/reconcile"
+	"github.com/kubernetes-sigs/controller-runtime/pkg/recorder"
 	"github.com/kubernetes-sigs/controller-runtime/pkg/runtime/inject"
 	logf "github.com/kubernetes-sigs/controller-runtime/pkg/runtime/log"
 	"github.com/kubernetes-sigs/controller-runtime/pkg/source"
@@ -34,6 +35,7 @@ import (
 	"k8s.io/apimachinery/pkg/util/wait"
 	"k8s.io/client-go/rest"
 	toolscache "k8s.io/client-go/tools/cache"
+	"k8s.io/client-go/tools/record"
 	"k8s.io/client-go/util/workqueue"
 )
 
@@ -87,6 +89,10 @@ type Controller struct {
 	// Started is true if the Controller has been Started
 	Started bool
 
+	// recorder is an event recorder for recording Event resources to the
+	// Kubernetes API.
+	recorder record.EventRecorder
+
 	// TODO(community): Consider initializing a logger with the Controller Name as the tag
 }
 
@@ -234,3 +240,14 @@ func (c *Controller) InjectFunc(f inject.Func) error {
 	c.SetFields = f
 	return nil
 }
+
+var _ inject.Recorder = &Controller{}
+
+// InjectRecorder is internal should be called only by the Controller.
+// InjectRecorder is used to inject the event recorder.
+func (c *Controller) InjectRecorder(scheme *runtime.Scheme, recorderGetFunc recorder.RecorderGetFuncType) error {
+	if c.recorder == nil {
+		c.recorder = recorderGetFunc(c.Name, scheme)
+	}
+	return nil
+}

pkg/manager/internal.go

@@ -93,6 +93,9 @@ func (cm *controllerManager) SetFields(i interface{}) error {
 	if _, err := inject.CacheInto(cm.cache, i); err != nil {
 		return err
 	}
+	if _, err := inject.RecorderInto(cm.scheme, i); err != nil {
+		return err
+	}
 	if _, err := inject.InjectorInto(cm.SetFields, i); err != nil {
 		return err
 	}

pkg/recorder/recorder.go

@@ -0,0 +1,41 @@
+/*
+Copyright 2018 The Kubernetes Authors.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+*/
+
+package recorder
+
+import (
+	"github.com/golang/glog"
+	"github.com/kubernetes-sigs/controller-runtime/pkg/client/config"
+	corev1 "k8s.io/api/core/v1"
+	"k8s.io/apimachinery/pkg/runtime"
+	typedcorev1 "k8s.io/client-go/kubernetes/typed/core/v1"
+	"k8s.io/client-go/tools/record"
+)
+
+type RecorderGetFuncType func(string, *runtime.Scheme) record.EventRecorder
+
+var eventBroadcaster record.EventBroadcaster
+
+func GetEventRecorderFor(name string, scheme *runtime.Scheme) record.EventRecorder {
+	if eventBroadcaster == nil {
+		cs := config.GetKubernetesClientSetOrDie()
+		eventBroadcaster := record.NewBroadcaster()
+		eventBroadcaster.StartLogging(glog.Infof)
+		eventBroadcaster.StartRecordingToSink(&typedcorev1.EventSinkImpl{Interface: cs.CoreV1().Events("")})
+	}
+
+	return eventBroadcaster.NewRecorder(scheme, corev1.EventSource{Component: name})
+}

pkg/runtime/inject/inject.go

@@ -19,6 +19,7 @@ package inject
 import (
 	"github.com/kubernetes-sigs/controller-runtime/pkg/cache"
 	"github.com/kubernetes-sigs/controller-runtime/pkg/client"
+	"github.com/kubernetes-sigs/controller-runtime/pkg/recorder"
 	"k8s.io/apimachinery/pkg/runtime"
 	"k8s.io/client-go/rest"
 )
@@ -83,6 +84,21 @@ func SchemeInto(scheme *runtime.Scheme, i interface{}) (bool, error) {
 	return false, nil
 }
 
+// Recorder is used by the ControllerManager to inject recorder into Sources, EventHandlers, Predicates, and
+// Reconciles
+type Recorder interface {
+	InjectRecorder(scheme *runtime.Scheme, recorderGetFunc recorder.RecorderGetFuncType) error
+}
+
+// RecorderInto will set recorder and return the result on i if it implements Recorder.  Returns
+// false if i does not implement Recorder.
+func RecorderInto(scheme *runtime.Scheme, i interface{}) (bool, error) {
+	if is, ok := i.(Recorder); ok {
+		return true, is.InjectRecorder(scheme, recorder.GetEventRecorderFor)
+	}
+	return false, nil
+}
+
 // Func injects dependencies into i.
 type Func func(i interface{}) error