gengliangwang opened a new pull request, #55973:
URL: https://github.com/apache/spark/pull/55973

   ### What changes were proposed in this pull request?
   
   Two related changes:
   
   1. Fold `ElementAtUtils.resolveArrayIndex` into the existing 
`ArrayExpressionUtils.java`, and remove `ElementAtUtils.java`. The 
per-expression naming chosen in SPARK-56916 didn't match the codebase's 
category-scoped utility-class convention (`ArrayExpressionUtils`, 
`BitmapExpressionUtils`, `ExpressionImplUtils`, ...) and there's now a natural 
home for any future array-expression ANSI helper.
   
   2. Refactor `GetArrayItem`'s ANSI codegen + eval paths to use a new 
`ArrayExpressionUtils.checkArrayIndex(int length, int index, QueryContext 
context)` helper, mirroring how `ElementAt` uses `resolveArrayIndex`. The 
helper throws `invalidArrayIndexError` for negative / out-of-bound ANSI indices 
and returns the validated 0-based position so the caller chains into 
`arr.get(idx, dataType)`. The non-ANSI branch keeps its inline form because it 
must return `null` (not throw) on out-of-bound.
   
   Net effect: the existing per-expression `ElementAtUtils.java` is removed; 
the existing `ArrayExpressionUtils.java` grows two `*ArrayIndex` helpers used 
by `ElementAt` and `GetArrayItem` codegen + eval.
   
   ### Why are the changes needed?
   
   Part of SPARK-56908 (umbrella). `arr[idx]` and `element_at(arr, idx)` share 
the same ANSI out-of-bound error shape; collapsing both into one-line helper 
calls keeps the codegen size small and avoids maintaining two parallel inline 
forms.
   
   ### Does this PR introduce _any_ user-facing change?
   
   No. The compiled behavior is identical; only the emitted Java source text 
changes.
   
   ### How was this patch tested?
   
   ```
   build/sbt "catalyst/testOnly *ComplexTypeSuite *CollectionExpressionsSuite"
   build/sbt "sql/testOnly *QueryExecutionAnsiErrorsSuite"
   ```
   
   All pass (83/83 catalyst, 21/21 sql).
   
   ### Was this patch authored or co-authored using generative AI tooling?
   
   Generated-by: Cursor 1.x
   


-- 
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

To unsubscribe, e-mail: [email protected]

For queries about this service, please contact Infrastructure at:
[email protected]


---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]

Reply via email to