Fix event processing during bootup (#240)

admiral/pkg/apis/admiral/routes/handler_test.go

@@ -15,6 +15,7 @@ import (
 	"net/http/httptest"
 	"strings"
 	"testing"
+	"time"
 )
 
 func TestReturnSuccessGET(t *testing.T) {
@@ -48,6 +49,7 @@ func TestGetClusters(t *testing.T) {
 					RemoteClusters: map[string]*secret.RemoteCluster{},
 				},
 			},
+			StartTime: time.Now(),
 		},
 	}
 	testCases := []struct {

admiral/pkg/clusters/handler.go

@@ -574,7 +574,8 @@ func addUpdateVirtualService(obj *v1alpha3.VirtualService, exist *v1alpha3.Virtu
 
 func addUpdateServiceEntry(obj *v1alpha3.ServiceEntry, exist *v1alpha3.ServiceEntry, namespace string, rc *RemoteController) {
 	var err error
-	var op string
+	var op, diff string
+	var skipUpdate bool
 	if obj.Annotations == nil {
 		obj.Annotations = map[string]string{}
 	}
@@ -589,7 +590,7 @@ func addUpdateServiceEntry(obj *v1alpha3.ServiceEntry, exist *v1alpha3.ServiceEn
 		exist.Labels = obj.Labels
 		exist.Annotations = obj.Annotations
 		op = "Update"
-		skipUpdate, diff := skipDestructiveUpdate(rc, obj, exist)
+		skipUpdate, diff = skipDestructiveUpdate(rc, obj, exist)
 		if diff != "" {
 			log.Infof(LogFormat+" diff=%s", op, "ServiceEntry", obj.Name, rc.ClusterID, "Diff in update", diff)
 		}

admiral/pkg/clusters/handler_test.go

@@ -529,6 +529,7 @@ func TestHandleVirtualServiceEvent(t *testing.T) {
 				CnameDependentClusterCache: cnameCache,
 				SeClusterCache:             common.NewMapOfMaps(),
 			},
+			StartTime: time.Now(),
 		},
 	}
 
@@ -548,6 +549,7 @@ func TestHandleVirtualServiceEvent(t *testing.T) {
 				CnameDependentClusterCache: goodCnameCache,
 				SeClusterCache:             common.NewMapOfMaps(),
 			},
+			StartTime: time.Now(),
 		},
 	}
 
@@ -574,6 +576,7 @@ func TestHandleVirtualServiceEvent(t *testing.T) {
 				CnameDependentClusterCache: goodCnameCache,
 				SeClusterCache:             common.NewMapOfMaps(),
 			},
+			StartTime: time.Now(),
 		},
 	}
 

admiral/pkg/clusters/registry.go

