Skip to content

[UUID 6/8] UUID multi-stage engine (planner + runtime) - #18874

Open
xiangfu0 wants to merge 4 commits into
apache:masterfrom
xiangfu0:uuid-split/06-mse-planner-runtime
Open

[UUID 6/8] UUID multi-stage engine (planner + runtime)#18874
xiangfu0 wants to merge 4 commits into
apache:masterfrom
xiangfu0:uuid-split/06-mse-planner-runtime

Conversation

@xiangfu0

@xiangfu0 xiangfu0 commented Jun 29, 2026

Copy link
Copy Markdown
Contributor

Parent tracking issue: #16619

What

Multi-stage (v2) engine support for UUID.

Changes

  • expressions.proto UUID / UUID_ARRAY enum values + planner type resolution, Rex parsing, proto serde, TypeFactory
  • HashJoinOperator / EnrichedHashJoinOperator + UuidLookupTable (LookupTable.normalizeKey), OneUuidKeyGroupIdGenerator, ServerPlanRequestUtils

Enables

UUID equi-joins and group-by in the multi-stage engine.

⚠️ Rolling-upgrade note: the new proto enum values are unknown to older brokers/servers; multi-stage queries with UUID literals fail on mixed-version clusters. Atomic-upgrade feature.

Depends on

#18869 and #18871.

About this PR / how to review

This is part 6 of 8 splitting #18140 (first-class logical UUID type) into layered PRs, as requested there.

The split is enabled by the v1 design: DataType.UUID has stored type BYTES, so most paths handle it automatically; each PR adds explicit UUID semantics to one subsystem. Head branch lives on xiangfu0/pinot.

This PR is stacked on #18873 (branch uuid-split/05-agg-groupby-distinct). Because GitHub PRs to apache must base on master, the Files-changed tab is cumulative (it includes layers 1–6) until the PRs below it merge. Review the commit titled [UUID 6/8] UUID multi-stage engine (planner + runtime) — that is this layer's change. Each parent merge shrinks this diff after a rebase.

Full stack (merge bottom → top)

  1. [UUID 1/8] Add logical UUID type foundation (pinot-spi) #18869 — [UUID 1/8] logical UUID type foundation (pinot-spi)
  2. [UUID 2/8] UUID ingest and segment storage #18870 — [UUID 2/8] UUID ingest and segment storage
  3. [UUID 3/8] UUID result rendering (DataSchema, Arrow/JSON encoders) #18871 — [UUID 3/8] UUID result rendering (DataSchema, Arrow/JSON encoders)
  4. [UUID 4/8] Server-side predicate evaluation for the logical UUID type #18872 — [UUID 4/8] UUID server-side predicates, CAST and transforms
  5. [UUID 5/8] UUID aggregation, group-by and distinct #18873 — [UUID 5/8] UUID aggregation, group-by and distinct
  6. [UUID 6/8] UUID multi-stage engine (planner + runtime) #18874 — [UUID 6/8] UUID multi-stage engine (planner + runtime)
  7. [UUID 7/8] UUID partitioning #18875 — [UUID 7/8] UUID scalar UDFs and partitioning
  8. [UUID 8/8] UUID integration tests, benchmarks and docs #18876 — [UUID 8/8] UUID integration tests, benchmarks and docs

After #18869 lands, PRs 2/4/7 (#18870, #18872, #18875) only depend on it and can be reviewed in parallel; PRs 5/6 (#18873, #18874) also need #18871; PR 8 (#18876) needs #18874.

Full feature description, v1 design contract, scope exclusions, and benchmark numbers: #18140.

@codecov-commenter

codecov-commenter commented Jun 30, 2026

Copy link
Copy Markdown

Codecov Report

❌ Patch coverage is 38.11881% with 250 lines in your changes missing coverage. Please review.
✅ Project coverage is 66.61%. Comparing base (e51b4e4) to head (f5ffcae).

Files with missing lines Patch % Lines
...pby/NoDictionarySingleColumnGroupKeyGenerator.java 42.00% 23 Missing and 6 partials ⚠️
.../function/DistinctCountULLAggregationFunction.java 0.00% 26 Missing ⚠️
...aggregation/function/AggregationFunctionUtils.java 20.00% 22 Missing and 2 partials ⚠️
...ion/DistinctCountCPCSketchAggregationFunction.java 0.00% 24 Missing ⚠️
...upby/NoDictionaryMultiColumnGroupKeyGenerator.java 26.66% 19 Missing and 3 partials ⚠️
...e/operator/groupby/OneUuidKeyGroupIdGenerator.java 0.00% 20 Missing ⚠️
...nction/DistinctCountBitmapAggregationFunction.java 24.00% 16 Missing and 3 partials ⚠️
...n/DistinctCountThetaSketchAggregationFunction.java 20.83% 18 Missing and 1 partial ⚠️
...ction/DistinctCountHLLPlusAggregationFunction.java 25.00% 15 Missing and 3 partials ⚠️
.../function/DistinctCountHLLAggregationFunction.java 50.00% 10 Missing and 2 partials ⚠️
... and 8 more
Additional details and impacted files
@@             Coverage Diff              @@
##             master   #18874      +/-   ##
============================================
- Coverage     66.65%   66.61%   -0.05%     
  Complexity     1423     1423              
