Search in sources :

Example 11 with JoinedSource

use of io.confluent.ksql.parser.tree.JoinedSource in project ksql by confluentinc.

the class KsqlParserTest method shouldSetWithinExpressionWithBeforeAndAfterAndGracePeriod.

@Test
public void shouldSetWithinExpressionWithBeforeAndAfterAndGracePeriod() {
    final String statementString = "CREATE STREAM foobar as SELECT * from TEST1 JOIN ORDERS " + "WITHIN (10 seconds, 20 minutes) GRACE PERIOD 10 minutes " + "ON TEST1.col1 = ORDERS.ORDERID ;";
    final Statement statement = KsqlParserTestUtil.buildSingleAst(statementString, metaStore).getStatement();
    assertThat(statement, instanceOf(CreateStreamAsSelect.class));
    final CreateStreamAsSelect createStreamAsSelect = (CreateStreamAsSelect) statement;
    final Query query = createStreamAsSelect.getQuery();
    assertThat(query.getFrom(), instanceOf(Join.class));
    final JoinedSource join = Iterables.getOnlyElement(((Join) query.getFrom()).getRights());
    assertTrue(join.getWithinExpression().isPresent());
    final WithinExpression withinExpression = join.getWithinExpression().get();
    assertThat(withinExpression.getBefore(), is(10L));
    assertThat(withinExpression.getAfter(), is(20L));
    assertThat(withinExpression.getBeforeTimeUnit(), is(TimeUnit.SECONDS));
    assertThat(withinExpression.getAfterTimeUnit(), is(TimeUnit.MINUTES));
    assertThat(withinExpression.getGrace(), is(Optional.of(new WindowTimeClause(10, TimeUnit.MINUTES))));
    assertThat(join.getType(), is(JoinedSource.Type.INNER));
}
Also used : JoinedSource(io.confluent.ksql.parser.tree.JoinedSource) Query(io.confluent.ksql.parser.tree.Query) TerminateQuery(io.confluent.ksql.parser.tree.TerminateQuery) Statement(io.confluent.ksql.parser.tree.Statement) PreparedStatement(io.confluent.ksql.parser.KsqlParser.PreparedStatement) Join(io.confluent.ksql.parser.tree.Join) WithinExpression(io.confluent.ksql.parser.tree.WithinExpression) WindowTimeClause(io.confluent.ksql.execution.windows.WindowTimeClause) Matchers.containsString(org.hamcrest.Matchers.containsString) CreateStreamAsSelect(io.confluent.ksql.parser.tree.CreateStreamAsSelect) Test(org.junit.Test)

Example 12 with JoinedSource

use of io.confluent.ksql.parser.tree.JoinedSource in project ksql by confluentinc.

the class KsqlParserTest method shouldHaveOuterJoinTypeWhenOuterKeywordIsSpecified.

@Test
public void shouldHaveOuterJoinTypeWhenOuterKeywordIsSpecified() {
    final String statementString = "CREATE STREAM foobar as SELECT * from TEST1 FULL OUTER JOIN " + "TEST2 ON TEST1.col1 = TEST2.col1;";
    final Statement statement = KsqlParserTestUtil.buildSingleAst(statementString, metaStore).getStatement();
    assertThat(statement, instanceOf(CreateStreamAsSelect.class));
    final CreateStreamAsSelect createStreamAsSelect = (CreateStreamAsSelect) statement;
    final Query query = createStreamAsSelect.getQuery();
    assertThat(query.getFrom(), instanceOf(Join.class));
    final JoinedSource join = Iterables.getOnlyElement(((Join) query.getFrom()).getRights());
    assertThat(join.getType(), is(JoinedSource.Type.OUTER));
}
Also used : JoinedSource(io.confluent.ksql.parser.tree.JoinedSource) Query(io.confluent.ksql.parser.tree.Query) TerminateQuery(io.confluent.ksql.parser.tree.TerminateQuery) Statement(io.confluent.ksql.parser.tree.Statement) PreparedStatement(io.confluent.ksql.parser.KsqlParser.PreparedStatement) Join(io.confluent.ksql.parser.tree.Join) Matchers.containsString(org.hamcrest.Matchers.containsString) CreateStreamAsSelect(io.confluent.ksql.parser.tree.CreateStreamAsSelect) Test(org.junit.Test)

Example 13 with JoinedSource

use of io.confluent.ksql.parser.tree.JoinedSource in project ksql by confluentinc.

the class StatementRewriterTest method shouldRewriteJoin.

@Test
public void shouldRewriteJoin() {
    // Given:
    final Join join = givenJoin(Optional.empty());
    // When:
    final AstNode rewritten = rewriter.rewrite(join, context);
    // Then:
    assertThat(rewritten, equalTo(new Join(location, rewrittenRelation, ImmutableList.of(new JoinedSource(Optional.empty(), rewrittenRightRelation, Type.LEFT, joinCriteria, Optional.empty())))));
}
Also used : JoinedSource(io.confluent.ksql.parser.tree.JoinedSource) Join(io.confluent.ksql.parser.tree.Join) AstNode(io.confluent.ksql.parser.tree.AstNode) Test(org.junit.Test)

Example 14 with JoinedSource

use of io.confluent.ksql.parser.tree.JoinedSource in project ksql by confluentinc.

the class SqlFormatterTest method shouldFormatMultipleJoinsWithDifferentTypes.

@Test
public void shouldFormatMultipleJoinsWithDifferentTypes() {
    final List<JoinedSource> rights = ImmutableList.of(new JoinedSource(Optional.empty(), rightAlias, JoinedSource.Type.LEFT, criteria, Optional.of(new WithinExpression(10, TimeUnit.SECONDS))), new JoinedSource(Optional.empty(), right2Alias, JoinedSource.Type.INNER, criteria2, Optional.of(new WithinExpression(10, TimeUnit.SECONDS))));
    final Join join = new Join(leftAlias, rights);
    final String expected = "`left` L" + "\nLEFT OUTER JOIN `right` R WITHIN 10 SECONDS ON (('left.col0' = 'right.col0'))" + "\nINNER JOIN `right2` R2 WITHIN 10 SECONDS ON (('left.col0' = 'right2.col0'))";
    assertEquals(expected, SqlFormatter.formatSql(join));
}
Also used : JoinedSource(io.confluent.ksql.parser.tree.JoinedSource) WithinExpression(io.confluent.ksql.parser.tree.WithinExpression) Join(io.confluent.ksql.parser.tree.Join) StringContains.containsString(org.hamcrest.core.StringContains.containsString) Test(org.junit.Test)

Example 15 with JoinedSource

use of io.confluent.ksql.parser.tree.JoinedSource in project ksql by confluentinc.

the class SqlFormatterTest method shouldFormatMultipleLeftJoinWithoutWindow.

@Test
public void shouldFormatMultipleLeftJoinWithoutWindow() {
    final List<JoinedSource> rights = ImmutableList.of(new JoinedSource(Optional.empty(), rightAlias, JoinedSource.Type.LEFT, criteria, Optional.empty()), new JoinedSource(Optional.empty(), right2Alias, JoinedSource.Type.LEFT, criteria2, Optional.empty()));
    final Join join = new Join(leftAlias, rights);
    final String expected = "`left` L" + "\nLEFT OUTER JOIN `right` R ON (('left.col0' = 'right.col0'))" + "\nLEFT OUTER JOIN `right2` R2 ON (('left.col0' = 'right2.col0'))";
    assertEquals(expected, SqlFormatter.formatSql(join));
}
Also used : JoinedSource(io.confluent.ksql.parser.tree.JoinedSource) Join(io.confluent.ksql.parser.tree.Join) StringContains.containsString(org.hamcrest.core.StringContains.containsString) Test(org.junit.Test)

Aggregations

Join (io.confluent.ksql.parser.tree.Join)20 JoinedSource (io.confluent.ksql.parser.tree.JoinedSource)20 Test (org.junit.Test)20 WithinExpression (io.confluent.ksql.parser.tree.WithinExpression)10 PreparedStatement (io.confluent.ksql.parser.KsqlParser.PreparedStatement)9 CreateStreamAsSelect (io.confluent.ksql.parser.tree.CreateStreamAsSelect)9 Query (io.confluent.ksql.parser.tree.Query)9 Statement (io.confluent.ksql.parser.tree.Statement)9 TerminateQuery (io.confluent.ksql.parser.tree.TerminateQuery)9 Matchers.containsString (org.hamcrest.Matchers.containsString)9 StringContains.containsString (org.hamcrest.core.StringContains.containsString)9 WindowTimeClause (io.confluent.ksql.execution.windows.WindowTimeClause)2 AstNode (io.confluent.ksql.parser.tree.AstNode)2