Skip to content

[SPARK-58313][SDP] Validate SCD2 track-history columns at AutoCDC flow construction#57524

Open
anew wants to merge 3 commits into
apache:masterfrom
anew:spark-58313-track-history-v2
Open

[SPARK-58313][SDP] Validate SCD2 track-history columns at AutoCDC flow construction#57524
anew wants to merge 3 commits into
apache:masterfrom
anew:spark-58313-track-history-v2

Conversation

@anew

@anew anew commented Jul 25, 2026

Copy link
Copy Markdown
Contributor

What changes were proposed in this pull request?

This PR replaces #57490, which had to be abandoned due to severe merge conflicts. It is an exact cherry-pick of the previous commits, plus a fix for the test cases that failed due to the merge conflicts.

An SCD2 AutoCDC flow can restrict which columns define a "run" via TRACK HISTORY ON (...), which populate ChangeArgs.trackHistorySelection. Until now, that selection was only resolved when the first microbatch ran, inside Scd2BatchProcessor.computeTrackedHistoryColumns during reconciliation. An unresolvable or ineligible tracking column — one that is absent from the source, is a key, is a reserved framework column, or was dropped by the flow's column_list — therefore surfaced mid-stream rather than at flow construction, unlike every other AutoCDC misconfiguration (keys, column selection, reserved names), which fail eagerly.

This PR validates trackHistorySelection at AutoCdcMergeFlow construction time, mirroring the existing requireKeysPresentInSelectedSchema check:

  • The eligibility + resolution logic is extracted from Scd2BatchProcessor.computeTrackedHistoryColumns into a schema-based companion helper Scd2BatchProcessor.computeTrackedHistoryColumns(schema, changeArgs, caseSensitive). Both the per-microbatch runtime path and the new construction-time validator call it, so the two can never diverge. The refactor is behavior-preserving.
  • AutoCdcMergeFlow gains requireTrackHistoryColumnsResolvableInSelectedSchema, invoked when deriving the user-selected schema (right after the key-presence check). It runs before the flow's schema is forced, so the actionable error surfaces ahead of the temporary AUTOCDC_SCD2_NOT_SUPPORTED gate and remains correct once SCD2 support lands.
  • No new error condition: an unresolvable selection reuses the existing AUTOCDC_COLUMNS_NOT_FOUND_IN_SCHEMA (schema name trackHistorySelection).
  • The check is a no-op when trackHistorySelection is None, which covers all SCD1 flows (enforced by ChangeArgs) and SCD2 flows that do not restrict tracking.

Why are the changes needed?

Deferring this validation to reconciliation means a simple typo or misconfiguration (TRACK HISTORY ON (typo), or tracking a key/excluded column) is not reported at graph analysis time; it only fails once data flows, with an error raised deep in the SCD2 batch processor. Validating at flow construction gives a fail-fast, user-actionable error consistent with the rest of the AutoCDC configuration surface (keys, column_list, reserved names).

Does this PR introduce any user-facing change?

Yes. An AutoCDC SCD2 flow whose TRACK HISTORY ON (...) references a column that is not an eligible history-tracking column (absent, a key, a framework column, or excluded by column_list) now fails at flow construction with AUTOCDC_COLUMNS_NOT_FOUND_IN_SCHEMA instead of failing when the first microbatch runs. Valid selections are unaffected, and there is no change for SCD1 flows. (Note: SCD2 AutoCDC flows are not yet generally supported on master — still gated by AUTOCDC_SCD2_NOT_SUPPORTED — so no released behavior changes.)

How was this patch tested?

New unit tests in AutoCdcFlowSuite covering: an SCD2 flow tracking a non-existent column, a key column (ineligible), and a column dropped by columnSelection are each rejected at construction with AUTOCDC_COLUMNS_NOT_FOUND_IN_SCHEMA; a resolvable selection passes the check (falling through to the SCD2-not-supported gate); and case-sensitive/insensitive resolution behavior.

Was this patch authored or co-authored using generative AI tooling?

Generated-by: Claude Code (Opus 4.8)

anew added 3 commits July 25, 2026 04:14
…w construction

An SCD2 AutoCDC flow's history-tracking columns (`TRACK HISTORY ON ...`,
i.e. ChangeArgs.trackHistorySelection) were only validated when the first
microbatch ran reconciliation. An unresolvable or ineligible tracking column
(one that is absent, a key, a framework column, or dropped by the column
selection) therefore surfaced mid-stream, deep inside Scd2BatchProcessor,
rather than eagerly at flow construction.

Validate the selection at AutoCdcMergeFlow construction time, mirroring the
existing key-presence check. The eligibility + resolution logic is extracted
into a schema-based Scd2BatchProcessor.computeTrackedHistoryColumns helper that
both the runtime path and the new construction-time validator call, so the two
can never diverge. An unresolvable selection fails with the existing
AUTOCDC_COLUMNS_NOT_FOUND_IN_SCHEMA error; the check is a no-op when
trackHistorySelection is None (all SCD1 flows and unrestricted SCD2 flows).

Adds tests in AutoCdcFlowSuite.

Co-authored-by: Isaac
… and test names

Remove the SPARK-58313 references from the source comments and test names/section
header, consistent with the other AutoCDC PRs (the merged PR records provenance).

Threading conf.resolver through the shared ColumnSelection.applyToSchema (rather than
deriving a resolver from a caseSensitive boolean) is left as a follow-up refactor,
SPARK-58347, since it reworks a boolean/getFieldIndex-based API shared across the SCD1
and SCD2 code paths and is out of scope here.

Co-authored-by: Opus 4.8
…ema support

SPARK-58319 (now on master) removed the AUTOCDC_SCD2_NOT_SUPPORTED gate, so an SCD2
flow with a resolvable track-history selection now constructs successfully instead of
throwing. Update the two tests that expected that gate to assert the flow builds and
its SCD2 schema resolves, and refresh the stale section comment. The case-sensitive
rejection test is unchanged: it still expects AUTOCDC_COLUMNS_NOT_FOUND_IN_SCHEMA from
the track-history check itself.

Co-authored-by: Opus 4.8
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.

1 participant