Kafka的多任务线程消费测试

╰+哭是因爲堅強的太久メ 2022-05-10 15:08 141阅读 0赞

## 代码: ##

package com.weichai.kafka;
    
    import kafka.consumer.ConsumerIterator;
    import kafka.consumer.KafkaStream;
    
    /**
     * 消费者的线程执行
     * @author lhy
     * @date 2018.10.09
     */
    public class Consumer implements Runnable {
    
    	private KafkaStream stream;
    	private int threadNumber;
    	
    	public Consumer(KafkaStream stream, int threadNumber) {
    		this.stream = stream;
    		this.threadNumber = threadNumber;
    	}
    
        /**
         * 线程执行
         */
    	@Override
    	public void run() {
    		// TODO Auto-generated method stub
    		ConsumerIterator<byte[], byte[]> it = stream.iterator();
    		while(it.hasNext()){
    			System.out.println("Thread " + threadNumber + ": " + new String(it.next().message()));
    		}
    		System.out.println("Shutting down Thread: " + threadNumber);
    	}
    
    }

package com.weichai.kafka;
    
    import java.util.HashMap;
    import java.util.List;
    import java.util.Map;
    import java.util.Properties;
    import java.util.concurrent.ExecutorService;
    import java.util.concurrent.Executors;
    import java.util.concurrent.TimeUnit;
    
    import kafka.consumer.ConsumerConfig;
    import kafka.consumer.KafkaStream;
    import kafka.javaapi.consumer.ConsumerConnector;
    
    /**
     * Kafka消费者的多线程调用,提高吞吐效率
     * @author lhy
     * @date 2018.10.09
     */
    public class ConsumerThread {
    
    	private final ConsumerConnector consumer;
    	private final String topic;
    	private ExecutorService executor;
    	
    	public ConsumerThread(String a_zookeeper, String a_groupId, String a_topic) {
    		consumer = kafka.consumer.Consumer.createJavaConsumerConnector(createConsumerConfig(a_zookeeper, a_groupId));
    		this.topic = a_topic;
    	}
    	
    	public void shutdown() {
    		if(consumer !=null){
    			consumer.shutdown();
    		}
    		if(executor !=null){
    			executor.shutdown();
    		}
    		try {
    			if(!executor.awaitTermination(5000, TimeUnit.MILLISECONDS)){
    				System.out.println("消费者线程等待超时,直接退出!");
    			}
    		} catch (InterruptedException e) {
    			// TODO Auto-generated catch block
    			System.out.println("系统异常中断，直接退出!");
    			e.printStackTrace();
    		}
    	}
    	
    	public void run(int a_numThreads) {
    		Map<String, Integer> topicCountMap = new HashMap<String, Integer>();
    		topicCountMap.put(topic, new Integer(a_numThreads));
    		Map<String, List<KafkaStream<byte[], byte[]>>> consumerMap = consumer.createMessageStreams(topicCountMap);
    		List<KafkaStream<byte[], byte[]>> streams = consumerMap.get(topic);
            
    		executor = Executors.newFixedThreadPool(a_numThreads);
    		
    		int threadNumber = 0;
    		for (final KafkaStream stream : streams) {
    			executor.submit(new Consumer(stream, threadNumber));
    			threadNumber++;
    		}
    	}
    	private ConsumerConfig createConsumerConfig(String a_zookeeper, String a_groupId) {
    		// TODO Auto-generated method stub
    		Properties props = new Properties();
    		props.put("zookeeper.connect", a_zookeeper);
    		props.put("group.id", a_groupId);
    		props.put("zookeeper.session.timeout.ms", "400");
    		props.put("zookeeper.sync.time.ms", "200");
    		props.put("auto.commit.interval.ms", "1000");
    		
    		return new ConsumerConfig(props);
    	}
    	
    	public static void main(String[] args) {
    		// TODO Auto-generated method stub
    		String zooKeeper = "localhost:2181";
    		String groupId = "0";
    		String topic = "SimpleNode";
    		int threads = 5; // 启动的线程数
    		
    		ConsumerThread thread = new ConsumerThread(zooKeeper, groupId, topic);
    		thread.run(threads);
    		try {
    			Thread.sleep(5000);      //线程休眠5秒后终止
    		} catch (InterruptedException e) {
    			// TODO Auto-generated catch block
    			e.printStackTrace();
    		}
    		thread.shutdown();
    	}
    }

### 运行截图: ###

![70][]

### 此处重温了并发编程的内容,使用到了多线程框架,开启5个线程吸收Topic,大大提高了Kafka消费Topic的效率. ###

[70]: /images/20220505/b0146e279c534e28968d2700be3a1cb6.png