如何在kafka 0.9.0中使用多线程消费者?

时间:2016-03-29 01:47:56

标签: java multithreading apache-kafka distributed-computing apache-flink

卡夫卡的文件给出了以下描述的方法:

  

每个线程一个消费者:一个简单的选择是为每个线程提供自己的消费者>实例。

我的代码:

public class KafkaConsumerRunner implements Runnable {

    private final AtomicBoolean closed = new AtomicBoolean(false);
    private final CloudKafkaConsumer consumer;
    private final String topicName;

    public KafkaConsumerRunner(CloudKafkaConsumer consumer, String topicName) {
        this.consumer = consumer;
        this.topicName = topicName;
    }

    @Override
    public void run() {
        try {
            this.consumer.subscribe(topicName);
            ConsumerRecords<String, String> records;
            while (!closed.get()) {
                synchronized (consumer) {
                    records = consumer.poll(100);
                }
                for (ConsumerRecord<String, String> tmp : records) {
                    System.out.println(tmp.value());
                }
            }
        } catch (WakeupException e) {
            // Ignore exception if closing
            System.out.println(e);
            //if (!closed.get()) throw e;
        }
    }

    // Shutdown hook which can be called from a separate thread
    public void shutdown() {
        closed.set(true);
        consumer.wakeup();
    }

    public static void main(String[] args) {
        CloudKafkaConsumer kafkaConsumer = KafkaConsumerBuilder.builder()
                .withBootstrapServers("172.31.1.159:9092")
                .withGroupId("test")
                .build();
        ExecutorService executorService = Executors.newFixedThreadPool(5);
        executorService.execute(new KafkaConsumerRunner(kafkaConsumer, "log"));
        executorService.execute(new KafkaConsumerRunner(kafkaConsumer, "log.info"));
        executorService.shutdown();
    }
}

但它不起作用并引发异常:

  

java.util.ConcurrentModificationException:KafkaConsumer对多线程访问不安全

此外,我还阅读了Flink(一个用于分布式流和批处理数据处理的开源平台)的来源。使用多线程消费者的Flink与我的相似。

long pollTimeout = Long.parseLong(flinkKafkaConsumer.properties.getProperty(KEY_POLL_TIMEOUT, Long.toString(DEFAULT_POLL_TIMEOUT)));
pollLoop: while (running) {
    ConsumerRecords<byte[], byte[]> records;
    //noinspection SynchronizeOnNonFinalField
    synchronized (flinkKafkaConsumer.consumer) {
        try {
            records = flinkKafkaConsumer.consumer.poll(pollTimeout);
        } catch (WakeupException we) {
            if (running) {
                throw we;
            }
            // leave loop
            continue;
        }
    }

flink code of mutli-thread

出了什么问题?

3 个答案:

答案 0 :(得分:13)

Kafka消费者不是线程安全的。正如您在问题中指出的那样,该文件指出

  

一个简单的选择是为每个线程提供自己的消费者实例

但是在您的代码中,您拥有由不同KafkaConsumerRunner实例包装的相同的使用者实例。因此,多个线程正在访问同一个消费者实例。卡夫卡文件明确说明

  

Kafka消费者不是线程安全的。所有网络I / O都发生在   打电话的应用程序的线程。这是责任   用户确保多线程访问正确   同步。将导致未同步的访问   ConcurrentModificationException的。

这正是你收到的例外。

答案 1 :(得分:3)

它会在您订阅的电话上抛出异常。 this.consumer.subscribe(topicName);

将该块移动到同步块中,如下所示:

@Override
public void run() {
    try {
        synchronized (consumer) {
            this.consumer.subscribe(topicName);
        }
        ConsumerRecords<String, String> records;
        while (!closed.get()) {
            synchronized (consumer) {
                records = consumer.poll(100);
            }
            for (ConsumerRecord<String, String> tmp : records) {
                System.out.println(tmp.value());
            }
        }
    } catch (WakeupException e) {
        // Ignore exception if closing
        System.out.println(e);
        //if (!closed.get()) throw e;
    }
}

答案 2 :(得分:2)

可能不是你的情况,但如果你是对多个主题的数据进行合金处理,那么你可以从同一个消费者的多个主题中读取数据。如果没有,那么最好创建消耗每个主题的单独作业。