adding defer statement to turn off maintenance mode

Signed-off-by: Jay Sharma <[email protected]>
chef · Oct 25, 2023 · 91e7c86 · 91e7c86
1 parent 870267d
commit 91e7c86
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/components/automate-cli/cmd/chef-automate/certRotate.go b/components/automate-cli/cmd/chef-automate/certRotate.go
@@ -1277,7 +1277,7 @@ func checkLagAndStopTraffic(infra *AutomateHAInfraDetails, sshConfig sshutils.SS
 	if err != nil {
 		return err
 	}
-	//////////////////////////////////////////////////////////////////////////
+
 	if userConsent {
 		agree, err := writer.Confirm(fmt.Sprintf(MAINTENANICE_ON_LAG, lag))
 		if err != nil {
@@ -1291,7 +1291,6 @@ func checkLagAndStopTraffic(infra *AutomateHAInfraDetails, sshConfig sshutils.SS
 	if err != nil {
 		return err
 	}
-	//////////////////////////////////////////////////////////////////////////
 
 	waitingStart := time.Now()
 	time.Sleep(waitTime * time.Second)
@@ -1358,6 +1357,14 @@ func (c *certRotateFlow) certRotateFromTemplate(clusterCertificateFile string, s
 		return err
 	}
 
+	defer func() {
+		c.log.Debug("==========================================================")
+		c.log.Debug("Defer Starting traffic MAINTENANICE MODE OFF")
+		c.log.Debug("==========================================================")
+		startTrafficOnAutomateNode(infra, configRes, c.sshUtil, c.log)
+		startTrafficOnChefServerNode(infra, configRes, c.sshUtil, c.log)
+	}()
+
 	if templateCerts != nil {
 		// rotating PG certs
 		start := time.Now()