From 75a4bd6c9578ac5d877ab28d3e2d6750b7ac7635 Mon Sep 17 00:00:00 2001
From: "Cuong. Duong Manh" <cuongpigerr@gmail.com>
Date: Mon, 1 Jul 2024 10:09:50 +0700
Subject: [PATCH] [build][enhance] decrease exponenital ebs

---
 pkg/driver/controller.go | 8 ++++++--
 1 file changed, 6 insertions(+), 2 deletions(-)

diff --git a/pkg/driver/controller.go b/pkg/driver/controller.go
index c3fe2cf..f54f451 100644
--- a/pkg/driver/controller.go
+++ b/pkg/driver/controller.go
@@ -471,7 +471,7 @@ func (s *controllerService) ControllerGetCapabilities(ctx lctx.Context, req *lcs
 }
 
 func (s *controllerService) ControllerExpandVolume(_ lctx.Context, preq *lcsi.ControllerExpandVolumeRequest) (*lcsi.ControllerExpandVolumeResponse, error) {
-	llog.V(4).InfoS("ControllerExpandVolume: called", "preq", *preq)
+	llog.V(4).InfoS("[INFO] - ControllerExpandVolume: Called", "request", *preq)
 
 	volumeID := preq.GetVolumeId()
 	if volumeID == "" {
@@ -480,9 +480,13 @@ func (s *controllerService) ControllerExpandVolume(_ lctx.Context, preq *lcsi.Co
 
 	// check if a request is already in-flight
 	if ok := s.inFlight.Insert(volumeID); !ok {
+		llog.InfoS("[INFO] - ControllerExpandVolume: Operation is already in-flight", "volumeID", volumeID)
 		return nil, ErrOperationAlreadyExists(volumeID)
 	}
-	defer s.inFlight.Delete(volumeID)
+	defer func() {
+		llog.InfoS("[INFO] - ControllerExpandVolume: Operation completed", "volumeID", volumeID)
+		s.inFlight.Delete(volumeID)
+	}()
 
 	capRange := preq.GetCapacityRange()
 	if capRange == nil {