wait only when state is loading

shamirchen · shamirchen · commit 0834da190d52 · 2025-09-25T14:07:00.000+08:00
diff --git a/velox/dwio/common/DirectBufferedInput.h b/velox/dwio/common/DirectBufferedInput.h
@@ -103,7 +103,7 @@ class DirectCoalescedLoad : public cache::CoalescedLoad {
 
   void cancel() override {
     folly::SemiFuture<bool> waitFuture(false);
-    if (!loadOrFuture(&waitFuture)) {
+    if (state() == State::kLoading && !loadOrFuture(&waitFuture)) {
       waitFuture.wait();
     }
     for (auto& request : requests_) {

Original file line number	Diff line number	Diff line change
`@@ -103,7 +103,7 @@ class DirectCoalescedLoad : public cache::CoalescedLoad {`
`103`	`103`
`104`	`104`	`void cancel() override {`
`105`	`105`	`folly::SemiFuture<bool> waitFuture(false);`
`106`		`- if (!loadOrFuture(&waitFuture)) {`
	`106`	`+ if (state() == State::kLoading && !loadOrFuture(&waitFuture)) {`
`107`	`107`	`waitFuture.wait();`
`108`	`108`	`}`
`109`	`109`	`for (auto& request : requests_) {`