Update backendconfig event handlers

kubernetes · Mar 11, 2020 · 274e02e · 274e02e
1 parent 3667931
commit 274e02e
Showing 1 changed file with 24 additions and 3 deletions.
diff --git a/pkg/controller/controller.go b/pkg/controller/controller.go
@@ -191,19 +191,40 @@ func NewLoadBalancerController(
 	// BackendConfig event handlers.
 	ctx.BackendConfigInformer.AddEventHandler(cache.ResourceEventHandlerFuncs{
 		AddFunc: func(obj interface{}) {
-			beConfig := obj.(*backendconfigv1beta1.BackendConfig)
+			beConfig, ok := obj.(*backendconfigv1beta1.BackendConfig)
+			if !ok {
+				klog.Errorf("Cannot cast obj to BackendConfig (obj type is %T)", obj)
+				return
+			}
 			ings := operator.Ingresses(ctx.Ingresses().List()).ReferencesBackendConfig(beConfig, operator.Services(ctx.Services().List())).AsList()
 			lbc.ingQueue.Enqueue(convert(ings)...)
 		},
 		UpdateFunc: func(old, cur interface{}) {
 			if !reflect.DeepEqual(old, cur) {
-				beConfig := cur.(*backendconfigv1beta1.BackendConfig)
+				beConfig, ok := cur.(*backendconfigv1beta1.BackendConfig)
+				if !ok {
+					klog.Errorf("Cannot cast obj to BackendConfig (obj type is %T)", cur)
+					return
+				}
 				ings := operator.Ingresses(ctx.Ingresses().List()).ReferencesBackendConfig(beConfig, operator.Services(ctx.Services().List())).AsList()
 				lbc.ingQueue.Enqueue(convert(ings)...)
 			}
 		},
 		DeleteFunc: func(obj interface{}) {
-			beConfig := obj.(*backendconfigv1beta1.BackendConfig)
+			beConfig, ok := obj.(*backendconfigv1beta1.BackendConfig)
+			if !ok {
+				klog.V(3).Infof("Cannot cast obj to BackendConfig (obj type is %T), casting into DeletedFinalStateUnknown", obj)
+				var tombstone cache.DeletedFinalStateUnknown
+				tombstone, ok = obj.(cache.DeletedFinalStateUnknown)
+				if !ok {
+					klog.Errorf("Cannot cast obj to DeletedFinalStateUnknown (obj type is %T)", obj)
+					return
+				}
+				if beConfig, ok = tombstone.Obj.(*backendconfigv1beta1.BackendConfig); !ok {
+					klog.Errorf("Cannot cast DeletedFinalStateUnknown.obj to BackendConfig (obj type is %T)", tombstone.Obj)
+					return
+				}
+			}
 			ings := operator.Ingresses(ctx.Ingresses().List()).ReferencesBackendConfig(beConfig, operator.Services(ctx.Services().List())).AsList()
 			lbc.ingQueue.Enqueue(convert(ings)...)
 		},