Casting to a struct type with repeated field names fails with "ArrowArray struct has 2 children (expected 1)"
I maintainer di solito rispondono entro 1 giorno
Valutazione
- Difficoltà
- 2/5
- Tempo stimato
- 1-3 ore
- Idoneità per principianti
- 68/100
Direzione di ricerca
Leggi il ramo struct di CometCast.isSupported e confrontalo con CometCreateNamedStruct, che già rifiuta i nomi di campo ripetuti; i rami array e map ricorrono al suo interno, quindi un controllo copre i casi annidati nel SQL di riproduzione. Riproduci con le tre query CAST che falliscono più i due controlli che passano, poi conferma che la correzione fa eseguire il fallback al cast in modo che i risultati corrispondano a Spark. Fatto sembra un nuovo caso che copre i nomi di campo struct ripetuti nella suite di test del cast e nessuna IllegalStateException da NativeUtil.importVector.
Scritto dal modello di indicizzazione a partire dal testo della issue.
Descrizione
Describe the bug
A cast whose target struct type repeats a field name fails the task when Comet runs it natively. Spark allows the cast as long as each pair of fields casts, so CAST(s AS STRUCT<x: INT, x: INT>) over a STRUCT<p: INT, q: INT> column returns {1, 10}. Comet's native cast_struct_to_struct builds the result with the target fields, and the projection itself works: to_json over the cast matches Spark with {"x":1,"x":10}. Importing the result on the JVM then fails, because Java Arrow keys struct children by name and the two x children collapse into one:
java.lang.IllegalStateException: ArrowArray struct has 2 children (expected 1)
at org.apache.arrow.util.Preconditions.checkState(Preconditions.java:562)
at org.apache.arrow.c.ArrayImporter.doImport(ArrayImporter.java:92)
at org.apache.arrow.c.ArrayImporter.importChild(ArrayImporter.java:83)
at org.apache.arrow.c.ArrayImporter.doImport(ArrayImporter.java:101)
at org.apache.arrow.c.ArrayImporter.importArray(ArrayImporter.java:68)
at org.apache.arrow.c.ArrowImporter.importVector(ArrowImporter.java:62)
at org.apache.comet.vector.NativeUtil.importVector(NativeUtil.scala:264)
at org.apache.comet.vector.NativeUtil.getNextBatch(NativeUtil.scala:211)
at org.apache.comet.CometExecIterator.getNextBatch(CometExecIterator.scala:236)
The same happens when the struct is nested in an array or a map value.
Steps to reproduce
Default configs, reproduced on main at fef94f6cd with Spark 4.1.3:
CREATE TABLE ts (
s STRUCT<p: INT, q: INT>,
arr ARRAY<STRUCT<p: INT, q: INT>>,
m MAP<STRING, STRUCT<p: INT, q: INT>>) USING parquet;
INSERT INTO ts VALUES
(named_struct('p', 1, 'q', 10), array(named_struct('p', 1, 'q', 10)),
map('k', named_struct('p', 1, 'q', 10))),
(named_struct('p', 2, 'q', NULL), array(), map()),
(NULL, NULL, NULL);
SELECT CAST(s AS STRUCT<x: INT, x: INT>) FROM ts; -- IllegalStateException
SELECT CAST(arr AS ARRAY<STRUCT<x: INT, x: INT>>) FROM ts; -- IllegalStateException
SELECT CAST(m AS MAP<STRING, STRUCT<x: INT, x: INT>>) FROM ts; -- IllegalStateException
SELECT CAST(s AS STRUCT<x: INT, y: INT>) FROM ts; -- matches Spark
SELECT to_json(CAST(s AS STRUCT<x: INT, x: INT>)) FROM ts; -- matches Spark
The plan is CometProject over CometNativeScan. Spark returns {1, 10}, {2, null} and null for the first query.
Expected behavior
Either match Spark or fall back. The smallest fix is probably for the struct branch of CometCast.isSupported to return Unsupported when the target struct repeats a field name, the same way CometCreateNamedStruct declines repeated names. The array and map branches already recurse into the struct branch, so one check there also covers the nested cases.
Additional context
This is the same Java Arrow limitation as #6251 (arrays_zip, fix in #6324), #5605 and #1015. Main declines structs with repeated field names wherever they would cross into Java Arrow: the shuffle and row conversion checks (#5866), the codegen dispatcher (#5766) and the cache serializer (#6004). Those checks guard the paths that bring a struct into Comet, though, and a native cast creates one. The broadcast exchange admits these types too (CometSink.convert calls supportedDataType with the default allowDuplicateStructFieldNames = true), so declining in the serde that produces the struct is what keeps it out.
Found while checking whether anything in #5603 was worth keeping.
- Lingua principale
- Scala
- Stelle
- 1.3k
- Fork
- 383
- Merge medio
- 2g 58m
- PR unite (30g)
- 399
Preparare l'ambiente
- Nessun Dockerfile né file Docker Compose
- Ha un modello di pull request
- Leggi la guida per i contributori
Come iniziare
- Leggi tutta la issue e poi la guida ai contributi del progetto.
- Commenta sulla issue per dire che te ne occupi tu — evita che due persone facciano lo stesso lavoro.
- Fai un fork del repository e lavora su un branch.
- Apri una pull request che faccia riferimento al numero della issue.
Altre issue di apache/datafusion-comet
-
area:aggregation bug priority:medium
Difficoltà 2/5 1-3 ore Idoneità per principianti 82/100
apache/datafusion-comet#6661 ·
I maintainer di solito rispondono entro 1 giorno
-
Native from_json with a schema that repeats a field name fails with "ArrowArray struct has 2 children (expected 1)"Forse già presa Una pull request collegata a questa issue è aperta o già unita. Apertaarea:expressions area:ffi bug good first issue priority:medium
Difficoltà 2/5 1-3 ore Idoneità per principianti 70/100
apache/datafusion-comet#6592 ·
I maintainer di solito rispondono entro 1 giorno
-
arrays_zip with two same-named inputs fails with "ArrowArray struct has 2 children (expected 1)"Forse già presa @mohitgurav20 l’ha presa 7 giorni fa. Apertaarea:expressions area:ffi bug priority:high
Difficoltà 2/5 1-3 ore Idoneità per principianti 78/100
apache/datafusion-comet#6251 · 4 commenti · 1 assegnatario ·
I maintainer di solito rispondono entro 1 giorno
-
ci: a transient Maven Central failure in lint-java is reported as a scalafix violationForse già presa @andygrove l’ha presa 15 giorni fa. Apertaarea:ci bug priority:low
Difficoltà 2/5 1-3 ore Idoneità per principianti 84/100
apache/datafusion-comet#6060 ·
I maintainer di solito rispondono entro 1 giorno
-
requires-triage
Difficoltà 2/5 1-3 ore Idoneità per principianti 68/100
apache/datafusion-comet#5661 ·
I maintainer di solito rispondono entro 1 giorno
Tutte le issue di apache/datafusion-comet
Issue simili
-
Difficoltà 2/5 1-3 ore Idoneità per principianti 78/100
snowflakedb/spark-snowflake#673 ·
-
bug
Difficoltà 2/5 1-3 ore Idoneità per principianti 86/100
salesforce/evalon#16 ·
-
Difficoltà 2/5 1-3 ore Idoneità per principianti 72/100
chipsalliance/chisel#5504 ·
I maintainer di solito rispondono entro 1 giorno
-
documentation good first issue
Difficoltà 2/5 Mezza giornata Idoneità per principianti 75/100
I maintainer di solito rispondono entro 1 giorno
-
Homebrew formula 2.1.26: 'cs completions bash' fails (exit 127) because bin/cs is not executableAperta
Difficoltà 2/5 1-3 ore Idoneità per principianti 85/100
coursier/coursier#3896 · 1 reazione ·
I maintainer di solito rispondono entro 1 giorno