Resolve lost wakeup in SpillPoolReader with multiple concurrent SpillPoolWriters - #23522
Conversation
That is interesting because I constantly facing the "Too many open files" when there is a heavy spill. I did not know it is a problem and I was thinking it is OK, just used |
| if !shared.current_write_files.is_empty() { | ||
| // Copy and clear `current_write_files` so we can release shared lock before locking files | ||
| let files = shared.current_write_files.clone(); | ||
| shared.current_write_files.clear(); |
There was a problem hiding this comment.
Do we really need to release shared lock before locking files
Does it make sense to take file out from current_write_files consecutively then we could avoid clone?
There was a problem hiding this comment.
Do we really need to release shared lock before locking files
I'm not entirely sure. My assumption was that this was already carefully tuned, so I didn't want to change the lock nesting. If I understood it correctly, the intention is to avoid IO while the shared lock is held. InProgressSpilFile::finish can do IO so the shared lock is released before calling finish.
Does it make sense to take file out from current_write_files consecutively then we could avoid clone?
The clone/clear is actually kind of pointless. Might as well just mem::take the entire VecDequeue here. I've made that change.
There was a problem hiding this comment.
I'm not entirely sure. My assumption was that this was already carefully tuned, so I didn't want to change the lock nesting. If I understood it correctly, the intention is to avoid IO while the shared lock is held. InProgressSpilFile::finish can do IO so the shared lock is released before calling finish.
we also need to be sure we don't have a deadlock due to lock inversion (aka try to take the file/shared locks in different orders across different code paths)
| file_shared.writer_finished = true; | ||
| // Wake reader waiting on this file (it's now finished) | ||
| file_shared.wake(); | ||
| // Don't put back current_write_file - let it rotate |
There was a problem hiding this comment.
| // Don't put back current_write_file - let it rotate | |
| // Don't put back current_write_files - let it rotate |
There was a problem hiding this comment.
Singular is actually correct here, but I'll change the wording a bit to clarify what's being done.
alamb
left a comment
There was a problem hiding this comment.
This looks good to me -- thank you @pepijnve and @jayzhan211 (BTW 👋 it is great to see you again). I think adding some more comments explaining the expected state of the various queues would help make this code easier to understand in the future, but doesn't have to happen as part of this PR
cc @adriangb who added I think added SpillPoolShared in the first place
cc @xanderbailey who added some of this share pool machnery
| use super::spill_manager::SpillManager; | ||
|
|
||
| /// Shared state between the writer and readers of a spill pool. | ||
| /// This contains the queue of files and coordination state. |
There was a problem hiding this comment.
I this these comments would be nice to update to reflect the new design
Specifically note that the current_write_files is a list of files that are currently open but not being acively written to. When a file is being actively written to, active writer count is incremented by one and an entry is removed from current_write_files
There was a problem hiding this comment.
I'll do another pass over the code and update the comments. I didn't pay sufficient attention to those.
|
|
||
| // No files in queue - check if writer is done | ||
| if shared.writer_dropped { | ||
| if shared.active_writer_count == 0 { |
There was a problem hiding this comment.
I wonder (maybe as a follow on PR) if adding some sort of RAAI guard for updating active_writer_count would make the code les error prone.
Or maybe even having a separate list of files being actively written (rather than just a count 🤔 )
There was a problem hiding this comment.
The SpillPoolWriter plays the role of RAAI guard itself, doesn't it? It's incremented when a SpillPoolWriter is cloned, and decremented on drop.
The name active_writer_count is a bit misleading. It does not reflect the number of writers currently writing a batch. It reflects the number of writers that still exist.
| if !shared.current_write_files.is_empty() { | ||
| // Copy and clear `current_write_files` so we can release shared lock before locking files | ||
| let files = shared.current_write_files.clone(); | ||
| shared.current_write_files.clear(); |
There was a problem hiding this comment.
I'm not entirely sure. My assumption was that this was already carefully tuned, so I didn't want to change the lock nesting. If I understood it correctly, the intention is to avoid IO while the shared lock is held. InProgressSpilFile::finish can do IO so the shared lock is released before calling finish.
we also need to be sure we don't have a deadlock due to lock inversion (aka try to take the file/shared locks in different orders across different code paths)
|
I've updated the documentation and code comments for correctness. I've deemphasised FIFO semantics quite a bit. The existing documentation stated that this was always guaranteed, but that was not actually the case. Prior to this PR the following chain of events could happen:
The reader will observe batches in the order This MR assumes that we can take this even further, since the FIFO guarantee already doesn't hold when multiple writers are present. Would be good to get some feedback from @adriangb indeed since the non-FIFO behaviour was already present in the first commit of this code unless I'm completely misreading it. I'm going to try to write a test that demonstrates the behaviour. |
|
I'm worried about dropping the FIFO semantics (even if they were broken). |
I'm probably missing a bit too much context. Could you help me define the exact order guarantee that's being promised by the spill pool? Single writer remains FIFO, but in the presence of multiple concurrent writers, what order can you actually guarantee for the reader? The best you can do I think is guarantee that the relative order of the batches per writer is retained. Is that sufficient? |
|
BTW, I had based myself a bit on this comment in When |
|
That makes sense, I was just checking that and came to the same conclusion. As long as SPSC preserves ordering / FIFO we are good 👍🏻 |
The spill pool's FIFO guarantee only holds for a single writer: with multiple concurrent `SpillPoolWriter` clones the reader can observe batches out of write order. That is fine for non-preserve-order `RepartitionExec` (the output is an unordered multiset), but for `preserve_order = true` the per-(input, output) stream feeds an order-sensitive `StreamingMerge`, so losing FIFO would silently produce wrong (unsorted) results. Previously the "single writer per ordered pool" invariant was upheld only by convention: one `preserve_order` bool drove two independent decisions (channel count vs. writer cloning) in two places, coupled only by comments. A future edit could break one without the other. Encode the invariant in the type system instead: - `channel()` now returns `SpillPoolWriter`, which is **not** `Clone`, so an ordered pool can only ever have one writer (enforced at compile time). It wraps the shared implementation and delegates `push_batch`. - `shared_channel()` returns the `Clone` `SharedSpillPoolWriter` for the multi-producer, per-writer-FIFO case. - `RepartitionExec` selects the topology in one place: `preserve_order` builds one dedicated ordered writer per input (moved, never cloned) via `PartitionSpillWriters::PerInput`; non-preserve builds one shared writer cloned across inputs via `PartitionSpillWriters::Shared`. Now feeding an ordered pool with a shared multi-producer writer simply does not compile. Adds `test_preserve_order_with_spill_file_rotation`, which forces a spill file per batch (`max_spill_file_size_bytes = 1`) and asserts each output partition stays sorted — exercising FIFO across file rotation. Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com> Claude-Session: https://claude.ai/code/session_01WsjHVeKZtrSHugLwNFURjL
|
I've added a |
Enforce single-writer spill pools for preserve_order via the type system
|
Thanks for the suggestion @adriangb I was thinking of doing something similar as well. I'll handle the merge conflict with |
# Conflicts: # datafusion/physical-plan/src/repartition/mod.rs
…e SpillWriter wrapper type
|
I flipped around the nesting of I've edited the documentation a bit to be less verbose. Claude seemed kind of happy to repeat the fact that indeed SharedSpillPoolWriter is Clone over and over again. |
| let mut file_shared = file.lock(); | ||
|
|
||
| // Finish the current writer if it exists | ||
| if let Some(mut writer) = file_shared.writer.take() { |
|
@jayzhan211 is back! 🎉
|
alamb
left a comment
There was a problem hiding this comment.
This PR is looking really nice to me @pepijnve @adriangb and @jayzhan211
However, since the scope has significantly grown since I first reviewed the PR I don't think it would be a good idea to backport this change as is without some "bake time" on main first...
Perhaps we can apply the earlier (smaller) fix for 54?
| /// The set of spill-pool writers for a single output partition, before they are handed to the | ||
| /// per-input tasks. The variant encodes the repartition mode so the wrong writer topology cannot | ||
| /// be constructed for a given mode. | ||
| enum PartitionSpillWriters { |
|
Thank you for opening this pull request! Reviewer note: cargo-semver-checks reported the current version number is not SemVer-compatible with the changes in this pull request (compared against the base branch). Details |
I can go back and redo just the code changes and test additions on a back port branch. Makes sense to keep it as small as possible for a patch. |
Codecov Report❌ Patch coverage is Additional details and impacted files@@ Coverage Diff @@
## main #23522 +/- ##
=======================================
Coverage ? 80.65%
=======================================
Files ? 1086
Lines ? 366430
Branches ? 366430
=======================================
Hits ? 295561
Misses ? 53244
Partials ? 17625 ☔ View full report in Codecov by Harness. 🚀 New features to boost your workflow:
|
|
(I am resolving the CI failures now) |
Does every patch now require 100% coverage of changed lines? |
No! We are sorting through the coverage reports (and they are somewhat useless now b/c there isn't a base to compare against): |
That's a relief. In the meantime I've created the minimal backport PR version of this. |
|
This is a very nice improvement, thanks for working on this! |
#23654) ## Which issue does this PR close? - part of #22547 54.x branch backport of fix for #23447 ## Rationale for this change See main PR #23522 ## What changes are included in this PR? See main PR #23522 ## Are these changes tested? Additional test cases added to verify existing behaviour Fix manually tested with reproduction code from #23447 ## Are there any user-facing changes? No
| /// Maximum size in bytes before rotating to a new file. | ||
| /// Typically set from configuration `datafusion.execution.max_spill_file_size_bytes`. | ||
| max_file_size_bytes: usize, | ||
| /// Shared state with readers (includes current_write_file for coordination) |
There was a problem hiding this comment.
nit:
(includes current_write_file for coordination)
is stale
There was a problem hiding this comment.
Since this branch has been merged already it would be best to make a PR with a fix
…PoolWriters (apache#23522) ## Which issue does this PR close? - Closes apache#23447 ## Rationale for this change When multiple `SpillPoolWriter` clones concurrently push batches to the same channel, more than one non-finished `SpillFile` can be in flight. This happens because each `SpillPoolWriter` clone takes the `current_write_file` at the start of `push_batch` and puts it back when it's done. When multiple `push_batch` calls happen concurrently, only the first one will be able to take the `current_write_file` and the others will all create their own new spill file. Which one gets put back for subsequent use is a race condition. If this occurred and the writers are all dropped before rotation happens in, multiple files in the `files` deque will be have `writer_finished == false`. The last writer drop logic in `SpillPoolWriter::drop` only finishes whatever file is the `current_write_file` as finished. This can lead to a stalled situation when `SpillPoolFile::poll_next` catches up with the writer and returns `Pending` because `writer_finished == false`. A waker for the file is registered, but since the last writer drop logic only finishes and wakes whatever happens to be `current_write_file`, which may not be the current read file, the waker may end up never being notified. There is a secondary waker that is registered on the spill pool itself, but due to fine grained locking, it is possible for the wake call in the last writer drop logic to be called before the waker registration. ## What changes are included in this PR? - Add support for tracking multiple unfinished write files - Close all unfinished write files when the last writer is dropped - Removed `writer_dropped` field which was an unnecessary denormalisation of `active_writer_count == 0` An additional benefit of tracking all unfinished write files is that excessive creation of tiny spill files is avoided when many writers are pushing batches concurrently. ## Are these changes tested? Reproduction case from linked issue was used to confirm fix ## Are there any user-facing changes? No --------- Co-authored-by: Adrian Garcia Badaracco <1755071+adriangb@users.noreply.github.com> Co-authored-by: Claude Opus 4.8 (1M context) <noreply@anthropic.com> Co-authored-by: Andrew Lamb <andrew@nerdnetworks.org>
…PC-DS validation (#62) * fix(bench,deploy): SF10-distributed harness, CFN ShufflePartitions, bootstrap fixes - register-glue.sh: fix SIGPIPE false-empty on multi-file table listings (grep -q closed the pipe early under pipefail; multi-file SF10 tables were falsely reported empty and skipped) - weft-cluster.yaml + deploy-stack.sh: ShufflePartitions parameter wired to the weft:shuffle-partitions tag (was hardcoded to WorkerCount) - remeasure-distributed.sh: SF-agnostic (SF env, GLUE_DB_* defaults derive from SF, --sf passthrough) - run-spark-connect.py: --query-timeout (env WEFT_QUERY_TIMEOUT) wraps collect in a ThreadPoolExecutor; on timeout the query fails fast and the Spark session is recreated so a wedged query can't hang the suite - packer bootstrap.sh: fix enable_role_unit deadlock + spill device (the shipped AMI still has the buggy copy; rebake pending) - docs: SF100/SF10 topology, memory invariant, WEFT_SHUFFLE_PARTITIONS - SF10-DISTRIBUTED-RESUME.md: live-cluster handoff state * fix(execution): distributed stage lifecycle, spill, and correctness fixes (KAN-15..33) - KAN-15/17: remove 30s Flight timeout, narrow is_retryable, WEFT_STAGE_TIMEOUT_MS, cancel path end-to-end - KAN-18/19: stage cleanup on all exit paths, ShuffleInputGuard, WEFT_STAGE_OUTPUT_TTL_SECS - KAN-20: completed_ops cap/TTL, real ReleaseExecute - KAN-21/23: worker-global spill budget + incremental .segN appends - KAN-22: correlated scalar subquery decorrelation (Q2 class) - KAN-25: hash-join build guard WEFT_HASH_JOIN_MAX_BUILD_FRACTION - KAN-26: Q12/Q13 join chains, agg-over-agg - KAN-27: one-row broadcast via __WEFT_SCALAR_STAGE__ literal injection - KAN-28: empty results return typed zero rows - KAN-29: semi/anti/IN decorrelation, Q17 residual, Q15 derived scalar, strict whole-fact-gather floor - KAN-32: intermediate bucket dispatch at 16 partitions/2 workers, streaming producers, per-partition dispatch, AQE row-count action - KAN-33: Q13/Q16/Q20/Q22 distributed shapes Regression tests: auto_distribute_decorrelate, auto_distribute_semi_anti, auto_distribute_kan33, distributed_high_card_groupby, stage_timeout, stage_cancel, stage_cleanup, worker_join_memory_guard. * fix(loom): key catalog provider cache by shard context (KAN-35) The catalog bridge cached resolved table providers by table name only, but a provider bakes in the shard/replicate decision from first resolution. Per-query auto-broadcast classification flips table roles between queries, so stale providers served full scans where the plan assumed sharded (rows x worker count — TPC-H Q4 exactly 2x at SF10) or shards where the plan assumed replicated (~0.5x loss in Q5/Q7, 0.25x in Q9). Cache is now keyed by (table_name, is_replicated). Regression test: crates/weft-loom/tests/catalog_shard_context.rs (role-flip in both directions; fails with the old name-only key). * fix(connect): distributed strict-mode surface + result-path alignment (KAN-15..33 support) * fix(bench): SF-agnostic remeasure harness, query-timeout flag, Q11 fraction (KAN-30) - run-spark-connect: --query-timeout, --skip-worker-preflight, jsonl results - q11.sql: HAVING fraction now 0.0001 / __WEFT_SF__ (TPC-H spec) - weft-bench: tpch-distributed WEFT_TPCH_ONLY/DEBUG, SF-agnostic remeasure - docs/runtime-contract.md updated to match * fix(execution): distribute Q13/Q22 under size-based auto-broadcast (KAN-36, KAN-38) The KAN-33 shapes only handled both-tables-sharded configs. The real cluster's resolve_replicated_tables replicates all but the largest table, so Q13/Q22 run with customer replicated and orders sharded: - Q13: relaxed broadcast for null-extended-side-sharded LEFT/RIGHT joins when every aggregate reads only null-extended-side columns (NULL extension contributes nothing to partials); routes the agg-over-agg composition at sharded.len()==1. 3-stage plan. - Q22: classify_scalar_conjunct accepts fully-replicated scalar bodies (partial runs once via ExchangeMode::Forward); body_projection allowed over fully-replicated outer, hash-shuffled by shared key. 6-stage plan. Regression tests: crates/weft-execution/tests/auto_distribute_kan36.rs (5 tests; fail before/pass after; 2-worker Flight e2e == single-node). tpch-distributed sf0.01: 22/22 distributed-ok, 0 mismatch. * fix(execution): serve only same-schema shuffle entries; lock AVG precision (KAN-39, KAN-40) KAN-39: Worker::read_shuffle unioned every stage-cache entry matching a stage id and took the first entry's schema. Stage ids repeat across queries, and a client-timed-out query's producer could insert after the driver's cleanup raced past it, so the next same-id stage served mixed-schema bucket sets and register_batches failed with 'Mismatch between schema and batches'. Now the freshest entry declares the served schema, benign nullability/metadata drift is aligned (align_batch_schema), and genuinely stale entries are skipped with a warning; plus a registration-side alignment guard in run_stage. KAN-40: not an engine bug — distributed AVG is bit-for-bit identical to single-node (Spark-faithful DECIMAL(19,6) typing); added distributed_avg.rs to lock distributed==single-node at full precision. Regression tests: stale_entry_from_prior_query_is_not_served (fails before the fix), crates/weft-execution/tests/distributed_avg.rs. * fix(execution): fuse grouped IN-subquery with outer aggregate (KAN-37) Q18's generic semi/anti plan shuffled the full 3-way customer-orders- lineitem join output (~60M wide rows at SF10) across 16 partitions, then grouped it all to discard everything but 624 orders — exceeding the 600s stage timeout. The outer sum(l_quantity) grouped by a key set containing o_orderkey IS the IN-subquery's per-key aggregate, so the fact never needs to join the dims: new try_in_agg_semi_join shape emits 3 stages (partial per-key agg -> combine+HAVING carrying r0 -> co-located join + exact local GROUP BY with sum(r0) tracking join multiplicity), strictly guarded so non-matching shapes fall through to the generic path. Live SF10: Q18 42.0s cold / 34.5s warm (was >600s), 100 rows, row-for-row exact vs DuckDB golden incl. ORDER BY/LIMIT boundary. Local sf3: 20.6s. Regression tests in auto_distribute_semi_anti.rs (plan shape fail-before, distributed==single-node, decline guards). * fix(execution): exact combine for agg-over-agg through CTE boundary (KAN-44) TPC-DS Q54 at SF10 split one customer's revenue across two shuffle partitions and never recombined by the group key: the single-sharded peels_to_inner_aggregate divert only fired for the KAN-36 null-extended outer-join shape, so Q54's inner GROUP BY c_customer_sk (inside the my_revenue CTE) fell through to the flat broadcast path and executed per worker over its local shard. The divert is now unconditional at sharded.len()==1 — partial hashed by inner key -> combine (one row per inner group) -> re-shuffle by outer key -> exact outer aggregate — subsuming the KAN-36 special case. Non-provable shapes (expression outer group keys, grouping sets) are declined instead of mis-planned. Regression tests: crates/weft-execution/tests/auto_distribute_kan44.rs (3 tests, fail before/pass after, distributed==single-node with customer rows spanning both shards). tpch-distributed 22/22 0 mismatch; TPC-DS execute ratchet 95/95 0 mismatch. * fix(connect): stream typed empty batch for zero-row distributed SQL results (KAN-42) The distributed SQL branch of base_relation_batches returned an empty Vec<RecordBatch> verbatim — every other result path honors the KAN-28 zero-row contract. For a zero-row distributed result (e.g. TPC-DS Q17), stream_batches emitted only ResultComplete: no ArrowBatch, no schema, no error, and PySpark died at 'assert table is not None'. Now mirrors the local path: re-derive the schema and push one typed empty batch. Regression test: distributed_zero_row_result_still_streams_typed_empty_batch (crates/weft-connect/tests/distributed_pyspark.rs) — fails before, passes after; asserts >=1 ArrowBatch response on a real ExecutePlan stream. * fix(execution): materialize replicated-only branches once; dedup identical CTE branches (KAN-41) TPC-DS Q58/Q78 inlined replicated-only aggregate branches (catalog_sales/ web_sales arms) in the per-partition outer stage, recomputing a full replicated-fact scan+join+aggregate 16x at SF10 (>600s); Q11 planned 4 structurally identical self-joined CTE branch DAGs (17 stages, each fact scanned 4x). dag_splitter now materializes replicated-only aggregate branches as single Forward stages (computed once; volatile branches and branches whose expression subqueries read sharded tables stay inline) and dedups identical CTE branches by plan fingerprint below SubqueryAlias. Q11: 17 -> 5 stages. Q72/Q93 investigated: no plan defect (already join-at-shard + partial aggregate; fast live-faithful) — harness artifact locally (all-MemTable workers lose stats/pushdown), slow-but-correct. Regression tests: crates/weft-execution/tests/auto_distribute_kan41.rs (5 tests, shape tests fail before/pass after) + 4 dag_splitter unit tests. Gates green; tpch-distributed 22/22; tpcds ratchets 95/99 + 95/95 held. * fix(loom,execution): make sort-merge fallback opt-in; fail fast over wedging (KAN-45) Root cause (local repro + live evidence): DataFusion 54 deadlocks under the bounded FairSpillPool — a spilling operator stalls while its consumer stops draining a RepartitionExec channel, parking the stage at 0% CPU until WEFT_STAGE_TIMEOUT_MS kills it (upstream delta-rs#4614 class; not SMJ-specific). The KAN-25 sort-merge fallback routed over-budget join builds into this path and turned them into 600s wedges. - WEFT_SORT_MERGE_FALLBACK (default false) gates the SMJ fallback at all three sites; over-budget builds now run hash and fail fast with an actionable error naming the knob - Engine::smj_fallback_enabled(); regression tests for default-off hash planning and actionable pool-overflow errors; guard tests serialized behind a tokio Mutex; worker_join_memory_guard opts in - docs/runtime-contract.md + CFN + Helm values recommend preferHashJoin Live SF10: Q93 19.7s golden-clean (was >600s). Q11/Q72 remain — same upstream deadlock class via spilling sort/agg, needs the DataFusion fix. * fix(connect): capture + stamp lakehouse snapshot pins on distributed stages (KAN-48) The Spark Connect distributed path left StageDef.lakehouse_snapshot_pins empty, so workers (which arm require_lakehouse_snapshot_pins) rejected every lakehouse scan: delta/iceberg TPC-H SF10 was 0/22 while parquet was 22/22. try_run_distributed now plans via logical_plan_with_lakehouse_snapshots and stamps the pins JSON on every stage (mirroring the bench plan_distributed path); DataFrame relation-tree path wraps to_plan in the new Engine::capture_lakehouse_snapshots. Regression tests: crates/weft-connect/tests/distributed_lakehouse.rs — SQL + DataFrame distributed group-by over a delta table match single-node (pre-fix: exact production 'omitted the snapshot pin' error), and the unpinned-scan guard stays armed. * chore(deps): datafusion 54.0.0 -> 54.1.0 (KAN-47) Picks up apache/datafusion#23522 (backported as #23654): lost-wakeup race in SpillPoolWriter with concurrent writers — the exact mechanism behind our bounded-FairSpillPool stage deadlock at SF10 (TPC-DS Q11/Q72/Q93, the Q58/Q93 join-config tradeoff, and the KAN-45 SMJ wedges). Also #23574: SMJ correctness fix on the KAN-25 fallback path. Verified before commit: full gates green; upgrade spike ran the wedging Q93 SMJ plan at 4GiB bounded pool — wedged on 54.0.0, completed 4/4 on 54.1.0. datafusion-proto aligned to 54.1.0 for the RecursiveQuery.schema breaking change (#23886). * feat(execution,loom): stage no-progress watchdog + stage observability (KAN-47 follow-up) Defense-in-depth for the DF spill-pool deadlock class: a per-stage monitor races alongside the wall-clock timeout and driver-cancel flag inside run_stage_bounded, sampling three progress signals — per-task batch heartbeat (output stages now stream instead of collect so they heartbeat too), memory-pool activity timestamp (new ProgressMemoryPool wrapper), and DF+shuffle spill bytes. If none advance for WEFT_STAGE_NO_PROGRESS_SECS (default 600) the stage aborts with an actionable KAN-47 error and frees its task slot, instead of burning the full timeout silently. Observability: per-stage summary line (stage/partition, batches, spill bytes, duration, status, last-progress age) matching existing eprintln style; WEFT_STAGE_NO_PROGRESS_SECS + WEFT_STAGE_TIMEOUT_MS documented in docs/runtime-contract.md. Tests: stall-abort at ~2s (was 60s wall timeout), slow-but-progressing not aborted, signal-logic units, summary-line fields; summary lines verified in distributed test logs. * feat(loom,execution): per-query join strategy selection + stall-retry (KAN-53) WEFT_PREFER_HASH_JOIN is now tri-state: auto (new default) chooses per query — hash when the KAN-25 build-budget estimate fits, plan-time SMJ reroute when it doesn't (allowed in auto mode now that DF 54.1.0 fixed the SpillPoolWriter deadlock); true/false keep the legacy session-wide force. Stats fallback chain: row count x width -> parquet total_byte_size -> hash fast path backstopped by the runtime pool-exhaustion retry. Stall-retry: a stage aborted by the KAN-47 no-progress watchdog is cleared of partial spill and retried ONCE with the join strategy flipped (with_join_strategy_flipped task-local, flip accounts for the plan-time reroute so the retry really is the opposite strategy); driver cancels and wall-clock timeouts stay final. Also fixes a latent KAN-25 bug: the SMJ fallback's cloned SessionConfig registered a fresh empty default catalog, wiping engine tables after any fallback (with_create_default_catalog_and_schema(false)). Tests: env parsing/override, auto small-build hash / large-build SMJ, runtime retry on underreported estimate, flip inversion, distributed auto-mode guard variant, flight E2E stall-retry (completes after flip; always-stall fails after exactly two attempts). Gates green; tpch 22/22; tpcds 95/99 + 95/95 with Q58 AND Q93 both execute-verified. * fix(loom): auto join mode reroutes to SMJ when build estimate is unknown (KAN-53 follow-up) Live SF10 evidence: with unknown Glue/S3 stats, auto mode picked hash, the unaccounted hash build (KAN-57) ballooned past the pool, and workers were OOM-killed — TPC-H Q16/Q21 regressed from always-passing. The runtime pool-exhaustion retry can't backstop this because the build is not pool-accounted, so the OOM killer fires first. New fallback order (bounded pool, auto or KAN-45 opt-in): (1) over-budget estimate => plan-time SMJ reroute; (2) UNKNOWN estimate => SMJ reroute (safe since DF 54.1.0 fixed the SMJ deadlock); (3) positive under-budget estimate => hash; (4) runtime exhaustion => one SMJ retry. Unbounded pool keeps hash always (unchanged). Tests: UnknownStatsExec/UnknownStatsTable stand-ins; unknown+bounded => SMJ with correct results (fails before), unknown+unbounded => hash, positive-fit still hash; all prior KAN-25/53 tests unchanged. Gates green; tpch 22/22; tpcds 95/99 + 95/95 (Q58+Q93 verified). * fix(execution): compact StringView buffers in shuffle buckets; loud spilled-read errors Two latent shuffle-spill bugs exposed by TPC-H Q16 at SF10 on the new stack (auto join + DF 54.1.0): 1. hash_partition's arrow take produces StringViewArray views retaining the source scan's entire string buffers (~140x bloat: 33MB for 4,765 rows). estimated_bucket_bytes counted retained buffers, tripping the 4GB shuffle spill threshold; reading bloated buckets at 8-16 concurrent tasks climbed worker RSS to 31GB -> OOM-kill. Fix: compact_views() gc()s Utf8View columns after partitioning so buckets carry only live strings. 2. BucketCache::read_partition served spilled-bucket read errors as legitimately empty buckets (unwrap_or_default): a stage retry adjacent to the spill lifecycle silently truncated c0 values -> wrong supplier_cnt with exact row count. Fix: read_partition returns Result; missing base file is still a legitimate empty bucket, anything else fails the ShuffleRead pull loudly with stage/partition context. Tests: string_view_buckets_do_not_retain_source_buffers (fails on HEAD); spill read-back tests updated. Gates green; tpch 22/22; tpcds 95/99 + 95/95. Live SF10: Q16 3/3 consecutive golden-exact row diffs, zero spill, no OOM, stage tasks ~20ms (was 5-7s). * feat(execution): reorder filtered dims first in inner-join chains (Q72 spill blowup) TPC-DS Q72 at SF10 never finished (>1800s, 36-38GB sort spill per worker): with no table statistics DataFusion keeps written join order, so stage 0 exploded catalog_sales x inventory on item_sk alone into ~1.1B row pairs before the selective single-table filters could apply, then external-sorted that intermediate. New reorder_filtered_dims_first (planner rewrite in plan/join_order.rs, hooked at the top of plan_distributed_logical): within each contiguous inner-join chain under a Filter, leaves carrying single-table filter conjuncts join first (greedy, dependency-checked against ON columns; no-op when order is unchanged). Q72's chain shrinks to ~1e5 rows before inventory is touched: 48.96s, zero spill, golden row-diff clean (was >1800s). Tests: 5 unit tests (fires, idempotent, no-filter no-op, already-optimal no-op, non-inner bail). Gates green; tpch 22/22; tpcds 95/99 + 95/95. Live SF10: Q72 48.96s golden-clean; Q11/Q78/Q93 controls clean. * feat(site,scripts): SF10 distributed results on the site + cluster lifecycle tooling - site/src/data/tpch.json: TPC-H SF10 distributed (2 workers) — 22/22, total 533s, golden-validated vs DuckDB - site/src/data/tpcds.json: replaces pending placeholder — 60/99 execute (total 1314s, avg 21.9s), 39 clean strict-mode refusals, zero timeouts - bench/sf100/results/to-site-sf10.py: generator from suite jsonls to site JSONs (provenance: no hand-entered figures); ENGINE_COLORS entry - PerformancePage: SF10 distributed framing, honest TPC-DS 60/99 wording, reproduce commands - scripts/sf10-{start,stop,attach-spill}.sh: ASG scale-to-0 cost control and 200G spill volume attach (KAN-57) - golden check/diff harness used for all validations --------- Co-authored-by: Vamsi Krishna Kothapalli <vamsi@Vamsis-Mac-mini.local> Co-authored-by: Vamsi Krishna Kothapalli <vamsi@Vamsis-Mac-mini.attlocal.net> Co-authored-by: Vamsi Krishna Kothapalli <vamsi@vamsis-mac-mini.tail08ec2b.ts.net>
…xes (#63) * fix(bench,deploy): SF10-distributed harness, CFN ShufflePartitions, bootstrap fixes - register-glue.sh: fix SIGPIPE false-empty on multi-file table listings (grep -q closed the pipe early under pipefail; multi-file SF10 tables were falsely reported empty and skipped) - weft-cluster.yaml + deploy-stack.sh: ShufflePartitions parameter wired to the weft:shuffle-partitions tag (was hardcoded to WorkerCount) - remeasure-distributed.sh: SF-agnostic (SF env, GLUE_DB_* defaults derive from SF, --sf passthrough) - run-spark-connect.py: --query-timeout (env WEFT_QUERY_TIMEOUT) wraps collect in a ThreadPoolExecutor; on timeout the query fails fast and the Spark session is recreated so a wedged query can't hang the suite - packer bootstrap.sh: fix enable_role_unit deadlock + spill device (the shipped AMI still has the buggy copy; rebake pending) - docs: SF100/SF10 topology, memory invariant, WEFT_SHUFFLE_PARTITIONS - SF10-DISTRIBUTED-RESUME.md: live-cluster handoff state * fix(execution): distributed stage lifecycle, spill, and correctness fixes (KAN-15..33) - KAN-15/17: remove 30s Flight timeout, narrow is_retryable, WEFT_STAGE_TIMEOUT_MS, cancel path end-to-end - KAN-18/19: stage cleanup on all exit paths, ShuffleInputGuard, WEFT_STAGE_OUTPUT_TTL_SECS - KAN-20: completed_ops cap/TTL, real ReleaseExecute - KAN-21/23: worker-global spill budget + incremental .segN appends - KAN-22: correlated scalar subquery decorrelation (Q2 class) - KAN-25: hash-join build guard WEFT_HASH_JOIN_MAX_BUILD_FRACTION - KAN-26: Q12/Q13 join chains, agg-over-agg - KAN-27: one-row broadcast via __WEFT_SCALAR_STAGE__ literal injection - KAN-28: empty results return typed zero rows - KAN-29: semi/anti/IN decorrelation, Q17 residual, Q15 derived scalar, strict whole-fact-gather floor - KAN-32: intermediate bucket dispatch at 16 partitions/2 workers, streaming producers, per-partition dispatch, AQE row-count action - KAN-33: Q13/Q16/Q20/Q22 distributed shapes Regression tests: auto_distribute_decorrelate, auto_distribute_semi_anti, auto_distribute_kan33, distributed_high_card_groupby, stage_timeout, stage_cancel, stage_cleanup, worker_join_memory_guard. * fix(loom): key catalog provider cache by shard context (KAN-35) The catalog bridge cached resolved table providers by table name only, but a provider bakes in the shard/replicate decision from first resolution. Per-query auto-broadcast classification flips table roles between queries, so stale providers served full scans where the plan assumed sharded (rows x worker count — TPC-H Q4 exactly 2x at SF10) or shards where the plan assumed replicated (~0.5x loss in Q5/Q7, 0.25x in Q9). Cache is now keyed by (table_name, is_replicated). Regression test: crates/weft-loom/tests/catalog_shard_context.rs (role-flip in both directions; fails with the old name-only key). * fix(connect): distributed strict-mode surface + result-path alignment (KAN-15..33 support) * fix(bench): SF-agnostic remeasure harness, query-timeout flag, Q11 fraction (KAN-30) - run-spark-connect: --query-timeout, --skip-worker-preflight, jsonl results - q11.sql: HAVING fraction now 0.0001 / __WEFT_SF__ (TPC-H spec) - weft-bench: tpch-distributed WEFT_TPCH_ONLY/DEBUG, SF-agnostic remeasure - docs/runtime-contract.md updated to match * fix(execution): distribute Q13/Q22 under size-based auto-broadcast (KAN-36, KAN-38) The KAN-33 shapes only handled both-tables-sharded configs. The real cluster's resolve_replicated_tables replicates all but the largest table, so Q13/Q22 run with customer replicated and orders sharded: - Q13: relaxed broadcast for null-extended-side-sharded LEFT/RIGHT joins when every aggregate reads only null-extended-side columns (NULL extension contributes nothing to partials); routes the agg-over-agg composition at sharded.len()==1. 3-stage plan. - Q22: classify_scalar_conjunct accepts fully-replicated scalar bodies (partial runs once via ExchangeMode::Forward); body_projection allowed over fully-replicated outer, hash-shuffled by shared key. 6-stage plan. Regression tests: crates/weft-execution/tests/auto_distribute_kan36.rs (5 tests; fail before/pass after; 2-worker Flight e2e == single-node). tpch-distributed sf0.01: 22/22 distributed-ok, 0 mismatch. * fix(execution): serve only same-schema shuffle entries; lock AVG precision (KAN-39, KAN-40) KAN-39: Worker::read_shuffle unioned every stage-cache entry matching a stage id and took the first entry's schema. Stage ids repeat across queries, and a client-timed-out query's producer could insert after the driver's cleanup raced past it, so the next same-id stage served mixed-schema bucket sets and register_batches failed with 'Mismatch between schema and batches'. Now the freshest entry declares the served schema, benign nullability/metadata drift is aligned (align_batch_schema), and genuinely stale entries are skipped with a warning; plus a registration-side alignment guard in run_stage. KAN-40: not an engine bug — distributed AVG is bit-for-bit identical to single-node (Spark-faithful DECIMAL(19,6) typing); added distributed_avg.rs to lock distributed==single-node at full precision. Regression tests: stale_entry_from_prior_query_is_not_served (fails before the fix), crates/weft-execution/tests/distributed_avg.rs. * fix(execution): fuse grouped IN-subquery with outer aggregate (KAN-37) Q18's generic semi/anti plan shuffled the full 3-way customer-orders- lineitem join output (~60M wide rows at SF10) across 16 partitions, then grouped it all to discard everything but 624 orders — exceeding the 600s stage timeout. The outer sum(l_quantity) grouped by a key set containing o_orderkey IS the IN-subquery's per-key aggregate, so the fact never needs to join the dims: new try_in_agg_semi_join shape emits 3 stages (partial per-key agg -> combine+HAVING carrying r0 -> co-located join + exact local GROUP BY with sum(r0) tracking join multiplicity), strictly guarded so non-matching shapes fall through to the generic path. Live SF10: Q18 42.0s cold / 34.5s warm (was >600s), 100 rows, row-for-row exact vs DuckDB golden incl. ORDER BY/LIMIT boundary. Local sf3: 20.6s. Regression tests in auto_distribute_semi_anti.rs (plan shape fail-before, distributed==single-node, decline guards). * fix(execution): exact combine for agg-over-agg through CTE boundary (KAN-44) TPC-DS Q54 at SF10 split one customer's revenue across two shuffle partitions and never recombined by the group key: the single-sharded peels_to_inner_aggregate divert only fired for the KAN-36 null-extended outer-join shape, so Q54's inner GROUP BY c_customer_sk (inside the my_revenue CTE) fell through to the flat broadcast path and executed per worker over its local shard. The divert is now unconditional at sharded.len()==1 — partial hashed by inner key -> combine (one row per inner group) -> re-shuffle by outer key -> exact outer aggregate — subsuming the KAN-36 special case. Non-provable shapes (expression outer group keys, grouping sets) are declined instead of mis-planned. Regression tests: crates/weft-execution/tests/auto_distribute_kan44.rs (3 tests, fail before/pass after, distributed==single-node with customer rows spanning both shards). tpch-distributed 22/22 0 mismatch; TPC-DS execute ratchet 95/95 0 mismatch. * fix(connect): stream typed empty batch for zero-row distributed SQL results (KAN-42) The distributed SQL branch of base_relation_batches returned an empty Vec<RecordBatch> verbatim — every other result path honors the KAN-28 zero-row contract. For a zero-row distributed result (e.g. TPC-DS Q17), stream_batches emitted only ResultComplete: no ArrowBatch, no schema, no error, and PySpark died at 'assert table is not None'. Now mirrors the local path: re-derive the schema and push one typed empty batch. Regression test: distributed_zero_row_result_still_streams_typed_empty_batch (crates/weft-connect/tests/distributed_pyspark.rs) — fails before, passes after; asserts >=1 ArrowBatch response on a real ExecutePlan stream. * fix(execution): materialize replicated-only branches once; dedup identical CTE branches (KAN-41) TPC-DS Q58/Q78 inlined replicated-only aggregate branches (catalog_sales/ web_sales arms) in the per-partition outer stage, recomputing a full replicated-fact scan+join+aggregate 16x at SF10 (>600s); Q11 planned 4 structurally identical self-joined CTE branch DAGs (17 stages, each fact scanned 4x). dag_splitter now materializes replicated-only aggregate branches as single Forward stages (computed once; volatile branches and branches whose expression subqueries read sharded tables stay inline) and dedups identical CTE branches by plan fingerprint below SubqueryAlias. Q11: 17 -> 5 stages. Q72/Q93 investigated: no plan defect (already join-at-shard + partial aggregate; fast live-faithful) — harness artifact locally (all-MemTable workers lose stats/pushdown), slow-but-correct. Regression tests: crates/weft-execution/tests/auto_distribute_kan41.rs (5 tests, shape tests fail before/pass after) + 4 dag_splitter unit tests. Gates green; tpch-distributed 22/22; tpcds ratchets 95/99 + 95/95 held. * fix(loom,execution): make sort-merge fallback opt-in; fail fast over wedging (KAN-45) Root cause (local repro + live evidence): DataFusion 54 deadlocks under the bounded FairSpillPool — a spilling operator stalls while its consumer stops draining a RepartitionExec channel, parking the stage at 0% CPU until WEFT_STAGE_TIMEOUT_MS kills it (upstream delta-rs#4614 class; not SMJ-specific). The KAN-25 sort-merge fallback routed over-budget join builds into this path and turned them into 600s wedges. - WEFT_SORT_MERGE_FALLBACK (default false) gates the SMJ fallback at all three sites; over-budget builds now run hash and fail fast with an actionable error naming the knob - Engine::smj_fallback_enabled(); regression tests for default-off hash planning and actionable pool-overflow errors; guard tests serialized behind a tokio Mutex; worker_join_memory_guard opts in - docs/runtime-contract.md + CFN + Helm values recommend preferHashJoin Live SF10: Q93 19.7s golden-clean (was >600s). Q11/Q72 remain — same upstream deadlock class via spilling sort/agg, needs the DataFusion fix. * fix(connect): capture + stamp lakehouse snapshot pins on distributed stages (KAN-48) The Spark Connect distributed path left StageDef.lakehouse_snapshot_pins empty, so workers (which arm require_lakehouse_snapshot_pins) rejected every lakehouse scan: delta/iceberg TPC-H SF10 was 0/22 while parquet was 22/22. try_run_distributed now plans via logical_plan_with_lakehouse_snapshots and stamps the pins JSON on every stage (mirroring the bench plan_distributed path); DataFrame relation-tree path wraps to_plan in the new Engine::capture_lakehouse_snapshots. Regression tests: crates/weft-connect/tests/distributed_lakehouse.rs — SQL + DataFrame distributed group-by over a delta table match single-node (pre-fix: exact production 'omitted the snapshot pin' error), and the unpinned-scan guard stays armed. * chore(deps): datafusion 54.0.0 -> 54.1.0 (KAN-47) Picks up apache/datafusion#23522 (backported as #23654): lost-wakeup race in SpillPoolWriter with concurrent writers — the exact mechanism behind our bounded-FairSpillPool stage deadlock at SF10 (TPC-DS Q11/Q72/Q93, the Q58/Q93 join-config tradeoff, and the KAN-45 SMJ wedges). Also #23574: SMJ correctness fix on the KAN-25 fallback path. Verified before commit: full gates green; upgrade spike ran the wedging Q93 SMJ plan at 4GiB bounded pool — wedged on 54.0.0, completed 4/4 on 54.1.0. datafusion-proto aligned to 54.1.0 for the RecursiveQuery.schema breaking change (#23886). * feat(execution,loom): stage no-progress watchdog + stage observability (KAN-47 follow-up) Defense-in-depth for the DF spill-pool deadlock class: a per-stage monitor races alongside the wall-clock timeout and driver-cancel flag inside run_stage_bounded, sampling three progress signals — per-task batch heartbeat (output stages now stream instead of collect so they heartbeat too), memory-pool activity timestamp (new ProgressMemoryPool wrapper), and DF+shuffle spill bytes. If none advance for WEFT_STAGE_NO_PROGRESS_SECS (default 600) the stage aborts with an actionable KAN-47 error and frees its task slot, instead of burning the full timeout silently. Observability: per-stage summary line (stage/partition, batches, spill bytes, duration, status, last-progress age) matching existing eprintln style; WEFT_STAGE_NO_PROGRESS_SECS + WEFT_STAGE_TIMEOUT_MS documented in docs/runtime-contract.md. Tests: stall-abort at ~2s (was 60s wall timeout), slow-but-progressing not aborted, signal-logic units, summary-line fields; summary lines verified in distributed test logs. * feat(loom,execution): per-query join strategy selection + stall-retry (KAN-53) WEFT_PREFER_HASH_JOIN is now tri-state: auto (new default) chooses per query — hash when the KAN-25 build-budget estimate fits, plan-time SMJ reroute when it doesn't (allowed in auto mode now that DF 54.1.0 fixed the SpillPoolWriter deadlock); true/false keep the legacy session-wide force. Stats fallback chain: row count x width -> parquet total_byte_size -> hash fast path backstopped by the runtime pool-exhaustion retry. Stall-retry: a stage aborted by the KAN-47 no-progress watchdog is cleared of partial spill and retried ONCE with the join strategy flipped (with_join_strategy_flipped task-local, flip accounts for the plan-time reroute so the retry really is the opposite strategy); driver cancels and wall-clock timeouts stay final. Also fixes a latent KAN-25 bug: the SMJ fallback's cloned SessionConfig registered a fresh empty default catalog, wiping engine tables after any fallback (with_create_default_catalog_and_schema(false)). Tests: env parsing/override, auto small-build hash / large-build SMJ, runtime retry on underreported estimate, flip inversion, distributed auto-mode guard variant, flight E2E stall-retry (completes after flip; always-stall fails after exactly two attempts). Gates green; tpch 22/22; tpcds 95/99 + 95/95 with Q58 AND Q93 both execute-verified. * fix(loom): auto join mode reroutes to SMJ when build estimate is unknown (KAN-53 follow-up) Live SF10 evidence: with unknown Glue/S3 stats, auto mode picked hash, the unaccounted hash build (KAN-57) ballooned past the pool, and workers were OOM-killed — TPC-H Q16/Q21 regressed from always-passing. The runtime pool-exhaustion retry can't backstop this because the build is not pool-accounted, so the OOM killer fires first. New fallback order (bounded pool, auto or KAN-45 opt-in): (1) over-budget estimate => plan-time SMJ reroute; (2) UNKNOWN estimate => SMJ reroute (safe since DF 54.1.0 fixed the SMJ deadlock); (3) positive under-budget estimate => hash; (4) runtime exhaustion => one SMJ retry. Unbounded pool keeps hash always (unchanged). Tests: UnknownStatsExec/UnknownStatsTable stand-ins; unknown+bounded => SMJ with correct results (fails before), unknown+unbounded => hash, positive-fit still hash; all prior KAN-25/53 tests unchanged. Gates green; tpch 22/22; tpcds 95/99 + 95/95 (Q58+Q93 verified). * fix(execution): compact StringView buffers in shuffle buckets; loud spilled-read errors Two latent shuffle-spill bugs exposed by TPC-H Q16 at SF10 on the new stack (auto join + DF 54.1.0): 1. hash_partition's arrow take produces StringViewArray views retaining the source scan's entire string buffers (~140x bloat: 33MB for 4,765 rows). estimated_bucket_bytes counted retained buffers, tripping the 4GB shuffle spill threshold; reading bloated buckets at 8-16 concurrent tasks climbed worker RSS to 31GB -> OOM-kill. Fix: compact_views() gc()s Utf8View columns after partitioning so buckets carry only live strings. 2. BucketCache::read_partition served spilled-bucket read errors as legitimately empty buckets (unwrap_or_default): a stage retry adjacent to the spill lifecycle silently truncated c0 values -> wrong supplier_cnt with exact row count. Fix: read_partition returns Result; missing base file is still a legitimate empty bucket, anything else fails the ShuffleRead pull loudly with stage/partition context. Tests: string_view_buckets_do_not_retain_source_buffers (fails on HEAD); spill read-back tests updated. Gates green; tpch 22/22; tpcds 95/99 + 95/95. Live SF10: Q16 3/3 consecutive golden-exact row diffs, zero spill, no OOM, stage tasks ~20ms (was 5-7s). * feat(execution): reorder filtered dims first in inner-join chains (Q72 spill blowup) TPC-DS Q72 at SF10 never finished (>1800s, 36-38GB sort spill per worker): with no table statistics DataFusion keeps written join order, so stage 0 exploded catalog_sales x inventory on item_sk alone into ~1.1B row pairs before the selective single-table filters could apply, then external-sorted that intermediate. New reorder_filtered_dims_first (planner rewrite in plan/join_order.rs, hooked at the top of plan_distributed_logical): within each contiguous inner-join chain under a Filter, leaves carrying single-table filter conjuncts join first (greedy, dependency-checked against ON columns; no-op when order is unchanged). Q72's chain shrinks to ~1e5 rows before inventory is touched: 48.96s, zero spill, golden row-diff clean (was >1800s). Tests: 5 unit tests (fires, idempotent, no-filter no-op, already-optimal no-op, non-inner bail). Gates green; tpch 22/22; tpcds 95/99 + 95/95. Live SF10: Q72 48.96s golden-clean; Q11/Q78/Q93 controls clean. * feat(site,scripts): SF10 distributed results on the site + cluster lifecycle tooling - site/src/data/tpch.json: TPC-H SF10 distributed (2 workers) — 22/22, total 533s, golden-validated vs DuckDB - site/src/data/tpcds.json: replaces pending placeholder — 60/99 execute (total 1314s, avg 21.9s), 39 clean strict-mode refusals, zero timeouts - bench/sf100/results/to-site-sf10.py: generator from suite jsonls to site JSONs (provenance: no hand-entered figures); ENGINE_COLORS entry - PerformancePage: SF10 distributed framing, honest TPC-DS 60/99 wording, reproduce commands - scripts/sf10-{start,stop,attach-spill}.sh: ASG scale-to-0 cost control and 200G spill volume attach (KAN-57) - golden check/diff harness used for all validations * feat(execution): distribute subquery-over-sharded shapes (KAN-55) - resolve_replicated_tables now sizes subquery-scanned tables too, so per-query classification is uniform (subquery-only facts replicate) - replicated-subquery conjuncts route verbatim (partition-independent) - global aggregates above semi/anti filters: recombinable partials + partition-0 gate for COUNT(DISTINCT) (exact empty-input row) - Q9 scalar-projection shape: uncorrelated global-agg scalars over the sharded fact become per-worker partials + one-row combine, AST-rewritten projection evaluated once behind a partition-0 gate - anti-only inline guard: NOT EXISTS over sharded key stream with replicated outer now declines to gather (was per-partition duplicates) 6 of 8 queries now plan distributed at SF10 strict (Q9,Q10,Q16,Q35,Q69, Q94); Q95 (self-join IN) and Q14 (ROLLUP+INTERSECT) keep refusing with locking tests. 15 new tests in auto_distribute_kan55.rs, all fail-before. Gates green; tpch 22/22; tpcds ratchets 95/99 + 95/95. * feat(execution): distribute mixed sharded/replicated UNION ALL (KAN-54) Nested-union flattening in split_union_by_sharding and plan_union (bag union is associative), plus a SUM-only guard for pre-aggregated sharded arms. 6 queries newly plan distributed under the real SF10 strict classification (Q4,Q27,Q33,Q56,Q60,Q76), all execute-verified distributed==single-node. Q5/Q77/Q80 deliberately keep refusing (grouping-set composition has a documented wrong-answer trap). Tests: auto_distribute_kan54.rs (7 tests, fail-before). Remainder mapped: window-over-agg ROLLUP family next (Q36,Q44,Q47,Q51,Q57,Q67,Q70,Q86). * fix(execution,connect): cancel+cleanup on client disconnect; reap uncommitted spill (KAN-46, KAN-31) Dropping the driver query future mid-run (client disconnect) skipped all cleanup: cancel watcher leaked, worker stages kept running, buckets stayed cached (17.8-26.5GB idle RSS observed), and uncommitted producer spill (25-38GB) orphaned into later ENOSPC. QueryAbortGuard on Drop aborts the watcher and runs the cancel+clear sweep; StageSpillReaper reaps a stage's spill scope (base + .segN segments) whenever a producer exits uncommitted. Committed caches still await clear_stages/TTL (positive control). Tests: query_abort_cleanup.rs (4), client_disconnect.rs (2) incl. slow-but-connected client is NOT killed. All fail-before. * test(loom): pin Spark default null ordering (KAN-52) Spark default: ASC => NULLS FIRST, DESC => NULLS LAST (opposite of DuckDB/Postgres). weft already matched via datafusion.sql_parser.default_null_ordering=nulls_min; this dedups the config block and locks the contract with 7 tests (defaults, explicit overrides, ORDER BY...LIMIT boundary, window ORDER BY, distributed finalize replay). Verified RED when the config is disabled. * feat(bench): MATCH/BENIGN/MISMATCH golden verdicts; parquet ListingTable harness (KAN-50, KAN-51) golden_common.py: verdict ladder — checksum match, verified numeric-scale (round-half-even/trunc at decimal scale, reproduces the recorded hash), boundary-tie (LIMIT + tie-zone enumeration, null-ordering sensitivity), numeric-drift heuristic, MISMATCH; --self-test (27 checks) and --strict-benign for CI. TPC-H final: 20 MATCH + 2 BENIGN + 0 MISMATCH; TPC-DS final: 47 MATCH + 13 BENIGN + 0 MISMATCH. tpcds_dist workers register replicated tables as parquet ListingTables (real stats, dynamic-filter pushdown — live-faithful) instead of MemTables; Q72 harness run 51.6s -> 17.8s at SF10. Ratchets 95/99 + 95/95 held; tpch 22/22. * fix(deploy): close bootstrap systemd cycle; atomic env; symmetric DNS; name-agnostic spill (KAN-58) Bootstrap no longer issues role-unit start jobs from inside weft-bootstrap.service (the cycle-closing edge; Requires=/After= kept as the env+DNS-before-start guarantee; UserData starts the role unit after bootstrap completes). write_env is atomic (mktemp+mv). Worker DNS: boot sync always includes own IP (fixes missing-self on cold start); ExecStop removes only own IP (fixes stale zombie IPs -> 'no free task slots'). find_spill_device accepts any unmounted non-root disk, largest wins. TimeoutStartSec=600 for the 240s peer wait. Offline harness 23/23; bash -n clean. NOTE: AMI rebake needed to roll this out. * fix(execution): nested-IN semi cascade accepts replicated dims (KAN-55 follow-up) KAN-55's uniform classification (subquery tables sized too) flips Q20's supplier/partsupp/part from sharded-by-default to replicated at SF10, and try_nested_in_semi's shardedness checks declined — strict mode refused with a whole-lineitem gather. Relaxed: replicated outer/middle/nested scans export once via ExchangeMode::Forward (KAN-36 precedent); the correlated scalar body still requires a sharded fact (a replicated body would multiply the threshold by worker count). Tests: q20_sf10_replicated_dims_plans_semi_cascade (fail-before at the SF10 config: 8 stages, Forward on 2/3/6, no gather) + q20_sf10_replicated_dims_distributed_matches_single_node. Gates green; tpch 22/22; tpcds 95/99 + 95/95; all KAN-55 suites re-verified. * feat(site): TPC-DS 72/99 + TPC-H re-run on the KAN-49 wave-2 stack Site data regenerated from the wave-2 validation runs: TPC-DS 72/99 execute (was 60/99; 27 strict refusals remain), TPC-H 22/22 re-run. to-site-sf10.py now reads the wave-2 jsonls (Q20 dedup ok-preferred). --------- Co-authored-by: Vamsi Krishna Kothapalli <vamsi@Vamsis-Mac-mini.local> Co-authored-by: Vamsi Krishna Kothapalli <vamsi@Vamsis-Mac-mini.attlocal.net> Co-authored-by: Vamsi Krishna Kothapalli <vamsi@vamsis-mac-mini.tail08ec2b.ts.net>
* fix(bench,deploy): SF10-distributed harness, CFN ShufflePartitions, bootstrap fixes - register-glue.sh: fix SIGPIPE false-empty on multi-file table listings (grep -q closed the pipe early under pipefail; multi-file SF10 tables were falsely reported empty and skipped) - weft-cluster.yaml + deploy-stack.sh: ShufflePartitions parameter wired to the weft:shuffle-partitions tag (was hardcoded to WorkerCount) - remeasure-distributed.sh: SF-agnostic (SF env, GLUE_DB_* defaults derive from SF, --sf passthrough) - run-spark-connect.py: --query-timeout (env WEFT_QUERY_TIMEOUT) wraps collect in a ThreadPoolExecutor; on timeout the query fails fast and the Spark session is recreated so a wedged query can't hang the suite - packer bootstrap.sh: fix enable_role_unit deadlock + spill device (the shipped AMI still has the buggy copy; rebake pending) - docs: SF100/SF10 topology, memory invariant, WEFT_SHUFFLE_PARTITIONS - SF10-DISTRIBUTED-RESUME.md: live-cluster handoff state * fix(execution): distributed stage lifecycle, spill, and correctness fixes (KAN-15..33) - KAN-15/17: remove 30s Flight timeout, narrow is_retryable, WEFT_STAGE_TIMEOUT_MS, cancel path end-to-end - KAN-18/19: stage cleanup on all exit paths, ShuffleInputGuard, WEFT_STAGE_OUTPUT_TTL_SECS - KAN-20: completed_ops cap/TTL, real ReleaseExecute - KAN-21/23: worker-global spill budget + incremental .segN appends - KAN-22: correlated scalar subquery decorrelation (Q2 class) - KAN-25: hash-join build guard WEFT_HASH_JOIN_MAX_BUILD_FRACTION - KAN-26: Q12/Q13 join chains, agg-over-agg - KAN-27: one-row broadcast via __WEFT_SCALAR_STAGE__ literal injection - KAN-28: empty results return typed zero rows - KAN-29: semi/anti/IN decorrelation, Q17 residual, Q15 derived scalar, strict whole-fact-gather floor - KAN-32: intermediate bucket dispatch at 16 partitions/2 workers, streaming producers, per-partition dispatch, AQE row-count action - KAN-33: Q13/Q16/Q20/Q22 distributed shapes Regression tests: auto_distribute_decorrelate, auto_distribute_semi_anti, auto_distribute_kan33, distributed_high_card_groupby, stage_timeout, stage_cancel, stage_cleanup, worker_join_memory_guard. * fix(loom): key catalog provider cache by shard context (KAN-35) The catalog bridge cached resolved table providers by table name only, but a provider bakes in the shard/replicate decision from first resolution. Per-query auto-broadcast classification flips table roles between queries, so stale providers served full scans where the plan assumed sharded (rows x worker count — TPC-H Q4 exactly 2x at SF10) or shards where the plan assumed replicated (~0.5x loss in Q5/Q7, 0.25x in Q9). Cache is now keyed by (table_name, is_replicated). Regression test: crates/weft-loom/tests/catalog_shard_context.rs (role-flip in both directions; fails with the old name-only key). * fix(connect): distributed strict-mode surface + result-path alignment (KAN-15..33 support) * fix(bench): SF-agnostic remeasure harness, query-timeout flag, Q11 fraction (KAN-30) - run-spark-connect: --query-timeout, --skip-worker-preflight, jsonl results - q11.sql: HAVING fraction now 0.0001 / __WEFT_SF__ (TPC-H spec) - weft-bench: tpch-distributed WEFT_TPCH_ONLY/DEBUG, SF-agnostic remeasure - docs/runtime-contract.md updated to match * fix(execution): distribute Q13/Q22 under size-based auto-broadcast (KAN-36, KAN-38) The KAN-33 shapes only handled both-tables-sharded configs. The real cluster's resolve_replicated_tables replicates all but the largest table, so Q13/Q22 run with customer replicated and orders sharded: - Q13: relaxed broadcast for null-extended-side-sharded LEFT/RIGHT joins when every aggregate reads only null-extended-side columns (NULL extension contributes nothing to partials); routes the agg-over-agg composition at sharded.len()==1. 3-stage plan. - Q22: classify_scalar_conjunct accepts fully-replicated scalar bodies (partial runs once via ExchangeMode::Forward); body_projection allowed over fully-replicated outer, hash-shuffled by shared key. 6-stage plan. Regression tests: crates/weft-execution/tests/auto_distribute_kan36.rs (5 tests; fail before/pass after; 2-worker Flight e2e == single-node). tpch-distributed sf0.01: 22/22 distributed-ok, 0 mismatch. * fix(execution): serve only same-schema shuffle entries; lock AVG precision (KAN-39, KAN-40) KAN-39: Worker::read_shuffle unioned every stage-cache entry matching a stage id and took the first entry's schema. Stage ids repeat across queries, and a client-timed-out query's producer could insert after the driver's cleanup raced past it, so the next same-id stage served mixed-schema bucket sets and register_batches failed with 'Mismatch between schema and batches'. Now the freshest entry declares the served schema, benign nullability/metadata drift is aligned (align_batch_schema), and genuinely stale entries are skipped with a warning; plus a registration-side alignment guard in run_stage. KAN-40: not an engine bug — distributed AVG is bit-for-bit identical to single-node (Spark-faithful DECIMAL(19,6) typing); added distributed_avg.rs to lock distributed==single-node at full precision. Regression tests: stale_entry_from_prior_query_is_not_served (fails before the fix), crates/weft-execution/tests/distributed_avg.rs. * fix(execution): fuse grouped IN-subquery with outer aggregate (KAN-37) Q18's generic semi/anti plan shuffled the full 3-way customer-orders- lineitem join output (~60M wide rows at SF10) across 16 partitions, then grouped it all to discard everything but 624 orders — exceeding the 600s stage timeout. The outer sum(l_quantity) grouped by a key set containing o_orderkey IS the IN-subquery's per-key aggregate, so the fact never needs to join the dims: new try_in_agg_semi_join shape emits 3 stages (partial per-key agg -> combine+HAVING carrying r0 -> co-located join + exact local GROUP BY with sum(r0) tracking join multiplicity), strictly guarded so non-matching shapes fall through to the generic path. Live SF10: Q18 42.0s cold / 34.5s warm (was >600s), 100 rows, row-for-row exact vs DuckDB golden incl. ORDER BY/LIMIT boundary. Local sf3: 20.6s. Regression tests in auto_distribute_semi_anti.rs (plan shape fail-before, distributed==single-node, decline guards). * fix(execution): exact combine for agg-over-agg through CTE boundary (KAN-44) TPC-DS Q54 at SF10 split one customer's revenue across two shuffle partitions and never recombined by the group key: the single-sharded peels_to_inner_aggregate divert only fired for the KAN-36 null-extended outer-join shape, so Q54's inner GROUP BY c_customer_sk (inside the my_revenue CTE) fell through to the flat broadcast path and executed per worker over its local shard. The divert is now unconditional at sharded.len()==1 — partial hashed by inner key -> combine (one row per inner group) -> re-shuffle by outer key -> exact outer aggregate — subsuming the KAN-36 special case. Non-provable shapes (expression outer group keys, grouping sets) are declined instead of mis-planned. Regression tests: crates/weft-execution/tests/auto_distribute_kan44.rs (3 tests, fail before/pass after, distributed==single-node with customer rows spanning both shards). tpch-distributed 22/22 0 mismatch; TPC-DS execute ratchet 95/95 0 mismatch. * fix(connect): stream typed empty batch for zero-row distributed SQL results (KAN-42) The distributed SQL branch of base_relation_batches returned an empty Vec<RecordBatch> verbatim — every other result path honors the KAN-28 zero-row contract. For a zero-row distributed result (e.g. TPC-DS Q17), stream_batches emitted only ResultComplete: no ArrowBatch, no schema, no error, and PySpark died at 'assert table is not None'. Now mirrors the local path: re-derive the schema and push one typed empty batch. Regression test: distributed_zero_row_result_still_streams_typed_empty_batch (crates/weft-connect/tests/distributed_pyspark.rs) — fails before, passes after; asserts >=1 ArrowBatch response on a real ExecutePlan stream. * fix(execution): materialize replicated-only branches once; dedup identical CTE branches (KAN-41) TPC-DS Q58/Q78 inlined replicated-only aggregate branches (catalog_sales/ web_sales arms) in the per-partition outer stage, recomputing a full replicated-fact scan+join+aggregate 16x at SF10 (>600s); Q11 planned 4 structurally identical self-joined CTE branch DAGs (17 stages, each fact scanned 4x). dag_splitter now materializes replicated-only aggregate branches as single Forward stages (computed once; volatile branches and branches whose expression subqueries read sharded tables stay inline) and dedups identical CTE branches by plan fingerprint below SubqueryAlias. Q11: 17 -> 5 stages. Q72/Q93 investigated: no plan defect (already join-at-shard + partial aggregate; fast live-faithful) — harness artifact locally (all-MemTable workers lose stats/pushdown), slow-but-correct. Regression tests: crates/weft-execution/tests/auto_distribute_kan41.rs (5 tests, shape tests fail before/pass after) + 4 dag_splitter unit tests. Gates green; tpch-distributed 22/22; tpcds ratchets 95/99 + 95/95 held. * fix(loom,execution): make sort-merge fallback opt-in; fail fast over wedging (KAN-45) Root cause (local repro + live evidence): DataFusion 54 deadlocks under the bounded FairSpillPool — a spilling operator stalls while its consumer stops draining a RepartitionExec channel, parking the stage at 0% CPU until WEFT_STAGE_TIMEOUT_MS kills it (upstream delta-rs#4614 class; not SMJ-specific). The KAN-25 sort-merge fallback routed over-budget join builds into this path and turned them into 600s wedges. - WEFT_SORT_MERGE_FALLBACK (default false) gates the SMJ fallback at all three sites; over-budget builds now run hash and fail fast with an actionable error naming the knob - Engine::smj_fallback_enabled(); regression tests for default-off hash planning and actionable pool-overflow errors; guard tests serialized behind a tokio Mutex; worker_join_memory_guard opts in - docs/runtime-contract.md + CFN + Helm values recommend preferHashJoin Live SF10: Q93 19.7s golden-clean (was >600s). Q11/Q72 remain — same upstream deadlock class via spilling sort/agg, needs the DataFusion fix. * fix(connect): capture + stamp lakehouse snapshot pins on distributed stages (KAN-48) The Spark Connect distributed path left StageDef.lakehouse_snapshot_pins empty, so workers (which arm require_lakehouse_snapshot_pins) rejected every lakehouse scan: delta/iceberg TPC-H SF10 was 0/22 while parquet was 22/22. try_run_distributed now plans via logical_plan_with_lakehouse_snapshots and stamps the pins JSON on every stage (mirroring the bench plan_distributed path); DataFrame relation-tree path wraps to_plan in the new Engine::capture_lakehouse_snapshots. Regression tests: crates/weft-connect/tests/distributed_lakehouse.rs — SQL + DataFrame distributed group-by over a delta table match single-node (pre-fix: exact production 'omitted the snapshot pin' error), and the unpinned-scan guard stays armed. * chore(deps): datafusion 54.0.0 -> 54.1.0 (KAN-47) Picks up apache/datafusion#23522 (backported as #23654): lost-wakeup race in SpillPoolWriter with concurrent writers — the exact mechanism behind our bounded-FairSpillPool stage deadlock at SF10 (TPC-DS Q11/Q72/Q93, the Q58/Q93 join-config tradeoff, and the KAN-45 SMJ wedges). Also #23574: SMJ correctness fix on the KAN-25 fallback path. Verified before commit: full gates green; upgrade spike ran the wedging Q93 SMJ plan at 4GiB bounded pool — wedged on 54.0.0, completed 4/4 on 54.1.0. datafusion-proto aligned to 54.1.0 for the RecursiveQuery.schema breaking change (#23886). * feat(execution,loom): stage no-progress watchdog + stage observability (KAN-47 follow-up) Defense-in-depth for the DF spill-pool deadlock class: a per-stage monitor races alongside the wall-clock timeout and driver-cancel flag inside run_stage_bounded, sampling three progress signals — per-task batch heartbeat (output stages now stream instead of collect so they heartbeat too), memory-pool activity timestamp (new ProgressMemoryPool wrapper), and DF+shuffle spill bytes. If none advance for WEFT_STAGE_NO_PROGRESS_SECS (default 600) the stage aborts with an actionable KAN-47 error and frees its task slot, instead of burning the full timeout silently. Observability: per-stage summary line (stage/partition, batches, spill bytes, duration, status, last-progress age) matching existing eprintln style; WEFT_STAGE_NO_PROGRESS_SECS + WEFT_STAGE_TIMEOUT_MS documented in docs/runtime-contract.md. Tests: stall-abort at ~2s (was 60s wall timeout), slow-but-progressing not aborted, signal-logic units, summary-line fields; summary lines verified in distributed test logs. * feat(loom,execution): per-query join strategy selection + stall-retry (KAN-53) WEFT_PREFER_HASH_JOIN is now tri-state: auto (new default) chooses per query — hash when the KAN-25 build-budget estimate fits, plan-time SMJ reroute when it doesn't (allowed in auto mode now that DF 54.1.0 fixed the SpillPoolWriter deadlock); true/false keep the legacy session-wide force. Stats fallback chain: row count x width -> parquet total_byte_size -> hash fast path backstopped by the runtime pool-exhaustion retry. Stall-retry: a stage aborted by the KAN-47 no-progress watchdog is cleared of partial spill and retried ONCE with the join strategy flipped (with_join_strategy_flipped task-local, flip accounts for the plan-time reroute so the retry really is the opposite strategy); driver cancels and wall-clock timeouts stay final. Also fixes a latent KAN-25 bug: the SMJ fallback's cloned SessionConfig registered a fresh empty default catalog, wiping engine tables after any fallback (with_create_default_catalog_and_schema(false)). Tests: env parsing/override, auto small-build hash / large-build SMJ, runtime retry on underreported estimate, flip inversion, distributed auto-mode guard variant, flight E2E stall-retry (completes after flip; always-stall fails after exactly two attempts). Gates green; tpch 22/22; tpcds 95/99 + 95/95 with Q58 AND Q93 both execute-verified. * fix(loom): auto join mode reroutes to SMJ when build estimate is unknown (KAN-53 follow-up) Live SF10 evidence: with unknown Glue/S3 stats, auto mode picked hash, the unaccounted hash build (KAN-57) ballooned past the pool, and workers were OOM-killed — TPC-H Q16/Q21 regressed from always-passing. The runtime pool-exhaustion retry can't backstop this because the build is not pool-accounted, so the OOM killer fires first. New fallback order (bounded pool, auto or KAN-45 opt-in): (1) over-budget estimate => plan-time SMJ reroute; (2) UNKNOWN estimate => SMJ reroute (safe since DF 54.1.0 fixed the SMJ deadlock); (3) positive under-budget estimate => hash; (4) runtime exhaustion => one SMJ retry. Unbounded pool keeps hash always (unchanged). Tests: UnknownStatsExec/UnknownStatsTable stand-ins; unknown+bounded => SMJ with correct results (fails before), unknown+unbounded => hash, positive-fit still hash; all prior KAN-25/53 tests unchanged. Gates green; tpch 22/22; tpcds 95/99 + 95/95 (Q58+Q93 verified). * fix(execution): compact StringView buffers in shuffle buckets; loud spilled-read errors Two latent shuffle-spill bugs exposed by TPC-H Q16 at SF10 on the new stack (auto join + DF 54.1.0): 1. hash_partition's arrow take produces StringViewArray views retaining the source scan's entire string buffers (~140x bloat: 33MB for 4,765 rows). estimated_bucket_bytes counted retained buffers, tripping the 4GB shuffle spill threshold; reading bloated buckets at 8-16 concurrent tasks climbed worker RSS to 31GB -> OOM-kill. Fix: compact_views() gc()s Utf8View columns after partitioning so buckets carry only live strings. 2. BucketCache::read_partition served spilled-bucket read errors as legitimately empty buckets (unwrap_or_default): a stage retry adjacent to the spill lifecycle silently truncated c0 values -> wrong supplier_cnt with exact row count. Fix: read_partition returns Result; missing base file is still a legitimate empty bucket, anything else fails the ShuffleRead pull loudly with stage/partition context. Tests: string_view_buckets_do_not_retain_source_buffers (fails on HEAD); spill read-back tests updated. Gates green; tpch 22/22; tpcds 95/99 + 95/95. Live SF10: Q16 3/3 consecutive golden-exact row diffs, zero spill, no OOM, stage tasks ~20ms (was 5-7s). * feat(execution): reorder filtered dims first in inner-join chains (Q72 spill blowup) TPC-DS Q72 at SF10 never finished (>1800s, 36-38GB sort spill per worker): with no table statistics DataFusion keeps written join order, so stage 0 exploded catalog_sales x inventory on item_sk alone into ~1.1B row pairs before the selective single-table filters could apply, then external-sorted that intermediate. New reorder_filtered_dims_first (planner rewrite in plan/join_order.rs, hooked at the top of plan_distributed_logical): within each contiguous inner-join chain under a Filter, leaves carrying single-table filter conjuncts join first (greedy, dependency-checked against ON columns; no-op when order is unchanged). Q72's chain shrinks to ~1e5 rows before inventory is touched: 48.96s, zero spill, golden row-diff clean (was >1800s). Tests: 5 unit tests (fires, idempotent, no-filter no-op, already-optimal no-op, non-inner bail). Gates green; tpch 22/22; tpcds 95/99 + 95/95. Live SF10: Q72 48.96s golden-clean; Q11/Q78/Q93 controls clean. * feat(site,scripts): SF10 distributed results on the site + cluster lifecycle tooling - site/src/data/tpch.json: TPC-H SF10 distributed (2 workers) — 22/22, total 533s, golden-validated vs DuckDB - site/src/data/tpcds.json: replaces pending placeholder — 60/99 execute (total 1314s, avg 21.9s), 39 clean strict-mode refusals, zero timeouts - bench/sf100/results/to-site-sf10.py: generator from suite jsonls to site JSONs (provenance: no hand-entered figures); ENGINE_COLORS entry - PerformancePage: SF10 distributed framing, honest TPC-DS 60/99 wording, reproduce commands - scripts/sf10-{start,stop,attach-spill}.sh: ASG scale-to-0 cost control and 200G spill volume attach (KAN-57) - golden check/diff harness used for all validations * feat(execution): distribute subquery-over-sharded shapes (KAN-55) - resolve_replicated_tables now sizes subquery-scanned tables too, so per-query classification is uniform (subquery-only facts replicate) - replicated-subquery conjuncts route verbatim (partition-independent) - global aggregates above semi/anti filters: recombinable partials + partition-0 gate for COUNT(DISTINCT) (exact empty-input row) - Q9 scalar-projection shape: uncorrelated global-agg scalars over the sharded fact become per-worker partials + one-row combine, AST-rewritten projection evaluated once behind a partition-0 gate - anti-only inline guard: NOT EXISTS over sharded key stream with replicated outer now declines to gather (was per-partition duplicates) 6 of 8 queries now plan distributed at SF10 strict (Q9,Q10,Q16,Q35,Q69, Q94); Q95 (self-join IN) and Q14 (ROLLUP+INTERSECT) keep refusing with locking tests. 15 new tests in auto_distribute_kan55.rs, all fail-before. Gates green; tpch 22/22; tpcds ratchets 95/99 + 95/95. * feat(execution): distribute mixed sharded/replicated UNION ALL (KAN-54) Nested-union flattening in split_union_by_sharding and plan_union (bag union is associative), plus a SUM-only guard for pre-aggregated sharded arms. 6 queries newly plan distributed under the real SF10 strict classification (Q4,Q27,Q33,Q56,Q60,Q76), all execute-verified distributed==single-node. Q5/Q77/Q80 deliberately keep refusing (grouping-set composition has a documented wrong-answer trap). Tests: auto_distribute_kan54.rs (7 tests, fail-before). Remainder mapped: window-over-agg ROLLUP family next (Q36,Q44,Q47,Q51,Q57,Q67,Q70,Q86). * fix(execution,connect): cancel+cleanup on client disconnect; reap uncommitted spill (KAN-46, KAN-31) Dropping the driver query future mid-run (client disconnect) skipped all cleanup: cancel watcher leaked, worker stages kept running, buckets stayed cached (17.8-26.5GB idle RSS observed), and uncommitted producer spill (25-38GB) orphaned into later ENOSPC. QueryAbortGuard on Drop aborts the watcher and runs the cancel+clear sweep; StageSpillReaper reaps a stage's spill scope (base + .segN segments) whenever a producer exits uncommitted. Committed caches still await clear_stages/TTL (positive control). Tests: query_abort_cleanup.rs (4), client_disconnect.rs (2) incl. slow-but-connected client is NOT killed. All fail-before. * test(loom): pin Spark default null ordering (KAN-52) Spark default: ASC => NULLS FIRST, DESC => NULLS LAST (opposite of DuckDB/Postgres). weft already matched via datafusion.sql_parser.default_null_ordering=nulls_min; this dedups the config block and locks the contract with 7 tests (defaults, explicit overrides, ORDER BY...LIMIT boundary, window ORDER BY, distributed finalize replay). Verified RED when the config is disabled. * feat(bench): MATCH/BENIGN/MISMATCH golden verdicts; parquet ListingTable harness (KAN-50, KAN-51) golden_common.py: verdict ladder — checksum match, verified numeric-scale (round-half-even/trunc at decimal scale, reproduces the recorded hash), boundary-tie (LIMIT + tie-zone enumeration, null-ordering sensitivity), numeric-drift heuristic, MISMATCH; --self-test (27 checks) and --strict-benign for CI. TPC-H final: 20 MATCH + 2 BENIGN + 0 MISMATCH; TPC-DS final: 47 MATCH + 13 BENIGN + 0 MISMATCH. tpcds_dist workers register replicated tables as parquet ListingTables (real stats, dynamic-filter pushdown — live-faithful) instead of MemTables; Q72 harness run 51.6s -> 17.8s at SF10. Ratchets 95/99 + 95/95 held; tpch 22/22. * fix(deploy): close bootstrap systemd cycle; atomic env; symmetric DNS; name-agnostic spill (KAN-58) Bootstrap no longer issues role-unit start jobs from inside weft-bootstrap.service (the cycle-closing edge; Requires=/After= kept as the env+DNS-before-start guarantee; UserData starts the role unit after bootstrap completes). write_env is atomic (mktemp+mv). Worker DNS: boot sync always includes own IP (fixes missing-self on cold start); ExecStop removes only own IP (fixes stale zombie IPs -> 'no free task slots'). find_spill_device accepts any unmounted non-root disk, largest wins. TimeoutStartSec=600 for the 240s peer wait. Offline harness 23/23; bash -n clean. NOTE: AMI rebake needed to roll this out. * fix(execution): nested-IN semi cascade accepts replicated dims (KAN-55 follow-up) KAN-55's uniform classification (subquery tables sized too) flips Q20's supplier/partsupp/part from sharded-by-default to replicated at SF10, and try_nested_in_semi's shardedness checks declined — strict mode refused with a whole-lineitem gather. Relaxed: replicated outer/middle/nested scans export once via ExchangeMode::Forward (KAN-36 precedent); the correlated scalar body still requires a sharded fact (a replicated body would multiply the threshold by worker count). Tests: q20_sf10_replicated_dims_plans_semi_cascade (fail-before at the SF10 config: 8 stages, Forward on 2/3/6, no gather) + q20_sf10_replicated_dims_distributed_matches_single_node. Gates green; tpch 22/22; tpcds 95/99 + 95/95; all KAN-55 suites re-verified. * feat(site): TPC-DS 72/99 + TPC-H re-run on the KAN-49 wave-2 stack Site data regenerated from the wave-2 validation runs: TPC-DS 72/99 execute (was 60/99; 27 strict refusals remain), TPC-H 22/22 re-run. to-site-sf10.py now reads the wave-2 jsonls (Q20 dedup ok-preferred). * feat(execution): distribute 8 more TPC-DS queries in strict mode (KAN-49 wave-3a) - Q1/Q30/Q81: decorrelate equality-correlated scalar thresholds into a derived per-key aggregate join (rewrite_correlated_scalar_subqueries) - Q28: exact global COUNT(DISTINCT) via distinct-key shuffle + combine - Q39: expression-alias substitution in branch HAVING/projection remap - Q47/Q57: chained stacked/ranking window layers over one aggregate - Q75: aggregate over Distinct(Union) with mixed sharded/replicated arms via per-arm raw export + hash-shuffle dedup + partial agg Each fix is gated by a strict-plan + distributed==single-node test in tests/auto_distribute_kan49a.rs (9 tests). Strict planner coverage 75 -> 83/99; non-strict ratchet holds 95/99; TPC-H 22/22 0 mismatch. * fix(execution): connect comma-join chains into keyed inner joins (KAN-49 Q6) TPC-DS Q6 OOM'd a SF10 worker with "Resources exhausted: CrossJoinExec with 16.0 GB already allocated". Its stage-0 SQL spliced the unoptimized comma join verbatim — customer_address CROSS JOIN date_dim CROSS JOIN item CROSS JOIN customer CROSS JOIN store_sales with the four equijoins in WHERE. DataFusion has no join reordering, and date_dim / item equijoin only against store_sales, so the emitted order left genuine cross joins under the fact join; CrossJoinExec buffers its entire left input (date_dim filtered to one d_month_seq x customer ⋈ customer_address) outside the KAN-25 hash-join build guard, which only inspects HashJoinExec builds. The correlated avg() scalar was NOT the cause — the worker's ScalarSubqueryToJoin decorrelates it into a hash join. join_order::connect_comma_join_chain rewrites a Filter over a key-less inner-join chain into a connected chain of keyed inner equijoins before stage SQL is emitted: cross-table equalities become ON keys, single-table predicates push onto their leaf's scan (KAN-26 convention — the shuffle-chain extractor walks past Filter nodes and would drop them), cross-table non-equalities become the residual ON filter of the join placing the last referenced leaf. The greedy spanning walk roots at the original leftmost leaf (KAN-26 shuffle shapes stay byte-stable) and attaches the most-filtered reachable leaf next. The rewrite declines — keeping plans byte-for-byte — for keyed/mixed chains, non-simple leaves (branch-DAG cross joins), ambiguous columns, disconnected join graphs (genuine cross products), and subquery-bearing conjuncts the dedicated paths own (cross-correlated, or scanning a non-replicated table; outer references attribute like plain columns since column_refs cannot see them). Q6's stage-0 SQL is now customer_address ⋈ customer ⋈ store_sales ⋈ date_dim ⋈ item, every join carrying its equijoin keys in ON; the worker's physical plan has no CrossJoinExec/NestedLoopJoinExec at all. Regression tests pin the structural property (the SF10 OOM itself is scale-dependent): no CROSS JOIN in any stage SQL, no cross/nested-loop join in the stage-0 physical plan, and distributed == single-node in strict mode with store_sales genuinely sharded. Gates: fmt, clippy -D warnings, cargo test -p weft-execution all green; tpcds-distributed sf0.01 ratchet 95/99 OK; strict sweep still plans Q6 (83/99 strict == unmodified base); tpcds execute 95-verified 0 mismatch; tpch-distributed 22/22 0 mismatch. * feat(execution): distribute ROLLUP-over-channel-UNION TPC-DS queries (KAN-49 wave-4) Resolve the KAN-54-documented wrong-answer trap for Q5/Q77/Q80 and plan them in strict mode; strict planner coverage 83/99 -> 86/99, all three execute-verified distributed==single-node at sf0.01. Root cause of the trap (reproduced, then fixed): - Q77's sharded arm joins a *replicated aggregate* above its own aggregate (ss LEFT JOIN sr). The naive union-split tail splices the whole arm per worker, attaching sr's per-key totals to every worker's partial row; the outer SUM re-adds them once per worker (store channel's returns doubled at sf0.01). Fix: try_split_union_agg_join — per-worker partials of the sharded aggregate co-locate with the replicated aggregate (computed once via Forward), and the join runs after the per-key recombine. - Q5's sharded arm is an aggregate over a *nested* mixed union (store_sales UNION ALL store_returns). Fix: try_split_union_nested distributes the arm with the ordinary machinery and adapts its exact rows into the outer partial schema. - Q80's row-level LEFT JOIN inside the sharded arm is already exact under the naive split; enabling grouping sets in the split covers it. - find_replicated_agg_join gates the unsafe shape: replicated join side producing aggregate rows + sharded side producing aggregate rows. Anything the compositions decline keeps the honest strict refusal. Supporting fixes: - union_of_arms falls back to Union::try_new_with_loose_types when the analyzer widened a decimal in only some arms (Q77's coalesce: Decimal128(17,2) vs (22,2)); the narrowed side is only unparsed for stage SQL, so the worker re-derives coercion from text. Without this the imprecise top-level bucketing dragged a replicated arm into the sharded side on real (decimal) data. - union_split_outer_partial emits bare column names (never Unparser-quoted) — a double-quoted "channel" was read as a string literal by the workers' Databricks dialect. Q14 (ROLLUP + INTERSECT with subqueries over the sharded fact) stays refused: the exact composition (INTERSECT as key shuffles + one-row broadcast of the mixed-sharding avg + per-arm semi joins) is mapped in the test comment; the gather fallback is documented-broken for this family. Refusal-guard test asserts the precise strict refusal. Tests: crates/weft-execution/tests/auto_distribute_kan49d.rs (4 tests, fail-before): Q5/Q77/Q80 strict-plan + assert_distributed_matches_ single_node on two in-process workers; Q14 refusal guard. Gates: fmt + clippy -D warnings clean; cargo test -p weft-execution all green; strict sweep 86/99 (was 83); non-strict ratchet holds (98/99 vs 95 baseline); execute sweep 98 verified / 0 mismatch; tpch-distributed 22/22, 0 mismatch. * feat(execution): distribute 6 whole-fact-gather TPC-DS queries in strict mode (KAN-49 wave-3b) Strict planner coverage 83/99 -> 89/99: Q24, Q38, Q49, Q87, Q95, Q97 no longer fall into the strict-refused whole-fact gather. New module plan/gather_shapes.rs with five co-location shapes: - try_global_count_over_set_op (Q38/Q87): each INTERSECT/EXCEPT DISTINCT branch exports rows hash-shuffled on the full row; the per-partition set op is exact and the global count recombines per-partition counts. - try_full_outer_join_global_agg (Q97): both distinct-key aggregate sides shuffle by the join key (replicated sides via a single Forward producer), so the per-partition FULL OUTER JOIN is exact and the sum(CASE ...) buckets recombine. - try_derived_having_scalar_threshold (Q24): the shared ssales derived aggregate distributes once; the uncorrelated 0.05*avg(netpaid) threshold decomposes off the combine into a KAN-27 one-row scalar broadcast, and the outer aggregate rides the same combine. - try_self_join_in_keys (Q95): the ws_wh self-join keys come from a shuffle-first distinct-key producer (fact hash-shuffled by the order key, self-joined locally); the outer scan exports the same key so the IN filters and count(DISTINCT order) evaluate exactly per partition. - try_ranked_union (Q49): per-channel arms distribute the per-item aggregate, gather the tiny result for the global rank() windows + top-N filter, and UNION-distinct concatenates/dedups. Also: a WEFT_TPCDS_DEBUG-gated plan-debug hook prints the primary shape error the gather fallback masks, and gather_shapes declines trace their reason under the same flag. KAN-55's Q95 refusal test becomes a positive strict-plan shape test; new end-to-end coverage in tests/auto_distribute_kan49c.rs (2-worker in-process cluster, strict mode, distributed == single-node per query). Gates: fmt/clippy/cargo test -p weft-execution clean; strict coverage 89/99; non-strict tpcds-distributed holds 95/99; execute with 2 workers 95 verified, 0 mismatch; tpch-distributed 22/22, 0 mismatch. * feat(execution): distribute 6 window-family TPC-DS queries in strict mode (KAN-49 wave-3b) Q36/Q44/Q51/Q67/Q70/Q86 now plan without the whole-fact gather (strict coverage 83 -> 89/99), each verified distributed == single-node end-to-end at sf0.01 on two in-process workers, gated by a regression test per query (auto_distribute_kan49b). - window over an aggregation now looks through SubqueryAlias / Projection / HAVING-Filter layers between the window and the GROUP BY (renames fold into the remap; pre-window filters apply before any window computes) - ROLLUP / CUBE / GROUPING SETS aggregates compose with windows: the combine reconstructs every rollup level (grouping() recomputed against the real GROUP BY ROLLUP via AggSpec::grouping_target) and the window re-shuffles the tiny combined output by its PARTITION BY key; expression keys (grouping()+grouping(), CASE WHEN ...) are materialized as computed columns on the producer stage - global ranking windows (no PARTITION BY) gather the tiny post-aggregate result to partition 0 and compute there (Q44) - a HAVING carrying an uncorrelated scalar subquery over the sharded fact plans the subquery as a partial/combine pair whose one-row output rides the window stage as a co-located input (Q44) - an uncorrelated IN subquery over the sharded fact in the aggregate input plans as a co-located key stream semi-filtering a scan-export stage (Q70) - framed aggregate windows (ROWS/RANGE BETWEEN ...) re-emit their frame; a FULL OUTER JOIN of two windowed aggregates plans as a co-located shuffle join (Q51) - a window over a UNION (DISTINCT) of aggregate arms plans each arm, hash-shuffles on the full row into a per-partition dedup, then windows over it (Q36) * fix(execution): Q24 shape fingerprint vs comma-join rewrite (KAN-49) The merged connect_comma_join_chain (Q6) rewrites the outer ssales CTE copy's Filter+comma-join into a keyed inner-join chain (ON keys, leaf filters, residual ON filter) but cannot touch the copy inside the HAVING scalar subquery, so try_derived_having_scalar_threshold's derived-table fingerprint stopped matching and Q24 fell back to the strict-refused whole-fact gather. plan_fingerprint now canonicalizes any inner-join region — key-less comma joins and keyed inner joins alike — into a JoinSet of all conjuncts (filter conjuncts, ON key pairs, residual ON filters, leaf filters; equality sides ordered) plus all leaves, each sorted. Both forms reduce identically whether or not the rewrite fires, and the rewrite's own bail-outs keep unmatched forms structural. Gates: kan49c 7/7 (Q24 fails without this fix, passes with it); full cargo test -p weft-execution green; fmt/clippy clean; strict sweep Q24 plans again, 92/99 with only Q14/Q36/Q44/Q51/Q67/Q70/Q86 skipped; non-strict ratchet holds (98/99 >= 95 baseline); execute 98 verified, 0 mismatch; tpch-distributed 22/22, 0 mismatch. * chore(bench): re-baseline TPC-DS ratchets to 98/99 + AggSpec merge fix (KAN-49) Planner 95 -> 98 supported, execute 95 -> 98 verified (0 mismatch). Only Q14 remains unsupported (documented ROLLUP+INTERSECT composition, guard test in auto_distribute_kan49d.rs). Also fixes the AggSpec initializer the window/union branch merge left non-compiling. * fix(connect): cast unsigned columns on the distributed SQL path (KAN-49) DataFusion ranking windows (rank/row_number/dense_rank) produce UInt64, which Spark has no type for: PySpark aborts the Arrow stream with UNSUPPORTED_DATA_TYPE_FOR_ARROW_CONVERSION. The DataFrame distributed path already normalized unsigned->signed via signed_columns; the SQL distributed path (the one PySpark spark.sql uses) did not. Found at SF10 on Q36/44/49/67/70/86, which executed correctly but could not be collected. Regression test: distributed_sql_rank_window_result_is_signed_for_pyspark drives a rank() window through in-process workers over Spark Connect, asserts the query went distributed (no local fallback) and that every client-visible column is signed (rank -> Int64). Fails before, passes after. * fix(scripts): sf10-attach-spill bash 3.2 + racy device pick - ${EXTRA[@]+...} form: macOS bash 3.2 under set -u aborts on an empty array expansion, which killed the script mid-run leaving the driver unmounted and the second worker unprocessed. - Pick the unmounted disk matching SIZE_GIB exactly instead of 'largest unmounted': NVMe enumeration is racy after stop/start and the old pick mounted a stale 100 GiB leftover at /var/lib/weft/spill on one worker. Verified against the live cluster (mounted 200G skipped, stale 100G matched only at 100 GiB). * feat(execution): distribute TPC-DS Q23/Q41 at the SF10 classification (KAN-49 wave-3f) Q23 and Q41 plan under the local per-fact sweep but refused at real SF10, where the query's largest table is the sharded one and everything else replicates. Two minimal compositions close the gap: - Q23 (store_sales sharded, channel facts replicated): a UNION ALL of per-channel arms whose only sharded inputs are three shared derived CTEs. New try_union_over_derived_ctes shape (gather_shapes.rs) plans each distinct CTE once (dag_splitter fingerprint dedup), gathers each into the arm stages (replace_branches placeholders), and concatenates the arms. best_ss_customer's single-row max_store_sales cross-join leaf becomes try_cross_scalar_threshold: the scalar's per-customer input distributes, a KAN-27 one-row broadcast computes the max, and the combine's HAVING compares against the injected SCALAR_TOKEN literal. - Q41 (item sharded — the local sweep never tries item): extend try_decorrelate_scalar_subquery with (a) a SELECT DISTINCT wrapper (re-applied after a full-row hash shuffle of the join output), (b) OR-factored correlation extraction for its (c AND A) OR (c AND B) disjunctive residual, and (c) mirrored compare operators for subquery-on-the-left compares ((SELECT count(*) ...) > 0), which the existing shape rendered with inverted operands. Tests (auto_distribute_kan49f.rs, fail-before verified): both queries plan under WEFT_DISTRIBUTED_STRICT=1 with the exact SF10 classification and match single-node end-to-end on two workers with cross-shard groups (the manufact=300 pair and every Q23 group deliberately straddle shards). Gates: fmt/clippy/test clean; strict sweep 98/99 held; non-strict ratchet 98/99 held; tpcds execute --workers 2: 98 verified, 0 mismatch; tpch-distributed: 22/22, 0 mismatch. * feat(site): SF10 final — TPC-DS 98/99, TPC-H 22/22 golden-validated (KAN-49) Wave-3 stack results: TPC-DS 98/99 execute (2265s hot, avg 23.1s), 79 MATCH + 19 BENIGN-verified, 0 mismatch, only Q14's documented strict refusal remaining. TPC-H 22/22 (593s hot, avg 26.9s), 20 MATCH + 2 BENIGN-verified, 0 mismatch. * fix(ci): line-tables-only debuginfo for the workspace test step main's clippy+test job died at link time (collect2: ld terminated with signal 7 [Bus error]) — parallel links of ~1-2 GB weft-execution test binaries exhaust the 16 GB hosted runner. debug=1 (line tables) keeps panic locations and backtraces while shrinking link inputs; assertions unchanged. * feat(execution): distribute TPC-DS Q14 in strict mode — 99/99 (KAN-49 wave-4) Q14 was the last strict-mode refusal of the 99: a ROLLUP over a per-channel UNION ALL whose arms read the sharded fact through two derived-CTE subqueries — the cross_items three-way INTERSECT (IN key set) and the avg_sales global AVG (HAVING threshold). The gather fallback is documented-broken for ROLLUP+UNION/INTERSECT, so strict mode refused. try_rollup_union_derived_subqueries (gather_shapes.rs) plans each derived table once (the per-arm inlined copies are fingerprint-checked identical) and rewrites the arms to read those stage outputs: - cross_items: the INTERSECT chain's raw arms are leaf producers hash-shuffled on the full (brand,class,category) triple (sharded channel per worker, replicated channels once via Forward), so the per-partition INTERSECT is exact; the item join-back is a per-partition broadcast join emitting the ss_item_sk key stream hash-shuffled on the key. - avg_sales: per-worker sum/count partials over the sharded arm plus one Forward partial over the replicated arms combine into the one-row global AVG gathered to partition 0, consumed as a co-located stream (the Q44 pattern) rather than a SCALAR_TOKEN literal. - each channel arm: scan export hashed by xx_item_sk, a per-partition semi join against the key stream feeding the partial aggregate (the Q70 composition; co-location keeps IN three-valued logic exact), and a gathered recombine applying HAVING against the co-located scalar. - the outer ROLLUP gathers the three tiny exact arm streams and rebuilds every grouping-set level; COUNT(*)>0 OR EXISTS(<scalar stream>) suppresses synthetic grand-total rows on empty partitions while keeping the genuine one on partition 0 over an empty input. The shape declines anything outside the exact family, so every other query keeps its current plan byte-for-byte. Tests: q14_rollup_intersect_plans_and_matches (SF10 classification, store_sales sharded) and _web_sharded (harness-local variant) replace the refusal guard; the fixture now puts item 3 into cross_items with a below-threshold November web sale, so the HAVING cut, the INTERSECT dedup, and the cross-shard key match are all exercised end-to-end at sf0.01 with two in-process workers. Gates: fmt/clippy/test -p weft-execution clean; strict sweep 99/99; non-strict ratchet 98/99 baseline held (+1 gain); execute 99 verified, 0 mismatch, 0 error; TPC-H 22/22, 0 mismatch. * test(execution): deflake auto_distribute worker-port allocation Two races shared a root cause: unique_worker_port seeded at 41000+, inside the Linux ephemeral source range (32768..=60999), so the harness's own outbound connections could steal a worker's port — serve_worker swallows the EADDRINUSE and await_worker_bind panicked ('w1 did not bind on port 41600' on CI). ephemeral_port()'s grab-release-rebind had the same exposure. - single OnceLock-seeded allocator, base 10000 + pid%15000, below the ephemeral floor and clear of other suites' fixed ports - all ephemeral_port() call sites switched to it; helper deleted - await_worker_bind ceiling 2s -> 10s for loaded runners Subsumes the open daily-maintenance drafts #61/#65. * fix(ci): raise clippy+test timeout to 180m The debug=1 switch invalidated the cached dev/test artifacts, so the first run with it recompiles the workspace from scratch and the old 90 min ceiling killed the job at the 90-min mark (run 30605262149, 'all but clippy+test green'). One completed run re-warms the cache. * chore(bench): re-baseline TPC-DS ratchets to 99/99 (KAN-49 Q14) * fix(bench): widen execute-sweep transport retry to 60x500ms The CI gates job kept flaking on 'connect worker: transport error' (Q30/Q80/Q81/Q91 across runs): the 4-vCPU runner can leave a worker unreachable for several seconds under load, and the 30x100ms=3s window gave up mid-hiccup, failing the whole execute ratchet. 30s ceiling + 'transport' marker; plan/SQL errors still fail immediately. * feat(site): TPC-DS 99/99 + TPC-H 22/22 golden-validated at SF10 (KAN-49) Q14 closed the matrix: TPC-DS 99/99 execute under strict mode (2438s hot, avg 24.6s), golden 80 MATCH + 19 BENIGN-verified, 0 mismatch. TPC-H 22/22 (623s hot, avg 28.3s), 20 MATCH + 2 BENIGN-verified, 0 mismatch. * test(execution): deflake worker-port allocation across all auto_distribute* suites Same fix as auto_distribute.rs, applied to the 14 sibling files whose allocators still seeded inside the Linux ephemeral source range (43000/46000/47000 + pid%) with the fetch_add-then-store init race: - OnceLock-seeded counter (no pre-seed values handed out) - per-file bases 12000..25000 (+pid%512), fully below 32768, distinct windows so concurrent test binaries cannot collide CI evidence: 'w1 did not bind on port 41600' (auto_distribute) and 'distributed run never succeeded' (auto_distribute_kan33) — both the swallowed-EADDRINUSE flake this removes. --------- Co-authored-by: Vamsi Krishna Kothapalli <vamsi@Vamsis-Mac-mini.local> Co-authored-by: Vamsi Krishna Kothapalli <vamsi@Vamsis-Mac-mini.attlocal.net> Co-authored-by: Vamsi Krishna Kothapalli <vamsi@vamsis-mac-mini.tail08ec2b.ts.net>

Which issue does this PR close?
Rationale for this change
When multiple
SpillPoolWriterclones concurrently push batches to the same channel, more than one non-finishedSpillFilecan be in flight. This happens because eachSpillPoolWriterclone takes thecurrent_write_fileat the start ofpush_batchand puts it back when it's done. When multiplepush_batchcalls happen concurrently, only the first one will be able to take thecurrent_write_fileand the others will all create their own new spill file. Which one gets put back for subsequent use is a race condition.If this occurred and the writers are all dropped before rotation happens in, multiple files in the
filesdeque will be havewriter_finished == false. The last writer drop logic inSpillPoolWriter::droponly finishes whatever file is thecurrent_write_fileas finished.This can lead to a stalled situation when
SpillPoolFile::poll_nextcatches up with the writer and returnsPendingbecausewriter_finished == false. A waker for the file is registered, but since the last writer drop logic only finishes and wakes whatever happens to becurrent_write_file, which may not be the current read file, the waker may end up never being notified.There is a secondary waker that is registered on the spill pool itself, but due to fine grained locking, it is possible for the wake call in the last writer drop logic to be called before the waker registration.
What changes are included in this PR?
writer_droppedfield which was an unnecessary denormalisation ofactive_writer_count == 0An additional benefit of tracking all unfinished write files is that excessive creation of tiny spill files is avoided when many writers are pushing batches concurrently.
Are these changes tested?
Reproduction case from linked issue was used to confirm fix
Are there any user-facing changes?
No