Remove recursion in csiAttacher#waitForVolumeAttachmentInternal

This commit is contained in:
Ted Yu 2019-07-23 14:29:22 -07:00 committed by Ted Yu
parent 5b496fe8f5
commit f4fbfcb46a
2 changed files with 5 additions and 11 deletions

View File

@ -169,13 +169,9 @@ func (c *csiAttacher) waitForVolumeAttachmentInternal(volumeHandle, attachID str
if err != nil {
return "", fmt.Errorf("watch error:%v for volume %v", err, volumeHandle)
}
var watcherClosed bool
ch := watcher.ResultChan()
defer func() {
if !watcherClosed {
watcher.Stop()
}
}()
defer watcher.Stop()
for {
select {
@ -201,11 +197,7 @@ func (c *csiAttacher) waitForVolumeAttachmentInternal(volumeHandle, attachID str
return "", errors.New("volume attachment has been deleted")
case watch.Error:
// close the watcher to avoid keeping the watcher too log
watcher.Stop()
watcherClosed = true
// start another cycle
return c.waitForVolumeAttachmentInternal(volumeHandle, attachID, timer, timeout)
klog.Warningf("waitForVolumeAttachmentInternal received watch error: %v", event)
}
case <-timer.C:

View File

@ -221,6 +221,8 @@ func TestAttacherAttach(t *testing.T) {
status.AttachError = &storage.VolumeError{
Message: "attacher error",
}
errStatus := apierrs.NewInternalError(fmt.Errorf("we got an error")).Status()
fakeWatcher.Error(&errStatus)
} else {
status.Attached = true
}