From 04005a2e0fd14e28d82963a30b2252e55832d905 Mon Sep 17 00:00:00 2001
From: Manuel de Brito Fontes <aledbf@gmail.com>
Date: Tue, 19 Sep 2017 17:09:33 -0300
Subject: [PATCH] Avoid issues with goroutines updating fields

---
 core/pkg/ingress/controller/controller.go | 35 ++++++++++++++++++-----
 core/pkg/ingress/controller/listers.go    |  6 ++--
 2 files changed, 31 insertions(+), 10 deletions(-)

diff --git a/core/pkg/ingress/controller/controller.go b/core/pkg/ingress/controller/controller.go
index 2f703234d..03f4c92a9 100644
--- a/core/pkg/ingress/controller/controller.go
+++ b/core/pkg/ingress/controller/controller.go
@@ -24,6 +24,7 @@ import (
 	"strconv"
 	"strings"
 	"sync"
+	"sync/atomic"
 	"time"
 
 	"github.com/golang/glog"
@@ -110,9 +111,9 @@ type GenericController struct {
 	// runningConfig contains the running configuration in the Backend
 	runningConfig *ingress.Configuration
 
-	forceReload bool
+	forceReload int32
 
-	initialSyncDone bool
+	initialSyncDone int32
 }
 
 // Configuration contains all the settings required by an Ingress controller
@@ -283,7 +284,7 @@ func (ic *GenericController) syncIngress(key interface{}) error {
 		PassthroughBackends: passUpstreams,
 	}
 
-	if !ic.forceReload && ic.runningConfig != nil && ic.runningConfig.Equal(&pcfg) {
+	if !ic.isForceReload() && ic.runningConfig != nil && ic.runningConfig.Equal(&pcfg) {
 		glog.V(3).Infof("skipping backend reload (no changes detected)")
 		return nil
 	}
@@ -302,7 +303,7 @@ func (ic *GenericController) syncIngress(key interface{}) error {
 	setSSLExpireTime(servers)
 
 	ic.runningConfig = &pcfg
-	ic.forceReload = false
+	ic.setForceReload(false)
 
 	return nil
 }
@@ -1185,7 +1186,7 @@ func (ic GenericController) Stop() error {
 }
 
 // Start starts the Ingress controller.
-func (ic GenericController) Start() {
+func (ic *GenericController) Start() {
 	glog.Infof("starting Ingress controller")
 
 	go ic.ingController.Run(ic.stopCh)
@@ -1224,14 +1225,14 @@ func (ic GenericController) Start() {
 
 	createDefaultSSLCertificate()
 
+	ic.setInitialSyncDone()
+
 	go ic.syncQueue.Run(time.Second, ic.stopCh)
 
 	if ic.syncStatus != nil {
 		go ic.syncStatus.Run(ic.stopCh)
 	}
 
-	ic.initialSyncDone = true
-
 	time.Sleep(5 * time.Second)
 	// force initial sync
 	ic.syncQueue.Enqueue(&extensions.Ingress{})
@@ -1239,6 +1240,26 @@ func (ic GenericController) Start() {
 	<-ic.stopCh
 }
 
+func (ic *GenericController) isForceReload() bool {
+	return atomic.LoadInt32(&ic.forceReload) != 0
+}
+
+func (ic *GenericController) setForceReload(shouldReload bool) {
+	if shouldReload {
+		atomic.StoreInt32(&ic.forceReload, 1)
+	} else {
+		atomic.StoreInt32(&ic.forceReload, 0)
+	}
+}
+
+func (ic *GenericController) isInitialSyncDone() bool {
+	return atomic.LoadInt32(&ic.initialSyncDone) != 0
+}
+
+func (ic *GenericController) setInitialSyncDone() {
+	atomic.StoreInt32(&ic.initialSyncDone, 1)
+}
+
 func createDefaultSSLCertificate() {
 	defCert, defKey := ssl.GetFakeSSLCert()
 	c, err := ssl.AddOrUpdateCertAndKey(fakeCertificate, defCert, defKey, []byte{})
diff --git a/core/pkg/ingress/controller/listers.go b/core/pkg/ingress/controller/listers.go
index 60cf3a92c..657c679ef 100644
--- a/core/pkg/ingress/controller/listers.go
+++ b/core/pkg/ingress/controller/listers.go
@@ -46,7 +46,7 @@ func (ic *GenericController) createListers(disableNodeLister bool) {
 			}
 			ic.recorder.Eventf(addIng, apiv1.EventTypeNormal, "CREATE", fmt.Sprintf("Ingress %s/%s", addIng.Namespace, addIng.Name))
 
-			if ic.initialSyncDone {
+			if ic.isInitialSyncDone() {
 				ic.syncQueue.Enqueue(obj)
 			}
 		},
@@ -142,7 +142,7 @@ func (ic *GenericController) createListers(disableNodeLister bool) {
 			if mapKey == ic.cfg.ConfigMapName {
 				glog.V(2).Infof("adding configmap %v to backend", mapKey)
 				ic.cfg.Backend.SetConfig(upCmap)
-				ic.forceReload = true
+				ic.setForceReload(true)
 			}
 		},
 		UpdateFunc: func(old, cur interface{}) {
@@ -152,7 +152,7 @@ func (ic *GenericController) createListers(disableNodeLister bool) {
 				if mapKey == ic.cfg.ConfigMapName {
 					glog.V(2).Infof("updating configmap backend (%v)", mapKey)
 					ic.cfg.Backend.SetConfig(upCmap)
-					ic.forceReload = true
+					ic.setForceReload(true)
 				}
 				// updates to configuration configmaps can trigger an update
 				if mapKey == ic.cfg.ConfigMapName || mapKey == ic.cfg.TCPConfigMapName || mapKey == ic.cfg.UDPConfigMapName {