split init controller; make controller config private

pkg/controller/controller.go

@@ -29,11 +29,10 @@ type Config struct {
 }
 
 type Controller struct {
-	Config
-
+	config   Config
 	opConfig *config.Config
-	logger   *logrus.Entry
 
+	logger         *logrus.Entry
 	KubeClient     k8sutil.KubernetesClient
 	RestClient     rest.Interface
 	TeamsAPIClient *teams.API
@@ -47,81 +46,75 @@ type Controller struct {
 	podCh              chan spec.PodEvent
 
 	clusterEventQueues  []*cache.FIFO
-
 	lastClusterSyncTime int64
 }
 
 func NewController(controllerConfig *Config) *Controller {
-	configMapData := make(map[string]string)
 	logger := logrus.New()
 
-	client, err := k8sutil.ClientSet(controllerConfig.RestConfig)
+	return &Controller{
+		config:   *controllerConfig,
+		opConfig: &config.Config{},
+		logger:   logger.WithField("pkg", "controller"),
+		clusters: make(map[spec.NamespacedName]*cluster.Cluster),
+		stopChs:  make(map[spec.NamespacedName]chan struct{}),
+		podCh:    make(chan spec.PodEvent),
+	}
+}
+
+func (c *Controller) initClients() {
+	client, err := k8sutil.ClientSet(c.config.RestConfig)
 	if err != nil {
-		logger.Fatalf("couldn't create client: %v", err)
+		c.logger.Fatalf("couldn't create client: %v", err)
 	}
+	c.KubeClient = k8sutil.NewFromKubernetesInterface(client)
 
-	restClient, err := k8sutil.KubernetesRestClient(*controllerConfig.RestConfig)
+	c.RestClient, err = k8sutil.KubernetesRestClient(*c.config.RestConfig)
 	if err != nil {
-		logger.Fatalf("couldn't create rest client: %v", err)
+		c.logger.Fatalf("couldn't create rest client: %v", err)
 	}
+}
+
+func (c *Controller) initOperatorConfig() {
+	configMapData := make(map[string]string)
 
-	if controllerConfig.ConfigMapName != (spec.NamespacedName{}) {
-		configMap, err := client.ConfigMaps(controllerConfig.ConfigMapName.Namespace).Get(controllerConfig.ConfigMapName.Name, meta_v1.GetOptions{})
+	if c.config.ConfigMapName != (spec.NamespacedName{}) {
+		configMap, err := c.KubeClient.ConfigMaps(c.config.ConfigMapName.Namespace).
+			Get(c.config.ConfigMapName.Name, meta_v1.GetOptions{})
 		if err != nil {
 			panic(err)
 		}
 
 		configMapData = configMap.Data
 	} else {
-		logger.Infoln("No ConfigMap specified. Loading default values")
+		c.logger.Infoln("No ConfigMap specified. Loading default values")
 	}
 
 	if configMapData["namespace"] == "" { // Namespace in ConfigMap has priority over env var
-		configMapData["namespace"] = controllerConfig.Namespace
+		configMapData["namespace"] = c.config.Namespace
 	}
-	if controllerConfig.NoDatabaseAccess {
+	if c.config.NoDatabaseAccess {
 		configMapData["enable_database_access"] = "false"
 	}
-	if controllerConfig.NoTeamsAPI {
+	if c.config.NoTeamsAPI {
 		configMapData["enable_teams_api"] = "false"
 	}
-	operatorConfig := config.NewFromMap(configMapData)
-
-	logger.Infof("Config: %s", operatorConfig.MustMarshal())
 
-	if operatorConfig.DebugLogging {
-		logger.Level = logrus.DebugLevel
+	c.opConfig = config.NewFromMap(configMapData)
 }
 
-	return &Controller{
-		Config:         *controllerConfig,
-		opConfig:       operatorConfig,
-		logger:         logger.WithField("pkg", "controller"),
-		clusters:       make(map[spec.NamespacedName]*cluster.Cluster),
-		stopChs:        make(map[spec.NamespacedName]chan struct{}),
-		podCh:          make(chan spec.PodEvent),
-		TeamsAPIClient: teams.NewTeamsAPI(operatorConfig.TeamsAPIUrl, logger),
-		KubeClient:     k8sutil.NewFromKubernetesInterface(client),
-		RestClient:     restClient,
-	}
-}
-
-func (c *Controller) Run(stopCh <-chan struct{}, wg *sync.WaitGroup) {
-	defer wg.Done()
-	wg.Add(1)
-
-	c.initController()
+func (c *Controller) initController() {
+	c.initClients()
+	c.initOperatorConfig()
 
-	go c.runInformers(stopCh)
+	c.logger.Infof("Config: %s", c.opConfig.MustMarshal())
 
-	for i := range c.clusterEventQueues {
-		go c.processClusterEventsQueue(i)
+	if c.opConfig.DebugLogging {
+		c.logger.Level = logrus.DebugLevel
 	}
 
-	c.logger.Info("Started working in background")
-}
+	c.TeamsAPIClient = teams.NewTeamsAPI(c.opConfig.TeamsAPIUrl, c.logger.Logger)
 
-func (c *Controller) initController() {
 	if err := c.createTPR(); err != nil {
 		c.logger.Fatalf("could not register ThirdPartyResource: %v", err)
 	}
@@ -129,7 +122,7 @@ func (c *Controller) initController() {
 	if infraRoles, err := c.getInfrastructureRoles(&c.opConfig.InfrastructureRolesSecretName); err != nil {
 		c.logger.Warningf("could not get infrastructure roles: %v", err)
 	} else {
-		c.InfrastructureRoles = infraRoles
+		c.config.InfrastructureRoles = infraRoles
 	}
 
 	// Postgresqls
@@ -179,6 +172,21 @@ func (c *Controller) initController() {
 	}
 }
 
+func (c *Controller) Run(stopCh <-chan struct{}, wg *sync.WaitGroup) {
+	defer wg.Done()
+	wg.Add(1)
+
+	c.initController()
+
+	go c.runInformers(stopCh)
+
+	for i := range c.clusterEventQueues {
+		go c.processClusterEventsQueue(i)
+	}
+
+	c.logger.Info("Started working in background")
+}
+
 func (c *Controller) runInformers(stopCh <-chan struct{}) {
 	go c.postgresqlInformer.Run(stopCh)
 	go c.podInformer.Run(stopCh)

pkg/controller/util.go

@@ -17,12 +17,12 @@ import (
 
 func (c *Controller) makeClusterConfig() cluster.Config {
 	infrastructureRoles := make(map[string]spec.PgUser)
-	for k, v := range c.InfrastructureRoles {
+	for k, v := range c.config.InfrastructureRoles {
 		infrastructureRoles[k] = v
 	}
 
 	return cluster.Config{
-		RestConfig:          c.RestConfig,
+		RestConfig:          c.config.RestConfig,
 		OpConfig:            config.Copy(c.opConfig),
 		InfrastructureRoles: infrastructureRoles,
 	}