Reduce busy-waiting when query contains pipeline blocking operators
- Dominant language
- Rust
- Stars
- 9.3k
- Forks
- 2.4k
- Avg merge
- 3d 7h
- Merged PRs (30d)
- 344
Description
### Is your feature request related to a problem or challenge?
When a query pipeline contains one or more pipeline blockers, the query will spend an extended period of time in the blocking phase of the query before it starts to emit data. During this time, polling the query stream can either block or return `Pending`.
Today quite a few pipeline blocking operators will block. While efficient in terms of polling, this prevents the query from being cancelled. The two possible solutions to this cancellation problem PR #16196 and #16301 fix this by ensuring query pipelines yield sufficiently providing the caller the opportunity to cancel.
A side effect of these changes (for both PR variants) is that the caller will now receive `Pending` results much more frequently effectively forcing the caller into a busy-waiting loop. The `Pending`s do not arrive at such a fast rate that this is overly problematic, but this is still rather wasteful.
As an illustration of this, I added wrapper stream in the CLI that prints out the poll results over time. Using a larger version of the test CSV file `main` at the time of writing shows `P` (`Pending`), `B` (`Ready(Some(OK(_)))`), and `` (`Ready(None)`).
```
> select a from annotated_data_infinite2 order by b desc limit 10;
PB
```
with the changes from the referenced PRs you get this instead.
```
> select a from annotated_data_infinite2 order by b desc limit 10;
PPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPPB
```
For complex queries, with many nested pipeline blockers, the polling call stack can get pretty deep making the waste larger. The `Pending` results typically will originate from the deepest point of the call stack. Every time this happens, the call stack is unwound and the polling call stack is rebuilt.
In an ideal world we would only return `Pending` once and wake the root caller when there is actually work to be done.
### Describe the solution you'd like
If a query is performing a long running task internally it should return `Pending` once and wake the caller only when data is actually available.
### Describe alternatives you've considered
None yet.
### Additional context
_No response_
Contributor guide
Research direction
Start by reading PRs #16196 and #16301 to understand the cancellation and yielding changes, then reproduce the issue with the CLI query against the larger test CSV while observing Pending and Ready results. The work is done when long-running blocking phases yield once and wake the root caller only when data is available, without preventing cancellation.
Written by the indexing model from the issue text.
Assessment
- Tech stack
- rust
- Domain
- backend-api-design, data-engineering
- Issue type
- Feature
- Difficulty
- 5/5
- Estimated time
- Over a week
- Activity status
- Stale
- Clarity
- Mostly clear
- Newbie friendliness
- 35/100