dev.snowdrop.vertx.sample.kafka.KafkaLog Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of vertx-spring-boot-sample-kafka Show documentation

There is a newer version: 1.4.1

Show newest version

package dev.snowdrop.vertx.sample.kafka;

import java.time.Duration;
import java.util.List;
import java.util.concurrent.CopyOnWriteArrayList;

import dev.snowdrop.vertx.kafka.ConsumerRecord;
import dev.snowdrop.vertx.kafka.KafkaConsumer;
import org.springframework.beans.factory.DisposableBean;
import org.springframework.beans.factory.InitializingBean;
import reactor.core.Disposable;

import static dev.snowdrop.vertx.sample.kafka.KafkaSampleApplication.LOG_TOPIC;

final class KafkaLog implements InitializingBean, DisposableBean {

    private final List messages = new CopyOnWriteArrayList<>();

    private final KafkaConsumer consumer;

    private Disposable consumerDisposer;

    KafkaLog(KafkaConsumer consumer) {
        this.consumer = consumer;
    }

    @Override
    public void afterPropertiesSet() {
        consumerDisposer = consumer.subscribe(LOG_TOPIC)
            .thenMany(consumer.flux())
            .log("Kafka log consumer")
            .map(ConsumerRecord::value)
            .subscribe(messages::add);
    }

    @Override
    public void destroy() {
        if (consumerDisposer != null) {
            consumerDisposer.dispose();
        }
        consumer.unsubscribe()
            .block(Duration.ofSeconds(2));
    }

    public List getMessages() {
        return messages;
    }
}