disk/status-waiter: add a hook to workaround credential refresh

should be reverted once we have a ecsClient that can refresh its credentials

pkg/disk/disk.go

@@ -25,6 +25,7 @@ import (
 	"github.com/aliyun/alibaba-cloud-sdk-go/services/ecs"
 	"github.com/container-storage-interface/spec/lib/go/csi"
 	snapClientset "github.com/kubernetes-csi/external-snapshotter/client/v4/clientset/versioned"
+	"github.com/kubernetes-sigs/alibaba-cloud-csi-driver/pkg/cloud"
 	"github.com/kubernetes-sigs/alibaba-cloud-csi-driver/pkg/cloud/metadata"
 	"github.com/kubernetes-sigs/alibaba-cloud-csi-driver/pkg/common"
 	"github.com/kubernetes-sigs/alibaba-cloud-csi-driver/pkg/features"
@@ -249,6 +250,9 @@ func newDiskStatusWaiter() DiskStatusWaiter {
 		return &SimpleDiskStatusWaiter{}
 	} else {
 		waiter := NewBatchedDiskStatusWaiter(GlobalConfigVar.EcsClient, clock.RealClock{})
+		waiter.PollHook = func() cloud.ECSInterface {
+			return updateEcsClient(GlobalConfigVar.EcsClient)
+		}
 		go waiter.Run(context.Background())
 		return waiter
 	}

pkg/disk/wait_status.go

@@ -34,6 +34,8 @@ type waitRequest struct {
 
 type BatchedDiskStatusWaiter struct {
 	ecsClient cloud.ECSInterface
+	// remove this once we have a ecsClient that can refresh its credentials
+	PollHook func() cloud.ECSInterface
 
 	requestChan chan *waitRequest
 	requests    map[string][]*waitRequest
@@ -91,6 +93,9 @@ func (w *BatchedDiskStatusWaiter) Run(ctx context.Context) {
 				pollChan = w.clk.After(pollInterval - w.clk.Since(lastPollTime))
 			}
 		case t := <-pollChan:
+			if w.PollHook != nil {
+				w.ecsClient = w.PollHook()
+			}
 			var err error
 			w.idQueue, err = w.poll(w.idQueue)
 			lastPollTime = t