Merge pull request kubernetes-csi#907 from andyzhangx/fix-goroutine-leak

fix: goroutine leak when timeout

Author: andyzhangx

go.mod

@@ -9,6 +9,7 @@ require (
 	github.com/onsi/gomega v1.37.0
 	github.com/pborman/uuid v1.2.1
 	github.com/stretchr/testify v1.10.0
+	go.uber.org/goleak v1.3.0
 	golang.org/x/net v0.39.0
 	google.golang.org/grpc v1.72.0
 	google.golang.org/protobuf v1.36.6

pkg/nfs/utils.go

@@ -272,7 +272,7 @@ type TimeoutFunc func() (err error)
 // WaitUntilTimeout waits for the exec function to complete or return timeout error
 func WaitUntilTimeout(timeout time.Duration, execFunc ExecFunc, timeoutFunc TimeoutFunc) error {
 	// Create a channel to receive the result of the exec function
-	done := make(chan bool)
+	done := make(chan bool, 1)
 	var err error
 
 	// Start the exec function in a goroutine

pkg/nfs/utils_test.go

@@ -23,6 +23,8 @@ import (
 	"strings"
 	"testing"
 	"time"
+
+	"go.uber.org/goleak"
 )
 
 var (
@@ -449,6 +451,7 @@ func TestRemoveEmptyDirs(t *testing.T) {
 }
 
 func TestWaitUntilTimeout(t *testing.T) {
+	defer goleak.VerifyNone(t)
 	tests := []struct {
 		desc        string
 		timeout     time.Duration
@@ -494,8 +497,11 @@ func TestWaitUntilTimeout(t *testing.T) {
 
 	for _, test := range tests {
 		err := WaitUntilTimeout(test.timeout, test.execFunc, test.timeoutFunc)
-		if err != nil && (err.Error() != test.expectedErr.Error()) {
-			t.Errorf("unexpected error: %v, expected error: %v", err, test.expectedErr)
+		if err != nil {
+			time.Sleep(2 * time.Second)
+			if err.Error() != test.expectedErr.Error() {
+				t.Errorf("unexpected error: %v, expected error: %v", err, test.expectedErr)
+			}
 		}
 	}
 }