fix goroutine leak on fetch sleep

hashicorp · Sep 29, 2022 · c677df8 · c677df8
1 parent 7e3c9ba
commit c677df8
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/dependency/vault_read.go b/dependency/vault_read.go
@@ -61,7 +61,12 @@ func (d *VaultReadQuery) Fetch(clients *ClientSet, opts *QueryOptions,
 	}
 	select {
 	case dur := <-d.sleepCh:
-		time.Sleep(dur)
+		select {
+		case <-time.After(dur):
+			break
+		case <-d.stopCh:
+			return nil, nil, ErrStopped
+		}
 	default:
 	}