Make it possible to insert test data

polypheny · Dec 18, 2024 · cc245ac · cc245ac
1 parent 34b2fb7
commit cc245ac
Showing 1 changed file with 17 additions and 1 deletion.
diff --git a/dbms/src/main/java/org/polypheny/db/routing/routers/BaseRouter.java b/dbms/src/main/java/org/polypheny/db/routing/routers/BaseRouter.java
@@ -196,10 +196,26 @@ protected List<RoutedAlgBuilder> handleGeneric( AlgNode node, List<RoutedAlgBuil
                 break;
             case 2:
                 builders.forEach(
-                        builder -> builder.replaceTop( node.copy( node.getTraitSet(), ImmutableList.of( builder.peek( 1 ), builder.peek( 0 ) ) ), 2 )
+                        builder -> builder.replaceTop(
+                                node.copy( node.getTraitSet(), ImmutableList.of( builder.peek( 1 ), builder.peek( 0 ) ) )
+                                , 2 )
                 );
                 break;
             default:
+                if ( node instanceof LogicalRelUnion l ) {
+                    builders.forEach(
+                            builder -> {
+                                List<AlgNode> inputs = new ArrayList<>();
+                                for ( int i = 0; i < node.getInputs().size(); i++ ) {
+                                    inputs.add( builder.peek( i ) );
+                                }
+                                builder.replaceTop(
+                                        node.copy( node.getTraitSet(), inputs )
+                                        , l.getInputs().size() );
+                            }
+                    );
+                    break;
+                }
                 throw new GenericRuntimeException( "Unexpected number of input elements: " + node.getInputs().size() );
         }
         return builders;