【Kafka十三】Kafka Simple Consumer
编程技术 / houtizong 发布于 3年前 61
代码中关于Host和Port是割裂开的，这会导致单机环境下的伪分布式Kafka集群环境下，这个例子没法运行。
实际情况是需要将host和port绑定到一起，
package kafka.examples.lowlevel;import kafka.api.FetchRequest;import kafka.api.FetchRequestBuilder;import kafka.api.PartitionOffsetRequestInfo;import kafka.common.ErrorMapping;import kafka.common.TopicAndPartition;import kafka.javaapi.*;import kafka.javaapi.consumer.SimpleConsumer;import kafka.message.MessageAndOffset;import java.nio.ByteBuffer;import java.util.*;public class KafkaLowLevelConsumer { //参数说明： /* Maximum number of messages to read (so we don’t loop forever) Topic to read from Partition to read from One broker to use for Metadata lookup Port the brokers listen on */ public static void main(String args[]) { KafkaLowLevelConsumer consumer = new KafkaLowLevelConsumer(); //读取的消息数 long maxReads = Long.parseLong(args[0]); //读取的topic名称 String topic = args[1]; //读取的partition，从0开始的 int partition = Integer.parseInt(args[2]); List<String> seeds = new ArrayList<String>(); seeds.add(args[3]); //seed broker的监听端口，每个Topic和Partition的信息是存放于zk目录：/brokers/topics/learn.topic.p8.r2 int port = Integer.parseInt(args[4]); try { consumer.run(maxReads, topic, partition, seeds, port); } catch (Exception e) { System.out.println("Oops:" + e); e.printStackTrace(); } } private List<String> replicaBrokers = new ArrayList<String>(); public KafkaLowLevelConsumer() { replicaBrokers = new ArrayList<String>(); } public void run(long maxReads, String topic, int partition, List<String> seedBrokers, int port) throws Exception { //获取指定topic和partition的元信息，PartitionMetadata的leader和replicas方法返回leader和replicas brokers PartitionMetadata metadata = findLeader(seedBrokers, port, topic, partition); if (metadata == null) { System.out.println("Can't find metadata for Topic and Partition. Exiting"); return; } //获取lead partition所在的broker if (metadata.leader() == null) { System.out.println("Can't find Leader for Topic and Partition. Exiting"); return; } //获取leader broker的host信息，不包括端口信息 String leadBroker = metadata.leader().host(); String clientName = "Client_" + topic + "_" + partition; //构造SimpleConsumer，为什么port和leaderBroker不一致？ //这里的leadBroker, port是配对的，应该是metadata.leader().port() SimpleConsumer consumer = new SimpleConsumer(leadBroker, port, 100000, 64 * 1024, clientName); //获取读取的offset long readOffset = getLastOffset(consumer, topic, partition, kafka.api.OffsetRequest.EarliestTime(), clientName); int numErrors = 0; while (maxReads > 0) { if (consumer == null) { consumer = new SimpleConsumer(leadBroker, port, 100000, 64 * 1024, clientName); } FetchRequest req = new FetchRequestBuilder() .clientId(clientName) .addFetch(topic, partition, readOffset, 100000) // Note: this fetchSize of 100000 might need to be increased if large batches are written to Kafka .build(); FetchResponse fetchResponse = consumer.fetch(req); //Since the SimpleConsumer doesn't handle lead Broker failures, you have to write a bit of code to handle it. if (fetchResponse.hasError()) { numErrors++; // Something went wrong! short code = fetchResponse.errorCode(topic, partition); System.out.println("Error fetching data from the Broker:" + leadBroker + " Reason: " + code); if (numErrors > 5) break; if (code == ErrorMapping.OffsetOutOfRangeCode()) { // We asked for an invalid offset. For simple case ask for the last element to reset readOffset = getLastOffset(consumer, topic, partition, kafka.api.OffsetRequest.LatestTime(), clientName); continue; } consumer.close(); consumer = null; leadBroker = findNewLeader(leadBroker, topic, partition, port); continue; } numErrors = 0; long numRead = 0; for (MessageAndOffset messageAndOffset : fetchResponse.messageSet(topic, partition)) { long currentOffset = messageAndOffset.offset(); if (currentOffset < readOffset) { System.out.println("Found an old offset: " + currentOffset + " Expecting: " + readOffset); continue; } readOffset = messageAndOffset.nextOffset(); ByteBuffer payload = messageAndOffset.message().payload(); byte[] bytes = new byte[payload.limit()]; payload.get(bytes); System.out.println(String.valueOf(messageAndOffset.offset()) + ": " + new String(bytes, "UTF-8")); numRead++; maxReads--; } //This method uses the findLeader() logic we defined earlier to find the new leader, // except here we only try to connect to one of the replicas for the topic/partition. // This way if we can’t reach any of the Brokers with the data we are interested in we give up and exit hard. //Since it may take a short time for ZooKeeper to detect the leader loss and assign a new leader, we sleep if we don’t get an answer. // In reality ZooKeeper often does the failover very quickly so you never sleep if (numRead == 0) { try { Thread.sleep(1000); } catch (InterruptedException ie) { } } } if (consumer != null) consumer.close(); } // Finding Starting Offset for Reads // Now define where to start reading data. Kafka includes two constants to help, // kafka.api.OffsetRequest.EarliestTime() finds the beginning of the data in the logs and starts streaming from there, // kafka.api.OffsetRequest.LatestTime() will only stream new messages. // Don’t assume that offset 0 is the beginning offset, since messages age out of the log over time. public static long getLastOffset(SimpleConsumer consumer, String topic, int partition, long whichTime, String clientName) { TopicAndPartition topicAndPartition = new TopicAndPartition(topic, partition); Map<TopicAndPartition, PartitionOffsetRequestInfo> requestInfo = new HashMap<TopicAndPartition, PartitionOffsetRequestInfo>(); requestInfo.put(topicAndPartition, new PartitionOffsetRequestInfo(whichTime, 1)); kafka.javaapi.OffsetRequest request = new kafka.javaapi.OffsetRequest( requestInfo, kafka.api.OffsetRequest.CurrentVersion(), clientName); //Get a list of valid offsets (up to maxSize) before the given time. OffsetResponse response = consumer.getOffsetsBefore(request); if (response.hasError()) { System.out.println("Error fetching data Offset Data the Broker. Reason: " + response.errorCode(topic, partition)); return 0; } long[] offsets = response.offsets(topic, partition); return offsets[0]; } //Since the SimpleConsumer doesn't handle lead Broker failures, you have to write a bit of code to handle it. //Here, once the fetch returns an error, we log the reason, close the consumer then try to figure out who the new leader is. private String findNewLeader(String oldLeader, String topic, int partition, int port) throws Exception { for (int i = 0; i < 3; i++) { boolean goToSleep; PartitionMetadata metadata = findLeader(replicaBrokers, port, topic, partition); if (metadata == null) { goToSleep = true; } else if (metadata.leader() == null) { goToSleep = true; } else if (oldLeader.equalsIgnoreCase(metadata.leader().host()) && i == 0) { // first time through if the leader hasn't changed give ZooKeeper a second to recover // second time, assume the broker did recover before failover, or it was a non-Broker issue // goToSleep = true; } else { return metadata.leader().host(); } if (goToSleep) { try { Thread.sleep(1000); } catch (InterruptedException ie) { } } } System.out.println("Unable to find new leader after Broker failure. Exiting"); throw new Exception("Unable to find new leader after Broker failure. Exiting"); } ///根据指定的Broker，查找指定topic和partition的Lead Partition //Finding the Lead Broker for a Topic and Partition //The easiest way to do this is to pass in a set of known Brokers to your logic, // either via a properties file or the command line. // These don’t have to be all the Brokers in the cluster, // rather just a set where you can start looking for a live Broker to query for Leader information. //seedBrokers是使用的replicaBrokers列表 //调用PartitionMetadata的leader和replicas方法可以得到该Partition对应的Leader和Replicas Broker信息 private PartitionMetadata findLeader(List<String> seedBrokers, int port, String topic, int partition) { PartitionMetadata returnMetaData = null; loop: for (String seed : seedBrokers) { SimpleConsumer consumer = null; try { /** class SimpleConsumer(val host: String, val port: Int, val soTimeout: Int, val bufferSize: Int, val clientId: String) */ consumer = new SimpleConsumer(seed, port, 100000, 64 * 1024, "leaderLookup"); List<String> topics = Collections.singletonList(topic); TopicMetadataRequest req = new TopicMetadataRequest(topics); kafka.javaapi.TopicMetadataResponse resp = consumer.send(req); //The call to topicsMetadata() asks the Broker you are connected to for all the details about the topic we are interested in List<TopicMetadata> metaData = resp.topicsMetadata(); for (TopicMetadata item : metaData) { //The loop on partitionsMetadata iterates through all the partitions until we find the one we want. Once we find it, we can break out of all the loops. for (PartitionMetadata part : item.partitionsMetadata()) { if (part.partitionId() == partition) { returnMetaData = part; break loop; } } } } catch (Exception e) { System.out.println("Error communicating with Broker [" + seed + "] to find Leader for [" + topic + ", " + partition + "] Reason: " + e); } finally { if (consumer != null) consumer.close(); } } if (returnMetaData != null) { replicaBrokers.clear(); ///将replicaBrokers进行缓存 for (kafka.cluster.Broker replica : returnMetaData.replicas()) { replicaBrokers.add(replica.host()); } } return returnMetaData; }} 参考：https://cwiki.apache.org/confluence/display/KAFKA/0.8.0+SimpleConsumer+Example
请勿发布不友善或者负能量的内容。与人为善，比聪明更重要！
技术博客集 - 网站简介：
前后端技术：
后端基于Hyperf2.1框架开发,前端使用Bootstrap可视化布局系统生成
网站主要作用：
1.编程技术分享及讨论交流，内置聊天系统;
2.测试交流框架问题，比如：Hyperf、Laravel、TP、beego;
3.本站数据是基于大数据采集等爬虫技术为基础助力分享知识，如有侵权请发邮件到站长邮箱，站长会尽快处理;
4.站长邮箱：[email protected];
文章归档
文章标签
友情链接
