From d1bfb3f8ffdd527f68b7565542387566e1ccb3a0 Mon Sep 17 00:00:00 2001
From: Marcin Wielgus <mwielgus@google.com>
Date: Wed, 31 Aug 2016 23:35:20 +0200
Subject: [PATCH] Events for federated replica set controller

---
 .../replicaset/replicasetcontroller.go        | 20 ++++++++++++++++++-
 .../replicaset/replicasetcontroller_test.go   |  1 +
 2 files changed, 20 insertions(+), 1 deletion(-)

diff --git a/federation/pkg/federation-controller/replicaset/replicasetcontroller.go b/federation/pkg/federation-controller/replicaset/replicasetcontroller.go
index cbd6f3a1969..72cd4f0e6d7 100644
--- a/federation/pkg/federation-controller/replicaset/replicasetcontroller.go
+++ b/federation/pkg/federation-controller/replicaset/replicasetcontroller.go
@@ -31,11 +31,13 @@ import (
 	fedclientset "k8s.io/kubernetes/federation/client/clientset_generated/federation_release_1_4"
 	planner "k8s.io/kubernetes/federation/pkg/federation-controller/replicaset/planner"
 	fedutil "k8s.io/kubernetes/federation/pkg/federation-controller/util"
+	"k8s.io/kubernetes/federation/pkg/federation-controller/util/eventsink"
 	"k8s.io/kubernetes/pkg/api"
 	apiv1 "k8s.io/kubernetes/pkg/api/v1"
 	extensionsv1 "k8s.io/kubernetes/pkg/apis/extensions/v1beta1"
 	"k8s.io/kubernetes/pkg/client/cache"
 	kubeclientset "k8s.io/kubernetes/pkg/client/clientset_generated/release_1_4"
+	"k8s.io/kubernetes/pkg/client/record"
 	"k8s.io/kubernetes/pkg/controller"
 	"k8s.io/kubernetes/pkg/controller/framework"
 	"k8s.io/kubernetes/pkg/runtime"
@@ -90,12 +92,18 @@ type ReplicaSetController struct {
 	fedUpdater fedutil.FederatedUpdater
 
 	replicaSetBackoff *flowcontrol.Backoff
+	// For events
+	eventRecorder record.EventRecorder
 
 	defaultPlanner *planner.Planner
 }
 
 // NewclusterController returns a new cluster controller
 func NewReplicaSetController(federationClient fedclientset.Interface) *ReplicaSetController {
+	broadcaster := record.NewBroadcaster()
+	broadcaster.StartRecordingToSink(eventsink.NewFederatedEventSink(federationClient))
+	recorder := broadcaster.NewRecorder(api.EventSource{Component: "federated-replicaset-controller"})
+
 	frsc := &ReplicaSetController{
 		fedClient:           federationClient,
 		replicasetDeliverer: fedutil.NewDelayingDeliverer(),
@@ -107,6 +115,7 @@ func NewReplicaSetController(federationClient fedclientset.Interface) *ReplicaSe
 				"*": {Weight: 1},
 			},
 		}),
+		eventRecorder: recorder,
 	}
 
 	replicaSetFedInformerFactory := func(cluster *fedv1.Cluster, clientset kubeclientset.Interface) (cache.Store, framework.ControllerInterface) {
@@ -464,6 +473,9 @@ func (frsc *ReplicaSetController) reconcileReplicaSet(key string) (reconciliatio
 
 		if !exists {
 			if replicas > 0 {
+				frsc.eventRecorder.Eventf(frs, api.EventTypeNormal, "CreateInCluster",
+					"Creating replicaset in cluster %s", clusterName)
+
 				operations = append(operations, fedutil.FederatedOperation{
 					Type:        fedutil.OperationTypeAdd,
 					Obj:         lrs,
@@ -475,6 +487,9 @@ func (frsc *ReplicaSetController) reconcileReplicaSet(key string) (reconciliatio
 			// Update existing replica set, if needed.
 			if !fedutil.ObjectMetaEquivalent(lrs.ObjectMeta, currentLrs.ObjectMeta) ||
 				!reflect.DeepEqual(lrs.Spec, currentLrs.Spec) {
+				frsc.eventRecorder.Eventf(frs, api.EventTypeNormal, "UpdateInCluster",
+					"Updating replicaset in cluster %s", clusterName)
+
 				operations = append(operations, fedutil.FederatedOperation{
 					Type:        fedutil.OperationTypeUpdate,
 					Obj:         lrs,
@@ -498,7 +513,10 @@ func (frsc *ReplicaSetController) reconcileReplicaSet(key string) (reconciliatio
 		// Everything is in order
 		return statusAllOk, nil
 	}
-	err = frsc.fedUpdater.Update(operations, updateTimeout)
+	err = frsc.fedUpdater.UpdateWithOnError(operations, updateTimeout, func(op fedutil.FederatedOperation, operror error) {
+		frsc.eventRecorder.Eventf(frs, api.EventTypeNormal, "FailedUpdateInCluster",
+			"Replicaset update in cluster %s failed: %v", op.ClusterName, operror)
+	})
 	if err != nil {
 		glog.Errorf("Failed to execute updates for %s: %v", key, err)
 		return statusError, err
diff --git a/federation/pkg/federation-controller/replicaset/replicasetcontroller_test.go b/federation/pkg/federation-controller/replicaset/replicasetcontroller_test.go
index 79b0b7ce8b9..e9f7362b5a1 100644
--- a/federation/pkg/federation-controller/replicaset/replicasetcontroller_test.go
+++ b/federation/pkg/federation-controller/replicaset/replicasetcontroller_test.go
@@ -166,6 +166,7 @@ func newReplicaSetWithReplicas(name string, replicas int32) *extensionsv1.Replic
 		ObjectMeta: apiv1.ObjectMeta{
 			Name:      name,
 			Namespace: apiv1.NamespaceDefault,
+			SelfLink:  "/api/v1/namespaces/default/replicasets/name",
 		},
 		Spec: extensionsv1.ReplicaSetSpec{
 			Replicas: &replicas,