============================================
  Files          3443     3446       +3     
  Lines        218632   218968     +336     
  Branches      34793    34885      +92     
============================================
+ Hits         145726   145857     +131     
- Misses        61192    61376     +184     
- Partials      11714    11735      +21     
Flag Coverage Δ
custom-integration1 ?
integration 100.00% <ø> (ø)
integration1 100.00% <ø> (ø)
integration2 ?
java-25 66.61% <38.11%> (-0.05%) ⬇️
temurin 66.61% <38.11%> (-0.05%) ⬇️
unittests 66.60% <38.11%> (-0.05%) ⬇️
unittests1 57.28% <38.11%> (+0.05%) ⬆️
unittests2 38.84% <0.74%> (-0.06%) ⬇️

Flags with carried forward coverage won't be shown. Click here to find out more.

☔ View full report in Codecov by Harness.
📢 Have feedback on the report? Share it here.

🚀 New features to boost your workflow:
  • 📦 JS Bundle Analysis: Save yourself from yourself by tracking and limiting bundle sizes in JS merges.

@xiangfu0
xiangfu0 force-pushed the uuid-split/06-mse-planner-runtime branch 7 times, most recently from 763c96a to 96284ae Compare July 7, 2026 07:08
@xiangfu0
xiangfu0 force-pushed the uuid-split/06-mse-planner-runtime branch 11 times, most recently from 3e1c95c to f41f8ea Compare July 14, 2026 08:03
@xiangfu0
xiangfu0 force-pushed the uuid-split/06-mse-planner-runtime branch 10 times, most recently from 1cf12f0 to bd7f7af Compare July 21, 2026 08:10
@xiangfu0
xiangfu0 force-pushed the uuid-split/06-mse-planner-runtime branch from bd7f7af to c6669fe Compare July 22, 2026 08:11
@xiangfu0
xiangfu0 force-pushed the uuid-split/06-mse-planner-runtime branch 6 times, most recently from 749583d to 7c23307 Compare August 7, 2026 09:28
xiangfu0 and others added 4 commits August 7, 2026 22:00
Adds UUID handling to the predicate evaluators, so =, !=, IN, NOT IN and range
predicates work against a UUID column on both the raw and the dictionary path.

UUID follows the pattern TIMESTAMP already uses: a logical type whose stored
type does the work. The literal is parsed to its 16-byte stored form once, when
the evaluator is built, and from there the existing BYTES evaluators apply --
no per-value conversion in the scan loop.

The dictionary path needs no UUID branch: Dictionary#getStoredValue returns hex
for a UUID column and indexOf(String) hex-decodes, so the existing String-keyed
lookup is already correct. PredicateUtils renders the literal to that hex form
for those String-typed lookup APIs.

Split into apache#19181 (CAST), apache#19182 (bloom filter pruning) and apache#19183 (transform
functions); this PR is now just the predicate evaluators.
The UUID branch accepted several input forms via UuidUtils.toBytes(Object),
which was inconsistent with every other branch -- they each cast to one type.

The input is in fact deterministic. GroupByDataTableReducer runs every column
through ColumnDataType#convert immediately before calling isMatch, and convert
returns UuidUtils.toUUID(...) for UUID, so the value is always a
java.util.UUID. The other reduce path cannot deliver a UUID here at all:
getConvertedKey has no UUID case and throws for it.

Adds testHavingFilterOnUuidColumn, which pins the contract -- this line was
previously untested in either direction. Verified it reaches the UUID branch
rather than passing vacuously.
Part 5/8 of splitting apache#18140 (logical UUID type). Rebased onto latest master; stacked on uuid-split/04-sse-predicates-cast.

Downstream references use the UuidKey class merged in apache#18869.
Part 6/8 of splitting apache#18140 (logical UUID type). Rebased onto latest master; stacked on uuid-split/05-agg-groupby-distinct.

Downstream references use the UuidKey class merged in apache#18869.
@xiangfu0
xiangfu0 force-pushed the uuid-split/06-mse-planner-runtime branch from 7c23307 to f5ffcae Compare August 8, 2026 09:36
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants