Examples with RelationSource io.crate.analyze.RelationSource used on opensource projects

Example 1 with RelationSource

use of io.crate.analyze.RelationSource in project crate by crate.

the class RelationNormalizerTest method testSubSelectOnLeftJoinWithFilterRewrittenToInner.

@Test
public void testSubSelectOnLeftJoinWithFilterRewrittenToInner() throws Exception {
    QueriedRelation relation = normalize("select col1, col2 from ( " + "  select t1.a as col1, t2.i as col2, t2.y as col3 " + "  from t1 left join t2 on t1.a = t2.b where t2.y > 60) as t " + "where col1 = 'a' order by col3");
    assertThat(relation, instanceOf(MultiSourceSelect.class));
    MultiSourceSelect mss = (MultiSourceSelect) relation;
    assertThat(mss.querySpec(), isSQL("SELECT doc.t1.a, doc.t2.i ORDER BY doc.t2.y"));
    assertThat(mss.joinPairs().get(0).joinType(), is(JoinType.INNER));
    assertThat(mss.joinPairs().get(0).condition(), isSQL("(doc.t1.a = doc.t2.b)"));
    // make sure that where clause was pushed down and didn't disappear somehow
    RelationSource t1 = mss.sources().get(T3.T1);
    assertThat(t1.querySpec().where().query(), isSQL("(doc.t1.a = 'a')"));
    RelationSource t2 = mss.sources().get(T3.T2);
    assertThat(t2.querySpec().where().query(), isSQL("(doc.t2.y > 60)"));
}

Example 2 with RelationSource

use of io.crate.analyze.RelationSource in project crate by crate.

the class RelationNormalizerTest method testFullJoinWithFiltersRewrittenToRight.

@Test
public void testFullJoinWithFiltersRewrittenToRight() throws Exception {
    QueriedRelation relation = normalize("select t1.a as col1, t2.i as col2 " + "from t1 full join t2 on t1.a = t2.b " + "where t1.x is null and t2.b = 'b'");
    assertThat(relation, instanceOf(MultiSourceSelect.class));
    MultiSourceSelect mss = (MultiSourceSelect) relation;
    assertThat(mss.querySpec(), isSQL("SELECT doc.t1.a, doc.t2.i WHERE (ISNULL doc.t1.x)"));
    assertThat(mss.joinPairs().get(0).joinType(), is(JoinType.RIGHT));
    assertThat(mss.joinPairs().get(0).left().toString(), is("doc.t1"));
    assertThat(mss.joinPairs().get(0).right().toString(), is("doc.t2"));
    assertThat(mss.joinPairs().get(0).condition(), isSQL("(doc.t1.a = doc.t2.b)"));
    // make sure that where clause for t2 wasn't pushed down since but be applied after the FULL join
    RelationSource t1 = mss.sources().get(T3.T1);
    assertThat(t1.querySpec().where().query(), is(nullValue()));
    RelationSource t2 = mss.sources().get(T3.T2);
    assertThat(t2.querySpec().where().query(), isSQL("(doc.t2.b = 'b')"));
}

Example 3 with RelationSource

use of io.crate.analyze.RelationSource in project crate by crate.

the class RelationNormalizerTest method testFullJoinWithFiltersRewrittenToInner.

@Test
public void testFullJoinWithFiltersRewrittenToInner() throws Exception {
    QueriedRelation relation = normalize("select t1.a, t2.i " + "from t1 full join t2 on t1.a = t2.b " + "where t2.y is not null and t1.a = 'a'");
    assertThat(relation, instanceOf(MultiSourceSelect.class));
    MultiSourceSelect mss = (MultiSourceSelect) relation;
    assertThat(mss.querySpec(), isSQL("SELECT doc.t1.a, doc.t2.i"));
    assertThat(mss.joinPairs().get(0).joinType(), is(JoinType.INNER));
    assertThat(mss.joinPairs().get(0).condition(), isSQL("(doc.t1.a = doc.t2.b)"));
    // make sure that the conditions of where clause were pushed down to the relations
    RelationSource t1 = mss.sources().get(T3.T1);
    assertThat(t1.querySpec().where().query(), isSQL("(doc.t1.a = 'a')"));
    RelationSource t2 = mss.sources().get(T3.T2);
    assertThat(t2.querySpec().where().query(), isSQL("(NOT (ISNULL doc.t2.y))"));
}

Example 4 with RelationSource

use of io.crate.analyze.RelationSource in project crate by crate.

the class RelationNormalizerTest method testFullJoinWithFiltersNotRewritten.

@Test
public void testFullJoinWithFiltersNotRewritten() throws Exception {
    QueriedRelation relation = normalize("select t1.a as col1, t2.i as col2, t2.y as col3 " + "from t1 full join t2 on t1.a = t2.b " + "where t1.x is null and t2.b = null order by col3");
    assertThat(relation, instanceOf(MultiSourceSelect.class));
    MultiSourceSelect mss = (MultiSourceSelect) relation;
    assertThat(mss.querySpec(), isSQL("SELECT doc.t1.a, doc.t2.i, doc.t2.y WHERE ((ISNULL doc.t1.x) AND NULL) ORDER BY doc.t2.y"));
    assertThat(mss.joinPairs().get(0).joinType(), is(JoinType.FULL));
    assertThat(mss.joinPairs().get(0).left().toString(), is("doc.t1"));
    assertThat(mss.joinPairs().get(0).right().toString(), is("doc.t2"));
    assertThat(mss.joinPairs().get(0).condition(), isSQL("(doc.t1.a = doc.t2.b)"));
    // make sure that where clause wasn't pushed down since but be applied after the FULL join
    RelationSource t1 = mss.sources().get(T3.T1);
    assertThat(t1.querySpec().where().query(), is(nullValue()));
    RelationSource t2 = mss.sources().get(T3.T2);
    assertThat(t2.querySpec().where().query(), is(nullValue()));
}

Example 5 with RelationSource

use of io.crate.analyze.RelationSource in project crate by crate.

the class MultiSourceAggregationConsumer method removeAggregationsAndLimitsFromMSS.

private static void removeAggregationsAndLimitsFromMSS(MultiSourceSelect mss, SplitPoints splitPoints) {
    QuerySpec querySpec = mss.querySpec();
    List<Symbol> outputs = Lists2.concatUnique(splitPoints.toCollect(), extractFieldsFromJoinConditions(mss));
    querySpec.outputs(outputs);
    querySpec.hasAggregates(false);
    // Limit & offset must be applied after the aggregation, so remove it from mss and sources.
    // OrderBy can be ignored because it's also applied after aggregation but there is always only 1 row so it
    // wouldn't have any effect.
    removeLimitOffsetAndOrder(querySpec);
    for (RelationSource relationSource : mss.sources().values()) {
        removeLimitOffsetAndOrder(relationSource.querySpec());
    }
    // need to change the types on the fields of the MSS to match the new outputs
    ListIterator<Field> fieldsIt = mss.fields().listIterator();
    Iterator<Function> outputsIt = splitPoints.aggregates().iterator();
    while (fieldsIt.hasNext()) {
        Field field = fieldsIt.next();
        Symbol output = outputsIt.next();
        fieldsIt.set(new Field(field.relation(), field.path(), output.valueType()));
    }
}