Handle review comments for Fix goroutine leak in federation service controller

federation/pkg/federation-controller/service/endpoint_helper.go

@@ -32,29 +32,28 @@ import (
 // It enforces that the syncHandler is never invoked concurrently with the same key.
 func (sc *ServiceController) clusterEndpointWorker() {
 	// process all pending events in endpointWorkerDoneChan
-	eventPending := true
+ForLoop:
-	for eventPending {
+	for {
 		select {
 		case clusterName := <-sc.endpointWorkerDoneChan:
 			sc.endpointWorkerMap[clusterName] = false
 		default:
 			// non-blocking, comes here if all existing events are processed
-			eventPending = false
+			break ForLoop
-			break
 		}
 	}
 
 	for clusterName, cache := range sc.clusterCache.clientMap {
-		workerExist, keyFound := sc.endpointWorkerMap[clusterName]
+		workerExist, found := sc.endpointWorkerMap[clusterName]
-		if keyFound && workerExist {
+		if found && workerExist {
 			continue
 		}
-		sc.endpointWorkerMap[clusterName] = true
 
 		// create a worker only if the previous worker has finished and gone out of scope
 		go func(cache *clusterCache, clusterName string) {
 			fedClient := sc.federationClient
 			for {
+				func() {
 					key, quit := cache.endpointQueue.Get()
 					// update endpoint cache
 					if quit {
@@ -67,8 +66,10 @@ func (sc *ServiceController) clusterEndpointWorker() {
 					if err != nil {
 						glog.V(2).Infof("Failed to sync endpoint: %+v", err)
 					}
+				}()
 			}
 		}(cache, clusterName)
+		sc.endpointWorkerMap[clusterName] = true
 	}
 }
 

federation/pkg/federation-controller/service/service_helper.go

@@ -36,29 +36,28 @@ import (
 // It enforces that the syncHandler is never invoked concurrently with the same key.
 func (sc *ServiceController) clusterServiceWorker() {
 	// process all pending events in serviceWorkerDoneChan
-	eventPending := true
+ForLoop:
-	for eventPending {
+	for {
 		select {
 		case clusterName := <-sc.serviceWorkerDoneChan:
 			sc.serviceWorkerMap[clusterName] = false
 		default:
 			// non-blocking, comes here if all existing events are processed
-			eventPending = false
+			break ForLoop
-			break
 		}
 	}
 
 	for clusterName, cache := range sc.clusterCache.clientMap {
-		workerExist, keyFound := sc.serviceWorkerMap[clusterName]
+		workerExist, found := sc.serviceWorkerMap[clusterName]
-		if keyFound && workerExist {
+		if found && workerExist {
 			continue
 		}
-		sc.serviceWorkerMap[clusterName] = true
 
 		// create a worker only if the previous worker has finished and gone out of scope
 		go func(cache *clusterCache, clusterName string) {
 			fedClient := sc.federationClient
 			for {
+				func() {
 					key, quit := cache.serviceQueue.Get()
 					if quit {
 						// send signal that current worker has finished tasks and is going out of scope
@@ -70,9 +69,10 @@ func (sc *ServiceController) clusterServiceWorker() {
 					if err != nil {
 						glog.Errorf("Failed to sync service: %+v", err)
 					}
-
+				}()
 			}
 		}(cache, clusterName)
+		sc.serviceWorkerMap[clusterName] = true
 	}
 }
 

federation/pkg/federation-controller/service/servicecontroller.go

@@ -68,7 +68,7 @@ const (
 	KubeAPIQPS    = 20.0
 	KubeAPIBurst  = 30
 
-	maxNoOfClusters = 256
+	maxNoOfClusters = 100
 )
 
 type cachedService struct {