Search in sources :

Example 1 with TopicWriteType

use of com.bakdata.quick.common.api.model.TopicWriteType in project quick by bakdata.

the class KafkaTopicService method createTopic.

@SuppressWarnings("RxReturnValueIgnored")
@Override
public Completable createTopic(final String name, final QuickTopicType keyType, final QuickTopicType valueType, final TopicCreationData requestData) {
    log.info("Create new topic {} with data {}", name, requestData);
    // we don't need the cache, so make sure we get the current information
    this.schemaRegistryClient.reset();
    // first, check if topic might already exist in topic registry or kafka itself and then make also sure there
    final Completable kafkaStateCheck = Completable.mergeArray(this.checkKafka(name), this.checkTopicRegistry(name));
    final Single<Optional<QuickSchemas>> keySchema = this.getQuickSchemas(requestData.getKeySchema()).cache();
    final Single<Optional<QuickSchemas>> valueSchema = this.getQuickSchemas(requestData.getValueSchema()).cache();
    final Completable schemaRegistryCheck = Completable.defer(() -> Completable.mergeArray(keySchema.flatMapCompletable(schema -> this.checkSchemaRegistry(name + "-key", schema)), valueSchema.flatMapCompletable(schema -> this.checkSchemaRegistry(name + "-value", schema))));
    final Completable stateCheck = kafkaStateCheck.andThen(schemaRegistryCheck);
    // create topic in kafka and deploy a mirror application
    final Completable kafkaTopicCreation = this.createKafkaTopic(name);
    final Completable mirrorCreation = this.createMirror(name, requestData.getRetentionTime());
    // default to mutable topic write type
    final TopicWriteType writeType = Objects.requireNonNullElse(requestData.getWriteType(), TopicWriteType.MUTABLE);
    // register at topic registry (value schema can be nullable)
    // todo evaluate whether the schema should be part of the topic registry
    final Completable topicRegister = Completable.defer(() -> {
        log.debug("Register subject '{}' with topic registry", name);
        return valueSchema.flatMapCompletable(schema -> {
            final String graphQLSchema = schema.map(QuickSchemas::getGraphQLSchema).orElse(null);
            final TopicData topicData = new TopicData(name, writeType, keyType, valueType, graphQLSchema);
            return this.topicRegistryClient.register(name, topicData);
        });
    });
    // register potential avro schema with the schema registry
    final Completable keyRegister = keySchema.flatMapCompletable(schemas -> this.registerSchema(name, schemas, KEY));
    final Completable valueRegister = valueSchema.flatMapCompletable(schemas -> this.registerSchema(name, schemas, VALUE));
    final Completable creationOperations = Completable.mergeArray(kafkaTopicCreation, mirrorCreation, topicRegister, keyRegister, valueRegister);
    return stateCheck.andThen(creationOperations.doOnError(ignored -> this.deleteTopic(name)));
}
Also used : MirrorCreationData(com.bakdata.quick.common.api.model.manager.creation.MirrorCreationData) TopicRegistryClient(com.bakdata.quick.common.api.client.TopicRegistryClient) SchemaRegistryClient(io.confluent.kafka.schemaregistry.client.SchemaRegistryClient) KEY(com.bakdata.quick.common.api.model.KeyValueEnum.KEY) TopicWriteType(com.bakdata.quick.common.api.model.TopicWriteType) Completable(io.reactivex.Completable) GatewayClient(com.bakdata.quick.common.api.client.GatewayClient) Singleton(javax.inject.Singleton) Single(io.reactivex.Single) Supplier(java.util.function.Supplier) GraphQLToAvroConverter(com.bakdata.quick.manager.graphql.GraphQLToAvroConverter) Value(lombok.Value) AdminClient(org.apache.kafka.clients.admin.AdminClient) KafkaConfig(com.bakdata.quick.common.config.KafkaConfig) MirrorService(com.bakdata.quick.manager.mirror.MirrorService) Duration(java.time.Duration) KeyValueEnum(com.bakdata.quick.common.api.model.KeyValueEnum) QuickTopicConfig(com.bakdata.quick.common.config.QuickTopicConfig) QuickTopicType(com.bakdata.quick.common.type.QuickTopicType) Schema(org.apache.avro.Schema) ResourcePrefix(com.bakdata.quick.manager.k8s.resource.QuickResources.ResourcePrefix) NewTopic(org.apache.kafka.clients.admin.NewTopic) GatewaySchema(com.bakdata.quick.common.api.model.manager.GatewaySchema) IOException(java.io.IOException) VALUE(com.bakdata.quick.common.api.model.KeyValueEnum.VALUE) BadArgumentException(com.bakdata.quick.common.exception.BadArgumentException) TopicCreationData(com.bakdata.quick.common.api.model.manager.creation.TopicCreationData) Objects(java.util.Objects) List(java.util.List) Slf4j(lombok.extern.slf4j.Slf4j) TopicData(com.bakdata.quick.common.api.model.TopicData) GatewayService(com.bakdata.quick.manager.gateway.GatewayService) Optional(java.util.Optional) CachedSchemaRegistryClient(io.confluent.kafka.schemaregistry.client.CachedSchemaRegistryClient) Completable(io.reactivex.Completable) Optional(java.util.Optional) TopicData(com.bakdata.quick.common.api.model.TopicData) TopicWriteType(com.bakdata.quick.common.api.model.TopicWriteType)

Aggregations

GatewayClient (com.bakdata.quick.common.api.client.GatewayClient)1 TopicRegistryClient (com.bakdata.quick.common.api.client.TopicRegistryClient)1 KeyValueEnum (com.bakdata.quick.common.api.model.KeyValueEnum)1 KEY (com.bakdata.quick.common.api.model.KeyValueEnum.KEY)1 VALUE (com.bakdata.quick.common.api.model.KeyValueEnum.VALUE)1 TopicData (com.bakdata.quick.common.api.model.TopicData)1 TopicWriteType (com.bakdata.quick.common.api.model.TopicWriteType)1 GatewaySchema (com.bakdata.quick.common.api.model.manager.GatewaySchema)1 MirrorCreationData (com.bakdata.quick.common.api.model.manager.creation.MirrorCreationData)1 TopicCreationData (com.bakdata.quick.common.api.model.manager.creation.TopicCreationData)1 KafkaConfig (com.bakdata.quick.common.config.KafkaConfig)1 QuickTopicConfig (com.bakdata.quick.common.config.QuickTopicConfig)1 BadArgumentException (com.bakdata.quick.common.exception.BadArgumentException)1 QuickTopicType (com.bakdata.quick.common.type.QuickTopicType)1 GatewayService (com.bakdata.quick.manager.gateway.GatewayService)1 GraphQLToAvroConverter (com.bakdata.quick.manager.graphql.GraphQLToAvroConverter)1 ResourcePrefix (com.bakdata.quick.manager.k8s.resource.QuickResources.ResourcePrefix)1 MirrorService (com.bakdata.quick.manager.mirror.MirrorService)1 CachedSchemaRegistryClient (io.confluent.kafka.schemaregistry.client.CachedSchemaRegistryClient)1 SchemaRegistryClient (io.confluent.kafka.schemaregistry.client.SchemaRegistryClient)1