@@ -27,6 +27,7 @@ func InitAdmiral(ctx context.Context, params common.AdmiralParams) (*RemoteRegis
 	common.InitializeConfig(params)
 	w := RemoteRegistry{
 		ctx: ctx,
+		StartTime: time.Now(),
 	}
 
 	wd := DependencyHandler{

admiral/pkg/clusters/registry_test.go

@@ -48,6 +48,7 @@ func TestDeleteCacheControllerThatDoesntExist(t *testing.T) {
 
 	w := RemoteRegistry{
 		RemoteControllers: make(map[string]*RemoteController),
+		StartTime: time.Now(),
 	}
 
 	err := w.deleteCacheController("I don't exit")
@@ -61,6 +62,7 @@ func TestDeleteCacheController(t *testing.T) {
 
 	w := RemoteRegistry{
 		RemoteControllers: make(map[string]*RemoteController),
+		StartTime: time.Now(),
 	}
 
 	r := rest.Config{

admiral/pkg/clusters/serviceentry.go

@@ -58,6 +58,10 @@ func modifyServiceEntryForNewServiceOrPod(event admiral.EventType, env string, s
 
 	defer util.LogElapsedTime("modifyServiceEntryForNewServiceOrPod", sourceIdentity, env, "")()
 
+	if IsCacheWarmupTime(remoteRegistry) {
+		log.Infof(LogFormat, event, env, sourceIdentity, "", "Processing skipped during cache warm up state")
+		return nil
+	}
 	//create a service entry, destination rule and virtual service in the local cluster
 	sourceServices := make(map[string]*k8sV1.Service)
 	sourceWeightedServices := make(map[string]map[string]*WeightedService)

admiral/pkg/clusters/serviceentry_test.go

@@ -289,6 +289,7 @@ func TestCreateServiceEntryForNewServiceOrPod(t *testing.T) {
 		KubeconfigPath: "testdata/fake.config",
 	}
 	rr, _ := InitAdmiral(context.Background(), p)
+	rr.StartTime = time.Now().Add(-60*time.Second)
 
 	config := rest.Config{
 		Host: "localhost",
@@ -875,6 +876,8 @@ func TestCreateServiceEntryForNewServiceOrPodRolloutsUsecase(t *testing.T) {
 
 	rr, _ := InitAdmiral(context.Background(), p)
 
+	rr.StartTime = time.Now().Add(-60*time.Second)
+
 	config := rest.Config{
 		Host: "localhost",
 	}
@@ -1011,6 +1014,7 @@ func TestCreateServiceEntryForBlueGreenRolloutsUsecase(t *testing.T) {
 	config := rest.Config{
 		Host: "localhost",
 	}
+	rr.StartTime = time.Now().Add(-60*time.Second)
 
 	d, e := admiral.NewDeploymentController("", make(chan struct{}), &test.MockDeploymentHandler{}, &config, time.Second*time.Duration(300))
 

admiral/pkg/clusters/types.go

@@ -60,6 +60,7 @@ type RemoteRegistry struct {
 	secretClient      k8s.Interface
 	ctx               context.Context
 	AdmiralCache      *AdmiralCache
+	StartTime         time.Time
 }
 
 func (r *RemoteRegistry) shutdown() {

admiral/pkg/clusters/util.go

@@ -10,6 +10,7 @@ import (
 	k8sV1 "k8s.io/api/core/v1"
 	"strconv"
 	"strings"
+	"time"
 )
 
 func GetMeshPorts(clusterName string, destService *k8sV1.Service,
@@ -127,3 +128,7 @@ func ValidateConfigmapBeforePutting(cm *k8sV1.ConfigMap) error {
 	}
 	return nil
 }
+
+func IsCacheWarmupTime(remoteRegistry *RemoteRegistry) bool {
+	return time.Since(remoteRegistry.StartTime) < common.GetAdmiralParams().CacheRefreshDuration
+}

admiral/pkg/controller/admiral/controller.go

@@ -102,8 +102,8 @@ func (c *Controller) Run(stopCh <-chan struct{}) {
 	}
 
 	log.Infof("Informer caches synced for controller=%v, current keys=%v", c.name, c.informer.GetStore().ListKeys())
-	//wait for 30 seconds for the first time (for all caches to sync)
-	wait.Until(c.runWorker, 30 * time.Second, stopCh)
+
+	wait.Until(c.runWorker, 5 * time.Second, stopCh)
 }
 
 func (c *Controller) runWorker() {

admiral/pkg/controller/admiral/service.go

@@ -60,14 +60,14 @@ func (s *serviceCache) Put(service *k8sV1.Service) {
 			Service:  make(map[string]map[string]*k8sV1.Service),
 			Identity: s.getKey(service),
 		}
-		s.cache[identity] = existing
 	}
 	namespaceServices := existing.Service[service.Namespace]
 	if namespaceServices == nil {
 		namespaceServices = make(map[string]*k8sV1.Service)
 	}
 	namespaceServices[service.Name] = service
 	existing.Service[service.Namespace] = namespaceServices
+	s.cache[identity] = existing
 
 }