use of org.apache.ignite.internal.sql.engine.rel.IgniteHashIndexSpool in project ignite-3 by apache.
the class FilterSpoolMergeToHashIndexSpoolRule method onMatch.
/**
* {@inheritDoc}
*/
@Override
public void onMatch(RelOptRuleCall call) {
final IgniteFilter filter = call.rel(0);
final IgniteTableSpool spool = call.rel(1);
RelOptCluster cluster = spool.getCluster();
RelTraitSet trait = spool.getTraitSet();
CorrelationTrait filterCorr = TraitUtils.correlation(filter);
if (filterCorr.correlated()) {
trait = trait.replace(filterCorr);
}
final RelNode input = spool.getInput();
List<RexNode> searchRow = RexUtils.buildHashSearchRow(cluster, filter.getCondition(), spool.getRowType());
if (nullOrEmpty(searchRow)) {
return;
}
RelNode res = new IgniteHashIndexSpool(cluster, trait.replace(RelCollations.EMPTY), input, searchRow, filter.getCondition());
call.transformTo(res);
}
use of org.apache.ignite.internal.sql.engine.rel.IgniteHashIndexSpool in project ignite-3 by apache.
the class HashIndexSpoolPlannerTest method testSingleKey.
/**
* Check equi-join on not colocated fields. CorrelatedNestedLoopJoinTest is applicable for this case only with
* IndexSpool.
*/
@Test
public void testSingleKey() throws Exception {
IgniteSchema publicSchema = new IgniteSchema("PUBLIC");
IgniteTypeFactory f = new IgniteTypeFactory(IgniteTypeSystem.INSTANCE);
publicSchema.addTable("T0", new TestTable(new RelDataTypeFactory.Builder(f).add("ID", f.createJavaType(Integer.class)).add("JID", f.createJavaType(Integer.class)).add("VAL", f.createJavaType(String.class)).build()) {
@Override
public IgniteDistribution distribution() {
return IgniteDistributions.affinity(0, "T0", "hash");
}
}.addIndex(RelCollations.of(ImmutableIntList.of(1, 0)), "t0_jid_idx"));
publicSchema.addTable("T1", new TestTable(new RelDataTypeFactory.Builder(f).add("ID", f.createJavaType(Integer.class)).add("JID", f.createJavaType(Integer.class)).add("VAL", f.createJavaType(String.class)).build()) {
@Override
public IgniteDistribution distribution() {
return IgniteDistributions.affinity(0, "T1", "hash");
}
}.addIndex(RelCollations.of(ImmutableIntList.of(1, 0)), "t1_jid_idx"));
String sql = "select * " + "from t0 " + "join t1 on t0.jid = t1.jid";
IgniteRel phys = physicalPlan(sql, publicSchema, "MergeJoinConverter", "NestedLoopJoinConverter", "FilterSpoolMergeToSortedIndexSpoolRule");
System.out.println("+++\n" + RelOptUtil.toString(phys));
IgniteHashIndexSpool idxSpool = findFirstNode(phys, byClass(IgniteHashIndexSpool.class));
List<RexNode> searchRow = idxSpool.searchRow();
assertNotNull(searchRow);
assertEquals(3, searchRow.size());
assertTrue(((RexLiteral) searchRow.get(0)).isNull());
assertTrue(((RexLiteral) searchRow.get(2)).isNull());
assertTrue(searchRow.get(1) instanceof RexFieldAccess);
}
use of org.apache.ignite.internal.sql.engine.rel.IgniteHashIndexSpool in project ignite-3 by apache.
the class HashIndexSpoolPlannerTest method testMultipleKeys.
@Test
public void testMultipleKeys() throws Exception {
IgniteSchema publicSchema = new IgniteSchema("PUBLIC");
IgniteTypeFactory f = new IgniteTypeFactory(IgniteTypeSystem.INSTANCE);
publicSchema.addTable("T0", new TestTable(new RelDataTypeFactory.Builder(f).add("ID", f.createJavaType(Integer.class)).add("JID0", f.createJavaType(Integer.class)).add("JID1", f.createJavaType(Integer.class)).add("VAL", f.createJavaType(String.class)).build()) {
@Override
public IgniteDistribution distribution() {
return IgniteDistributions.affinity(0, "T0", "hash");
}
});
publicSchema.addTable("T1", new TestTable(new RelDataTypeFactory.Builder(f).add("ID", f.createJavaType(Integer.class)).add("JID0", f.createJavaType(Integer.class)).add("JID1", f.createJavaType(Integer.class)).add("VAL", f.createJavaType(String.class)).build()) {
@Override
public IgniteDistribution distribution() {
return IgniteDistributions.affinity(0, "T1", "hash");
}
}.addIndex(RelCollations.of(ImmutableIntList.of(1, 0)), "t1_jid0_idx"));
String sql = "select * " + "from t0 " + "join t1 on t0.jid0 = t1.jid0 and t0.jid1 = t1.jid1";
IgniteRel phys = physicalPlan(sql, publicSchema, "MergeJoinConverter", "NestedLoopJoinConverter", "FilterSpoolMergeToSortedIndexSpoolRule");
IgniteHashIndexSpool idxSpool = findFirstNode(phys, byClass(IgniteHashIndexSpool.class));
List<RexNode> searcRow = idxSpool.searchRow();
assertNotNull(searcRow);
assertEquals(4, searcRow.size());
assertTrue(((RexLiteral) searcRow.get(0)).isNull());
assertTrue(searcRow.get(1) instanceof RexFieldAccess);
assertTrue(searcRow.get(2) instanceof RexFieldAccess);
assertTrue(((RexLiteral) searcRow.get(3)).isNull());
}
use of org.apache.ignite.internal.sql.engine.rel.IgniteHashIndexSpool in project ignite-3 by apache.
the class HashIndexSpoolPlannerTest method testSourceWithoutCollation.
/**
* Check equi-join on not colocated fields without indexes.
*/
@Test
public void testSourceWithoutCollation() throws Exception {
IgniteSchema publicSchema = new IgniteSchema("PUBLIC");
IgniteTypeFactory f = new IgniteTypeFactory(IgniteTypeSystem.INSTANCE);
publicSchema.addTable("T0", new TestTable(new RelDataTypeFactory.Builder(f).add("ID", f.createJavaType(Integer.class)).add("JID", f.createJavaType(Integer.class)).add("VAL", f.createJavaType(String.class)).build()) {
@Override
public IgniteDistribution distribution() {
return IgniteDistributions.affinity(0, "T0", "hash");
}
});
publicSchema.addTable("T1", new TestTable(new RelDataTypeFactory.Builder(f).add("ID", f.createJavaType(Integer.class)).add("JID", f.createJavaType(Integer.class)).add("VAL", f.createJavaType(String.class)).build()) {
@Override
public IgniteDistribution distribution() {
return IgniteDistributions.affinity(0, "T1", "hash");
}
});
String sql = "select * " + "from t0 " + "join t1 on t0.jid = t1.jid";
IgniteRel phys = physicalPlan(sql, publicSchema, "MergeJoinConverter", "NestedLoopJoinConverter");
IgniteHashIndexSpool idxSpool = findFirstNode(phys, byClass(IgniteHashIndexSpool.class));
List<RexNode> searchRow = idxSpool.searchRow();
assertNotNull(searchRow);
assertEquals(3, searchRow.size());
assertTrue(((RexLiteral) searchRow.get(0)).isNull());
assertTrue(((RexLiteral) searchRow.get(2)).isNull());
assertTrue(searchRow.get(1) instanceof RexFieldAccess);
}
Aggregations