Examples with QueryDescriptionEntity io.confluent.ksql.rest.entity.QueryDescriptionEntity used on opensource projects

Example 1 with QueryDescriptionEntity

use of io.confluent.ksql.rest.entity.QueryDescriptionEntity in project ksql by confluentinc.

the class KsqlResourceTest method shouldExplainQueryStatement.

@Test
public void shouldExplainQueryStatement() {
    // Given:
    final String ksqlQueryString = "SELECT * FROM test_stream EMIT CHANGES;";
    final String ksqlString = "EXPLAIN " + ksqlQueryString;
    // When:
    final QueryDescriptionEntity query = makeSingleRequest(ksqlString, QueryDescriptionEntity.class);
    // Then:
    validateTransientQueryDescription(ksqlQueryString, emptyMap(), query);
}

Example 2 with QueryDescriptionEntity

use of io.confluent.ksql.rest.entity.QueryDescriptionEntity in project ksql by confluentinc.

the class KsqlResourceTest method shouldExplainQueryId.

@Test
public void shouldExplainQueryId() {
    // Given:
    final Map<String, Object> overriddenProperties = Collections.singletonMap("ksql.streams.auto.offset.reset", "earliest");
    final PersistentQueryMetadata queryMetadata = createQuery("CREATE STREAM test_explain AS SELECT * FROM test_stream;", overriddenProperties);
    // When:
    final QueryDescriptionEntity query = makeSingleRequest("EXPLAIN " + queryMetadata.getQueryId() + ";", QueryDescriptionEntity.class);
    // Then:
    validateQueryDescription(queryMetadata, overriddenProperties, query);
}

Example 3 with QueryDescriptionEntity

use of io.confluent.ksql.rest.entity.QueryDescriptionEntity in project ksql by confluentinc.

the class ConsoleTest method shouldPrintExplainQueryWithError.

@Test
public void shouldPrintExplainQueryWithError() {
    final long timestamp = 1596644936314L;
    // Given:
    final QueryDescriptionEntity queryEntity = new QueryDescriptionEntity("statement", new QueryDescription(new QueryId("id"), "statement", Optional.empty(), ImmutableList.of(new FieldInfo("name", new SchemaInfo(SqlBaseType.STRING, ImmutableList.of(), null), Optional.empty())), ImmutableSet.of("source"), ImmutableSet.of("sink"), "topology", "executionPlan", ImmutableMap.of("overridden.prop", 42), ImmutableMap.of(new KsqlHostInfoEntity("foo", 123), KsqlQueryStatus.ERROR), KsqlQueryType.PERSISTENT, ImmutableList.of(new QueryError(timestamp, "error", Type.SYSTEM)), ImmutableSet.of(new StreamsTaskMetadata("test", Collections.emptySet(), Optional.empty())), "consumerGroupId"));
    final KsqlEntityList entityList = new KsqlEntityList(ImmutableList.of(queryEntity));
    // When:
    console.printKsqlEntityList(entityList);
    // Then:
    final String output = terminal.getOutputString();
    Approvals.verify(output, approvalOptions);
}

Example 4 with QueryDescriptionEntity

use of io.confluent.ksql.rest.entity.QueryDescriptionEntity in project ksql by confluentinc.

the class ExplainExecutorTest method shouldExplainStatement.

@Test
public void shouldExplainStatement() {
    // Given:
    engine.givenSource(DataSourceType.KSTREAM, "Y");
    final String statementText = "SELECT * FROM Y EMIT CHANGES;";
    final ConfiguredStatement<Explain> explain = (ConfiguredStatement<Explain>) engine.configure("EXPLAIN " + statementText);
    // When:
    final QueryDescriptionEntity query = (QueryDescriptionEntity) customExecutors.explain().execute(explain, sessionProperties, engine.getEngine(), engine.getServiceContext()).getEntity().orElseThrow(IllegalStateException::new);
    // Then:
    assertThat(query.getQueryDescription().getStatementText(), equalTo(statementText));
    assertThat(query.getQueryDescription().getSources(), containsInAnyOrder("Y"));
    assertThat(query.getQueryDescription().getKsqlHostQueryStatus(), equalTo(Collections.emptyMap()));
}

Example 5 with QueryDescriptionEntity

use of io.confluent.ksql.rest.entity.QueryDescriptionEntity in project ksql by confluentinc.

the class ExplainExecutorTest method shouldExplainQueryId.

@Test
public void shouldExplainQueryId() {
    // Given:
    final ConfiguredStatement<Explain> explain = (ConfiguredStatement<Explain>) engine.configure("EXPLAIN id;");
    final PersistentQueryMetadata metadata = givenPersistentQuery("id");
    final KsqlEngine engine = mock(KsqlEngine.class);
    when(engine.getPersistentQuery(metadata.getQueryId())).thenReturn(Optional.of(metadata));
    // When:
    final QueryDescriptionEntity query = (QueryDescriptionEntity) customExecutors.explain().execute(explain, sessionProperties, engine, this.engine.getServiceContext()).getEntity().orElseThrow(IllegalStateException::new);
    // Then:
    assertThat(query.getQueryDescription(), equalTo(QueryDescriptionFactory.forQueryMetadata(metadata, Collections.singletonMap(new KsqlHostInfoEntity(LOCAL_HOST), STATE))));
}