Examples with RegisterTopic io.confluent.ksql.parser.tree.RegisterTopic used on opensource projects

Example 6 with RegisterTopic

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

the class CliUtils method getAvroSchemaIfAvroTopic.

public Optional<String> getAvroSchemaIfAvroTopic(SqlBaseParser.RegisterTopicContext registerTopicContext) {
    AstBuilder astBuilder = new AstBuilder(null);
    RegisterTopic registerTopic = (RegisterTopic) astBuilder.visitRegisterTopic(registerTopicContext);
    if (registerTopic.getProperties().get(DdlConfig.VALUE_FORMAT_PROPERTY) == null) {
        throw new KsqlException("VALUE_FORMAT is not set for the topic.");
    }
    if (registerTopic.getProperties().get(DdlConfig.VALUE_FORMAT_PROPERTY).toString().equalsIgnoreCase("'AVRO'")) {
        if (registerTopic.getProperties().containsKey(DdlConfig.AVRO_SCHEMA_FILE)) {
            String avroSchema = getAvroSchema(AstBuilder.unquote(registerTopic.getProperties().get(DdlConfig.AVRO_SCHEMA_FILE).toString(), "'"));
            return Optional.of(avroSchema);
        } else {
            throw new KsqlException("You need to provide avro schema file path for topics in avro format.");
        }
    }
    return Optional.empty();
}

Example 7 with RegisterTopic

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

the class KsqlResourceTest method testInstantRegisterTopic.

@Test
public void testInstantRegisterTopic() throws Exception {
    KsqlResource testResource = TestKsqlResourceUtil.get(ksqlEngine, ksqlRestConfig);
    final String ksqlTopic = "FOO";
    final String kafkaTopic = "bar";
    final String format = "json";
    final String ksqlString = String.format("REGISTER TOPIC %s WITH (kafka_topic='%s', value_format='%s');", ksqlTopic, kafkaTopic, format);
    final Map<String, Expression> createTopicProperties = new HashMap<>();
    createTopicProperties.put(DdlConfig.KAFKA_TOPIC_NAME_PROPERTY, new StringLiteral(kafkaTopic));
    createTopicProperties.put(DdlConfig.VALUE_FORMAT_PROPERTY, new StringLiteral(format));
    final RegisterTopic ksqlStatement = new RegisterTopic(QualifiedName.of(ksqlTopic), false, createTopicProperties);
    final CommandId commandId = new CommandId(CommandId.Type.TOPIC, ksqlTopic, CommandId.Action.CREATE);
    final CommandStatus commandStatus = new CommandStatus(CommandStatus.Status.QUEUED, "Statement written to command topic");
    final CommandStatusEntity expectedCommandStatusEntity = new CommandStatusEntity(ksqlString, commandId, commandStatus);
    final Map<String, Object> streamsProperties = Collections.emptyMap();
    KsqlEntity testKsqlEntity = makeSingleRequest(testResource, ksqlString, ksqlStatement, streamsProperties, KsqlEntity.class);
    assertEquals(expectedCommandStatusEntity, testKsqlEntity);
}