com.jeesuite.kafka.producer.ConsumerAckWatcher Maven / Gradle / Ivy

Go to download
package com.jeesuite.kafka.producer;

import java.util.Objects;
import java.util.concurrent.CountDownLatch;
import java.util.concurrent.TimeUnit;

import org.I0Itec.zkclient.IZkDataListener;
import org.I0Itec.zkclient.ZkClient;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import com.jeesuite.kafka.KafkaConst;

/**
 * 消费回执观察者
 * @description 

 * @author vakin
 * @date 2017年11月27日
 */
public class ConsumerAckWatcher {

	private static final Logger log = LoggerFactory.getLogger(ConsumerAckWatcher.class);

	private CountDownLatch latch;
	private String consumerGroup;
	
	public ConsumerAckWatcher(String messageId, ZkClient zkClient) {
		this.latch = new CountDownLatch(1);
		
		String path = KafkaConst.ZK_PRODUCER_ACK_PATH + messageId;
		zkClient.createEphemeral(path);
		
		zkClient.subscribeDataChanges(path, new IZkDataListener() {
			@Override
			public void handleDataDeleted(String dataPath) throws Exception {}
			@Override
			public void handleDataChange(String dataPath, Object data) throws Exception {
				latch.countDown();
				consumerGroup = Objects.toString(data);
				log.debug("recv_consumer_ack messageId:{}，from group:{}",messageId,consumerGroup);
				try {zkClient.delete(dataPath);} catch (Exception e) {}
			}
		});
	}
	
	
	/**
	 * 等待应答
	 * @return 返回消费者groupName
	 */
	public String waitAck(){
		try {
			this.latch.await(5000, TimeUnit.MILLISECONDS);
			return consumerGroup;
		} catch (Exception e) {} 
		return null;
	}
	
}