[
https://issues.apache.org/jira/browse/BEAM-14026?focusedWorklogId=736246&page=com.atlassian.jira.plugin.system.issuetabpanels:worklog-tabpanel#worklog-736246
]
ASF GitHub Bot logged work on BEAM-14026:
-----------------------------------------
Author: ASF GitHub Bot
Created on: 03/Mar/22 19:33
Start Date: 03/Mar/22 19:33
Worklog Time Spent: 10m
Work Description: TheNeuralBit commented on a change in pull request
#16988:
URL: https://github.com/apache/beam/pull/16988#discussion_r818990956
##########
File path: sdks/java/extensions/sql/build.gradle
##########
@@ -206,6 +206,14 @@ task runBasicExample(type: JavaExec) {
args = ["--runner=DirectRunner"]
}
+// Run basic SQL example
+task runNestedRowInArrayExample(type: JavaExec) {
+ description = "Run basic SQL example"
+ mainClass = "org.apache.beam.sdk.extensions.sql.example.BeamSqlUnnestExample"
+ classpath = sourceSets.main.runtimeClasspath
+ args = ["--runner=DirectRunner"]
+}
Review comment:
Do we need this example? It looks like the same behavior is verified
continuously in
[BeamSqlDslUnnestRowsTest.java](https://github.com/apache/beam/pull/16988/files#diff-8c0c041856075a2f10a667e185965b258fb8dd695a9a7ab9e4fdb417499fd424)
##########
File path: sdks/java/core/src/main/java/org/apache/beam/sdk/values/Row.java
##########
@@ -108,11 +109,32 @@
/** Return the list of data values. */
public abstract List<Object> getValues();
+ /** This is recursive call to get all the values of the nested rows.
+ The recusion is bounded by the amount of nesting with in the data
+ This mirrors the unnest behavior of calcite towards schema **/
+ public List<Object> getNestedRowBaseValues() {
+ return IntStream.range(0, getFieldCount())
+ .mapToObj(i -> {
+ List<Object> values = new ArrayList<>();
+ FieldType fieldType = this.getSchema().getField(i).getType();
+ if(fieldType.getTypeName().equals(TypeName.ROW)) {
+ Row row = this.getBaseValue(i, Row.class);
+ List<Object> rowValues = row.getNestedRowBaseValues();
+ if(null != rowValues) {
+ values.addAll(rowValues);
+ }
+ } else {
+ values.add(this.getBaseValue(i));
+ }
+ return values.stream();
+ }).flatMap(Function.identity()).collect(Collectors.toList());
+ }
Review comment:
I'd prefer not to add this logic to Row's public API. It's specific to
the needs in unnest. Would it be possible to just define it as a helper there?
##########
File path: sdks/java/core/src/main/java/org/apache/beam/sdk/values/Row.java
##########
@@ -537,7 +559,7 @@ static int deepHashCodeForMap(
}
static boolean deepEqualsForCollection(
- Collection<Object> a, Collection<Object> b, Schema.FieldType
elementType) {
+ Collection<Object> a, Collection<Object> b, Schema.FieldType
elementType) {
Review comment:
I think unnecessary whitespace changes like this are what's making our
CI unhappy. Can you run spotless to format the code? `./gradlew spotlessApply`
You might also double check there aren't other checkstyle failures:
`./gradlew :sdks:java:extensions:sql:checkstyleMain
:sdks:java:extensions:sql:checkstyleTest`
--
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]
Issue Time Tracking
-------------------
Worklog Id: (was: 736246)
Time Spent: 20m (was: 10m)
> Using unnest on a array with rows having nested rows throws
> IllegalArgumentException
> ------------------------------------------------------------------------------------
>
> Key: BEAM-14026
> URL: https://issues.apache.org/jira/browse/BEAM-14026
> Project: Beam
> Issue Type: Bug
> Components: dsl-sql, sdk-java-core
> Affects Versions: 2.36.0
> Reporter: Abhijeet
> Priority: P2
> Fix For: Not applicable
>
> Time Spent: 20m
> Remaining Estimate: 0h
>
> Using unnest with example sql statement like
> "select t.a1, t.a2, t.a3, d.b1, d.b2, d.b4, d.b3.c1, d.b3.c2, d.b3.c3 from
> test t cross join unnest(t.a4) d"
> on the following dataset throws Illegal argument exception
> Java code snippet
> ----
> // define the input row format level3
> Schema level3Type =
> Schema.builder().addInt32Field("c1").addStringField("c2").addDoubleField("c3").build();
> Row level3Row1 = Row.withSchema(level3Type).addValues(1, "row", 1.0).build();
> Row level3Row2 = Row.withSchema(level3Type).addValues(2, "row", 2.0).build();
> Row level3Row3 = Row.withSchema(level3Type).addValues(3, "row", 3.0).build();
> // define the input row format level3
> Schema level2Type =
> Schema.builder().addInt32Field("b1")
> .addStringField("b2")
> .addRowField("b3", level3Type)
> .addDoubleField("b4").build();
> Row level2Row1 = Row.withSchema(level2Type).addValues(1, "row", level3Row1,
> 1.0).build();
> Row level2Row2 = Row.withSchema(level2Type).addValues(2, "row", level3Row2,
> 2.0).build();
> Row level2Row3 = Row.withSchema(level2Type).addValues(3, "row", level3Row3,
> 3.0).build();
> // define the input row format level3
> Schema level1Type =
> Schema.builder().addInt32Field("a1")
> .addStringField("a2")
> .addDoubleField("a3")
> .addArrayField("a4", Schema.FieldType.row(level2Type))
> .build();
> Row level1Row1 = Row.withSchema(level1Type).addValues(1, "row", 1.0,
> Arrays.asList(level2Row1, level2Row2, level2Row3)).build();
> Row level1Row2 = Row.withSchema(level1Type).addValues(2, "row", 2.0,
> Arrays.asList(level2Row1, level2Row2, level2Row3)).build();
> Row level1Row3 = Row.withSchema(level1Type).addValues(3, "row", 3.0,
> Arrays.asList(level2Row1, level2Row2, level2Row3)).build();
> // create a source PCollection with Create.of();
> PCollection<Row> inputTable =
> PBegin.in(p).apply(Create.of(level1Row1, level1Row2,
> level1Row3).withRowSchema(level1Type));
> ----
> Throws following exception
> java.lang.IllegalArgumentException: Row expected 10 fields. initialized with
> 8 fields
--
This message was sent by Atlassian Jira
(v8.20.1#820001)
