新的api完整实例如下:
[java] view plain copy package com.cuicui.kafkademon; import java.util.ArrayList; import java.util.List; import java.util.Properties; import kafka.javaapi.producer.Producer; import kafka.producer.KeyedMessage; import kafka.producer.ProducerConfig; /** * @author <a href="mailto:leicui001@126.com">崔磊</a> * @date 2015年11月4日 上午11:44:15 */ public class MyProducer { public static void main(String[] args) throws InterruptedException { Properties props = new Properties(); props.put("serializer.class", "kafka.serializer.StringEncoder"); props.put("metadata.broker.list", KafkaProperties.BROKER_CONNECT); props.put("partitioner.class", "com.cuicui.kafkademon.MyPartitioner"); props.put("request.required.acks", "1"); ProducerConfig config = new ProducerConfig(props); Producer<String, String> producer = new Producer<String, String>(config); // 单个发送 for (int i = 0; i <= 1000000; i++) { KeyedMessage<String, String> message = new KeyedMessage<String, String>(KafkaProperties.TOPIC, i + "", "Message" + i); producer.send(message); Thread.sleep(5000); } // 批量发送 List<KeyedMessage<String, String>> messages = new ArrayList<KeyedMessage<String, String>>(100); for (int i = 0; i <= 10000; i++) { KeyedMessage<String, String> message = new KeyedMessage<String, String>(KafkaProperties.TOPIC, i + "", "Message" + i); messages.add(message); if (i % 100 == 0) { producer.send(messages); messages.clear(); } } producer.send(messages); } } 下面这个是用到的分区函数:
[java] view plain copy import kafka.producer.Partitioner; import kafka.utils.VerifiableProperties; public class MyPartitioner implements Partitioner { public MyPartitioner(VerifiableProperties props) { } /* * @see kafka.producer.Partitioner#partition(java.lang.Object, int) */ @Override public int partition(Object key, int partitionCount) { return Integer.valueOf((String) key) % partitionCount; } }
高级别的API隐藏了和brokers连接的细节,在不必关心服务端架构的情况下和服务端通信。还可以自己维护消费状态,并可以通过一些条件指定订阅特定的topic,比如白名单黑名单或者正则表达式。
[java] view plain copy package com.cuicui.kafkademon; import java.nio.ByteBuffer; import java.util.Collections; import java.util.HashMap; import java.util.List; import java.util.Map; import kafka.api.FetchRequest; import kafka.api.FetchRequestBuilder; import kafka.api.PartitionOffsetRequestInfo; import kafka.cluster.Broker; import kafka.common.TopicAndPartition; import kafka.javaapi.FetchResponse; import kafka.javaapi.OffsetRequest; import kafka.javaapi.OffsetResponse; import kafka.javaapi.PartitionMetadata; import kafka.javaapi.TopicMetadata; import kafka.javaapi.TopicMetadataRequest; import kafka.javaapi.TopicMetadataResponse; import kafka.javaapi.consumer.SimpleConsumer; import kafka.javaapi.message.ByteBufferMessageSet; import kafka.message.Message; import kafka.message.MessageAndOffset; /** * offset自己维护 目标topic、partition均由自己分配 * * @author <a href="mailto:leicui001@126.com">崔磊</a> * @date 2015年11月4日 上午11:44:15 * */ public class MySimpleConsumer { public static void main(String[] args) { new MySimpleConsumer().consume(); } /** * 消费消息 */ public void consume() { int partition = 0; // 找到leader Broker leaderBroker = findLeader(KafkaProperties.BROKER_CONNECT, KafkaProperties.TOPIC, partition); // 从leader消费 SimpleConsumer simpleConsumer = new SimpleConsumer(leaderBroker.host(), leaderBroker.port(), 20000, 10000, "mySimpleConsumer"); long startOffet = 1; int fetchSize = 1000; while (true) { long offset = startOffet; // 添加fetch指定目标tipic,分区,起始offset及fetchSize(字节),可以添加多个fetch FetchRequest req = new FetchRequestBuilder().addFetch(KafkaProperties.TOPIC, 0, startOffet, fetchSize).build(); // 拉取消息 FetchResponse fetchResponse = simpleConsumer.fetch(req); ByteBufferMessageSet messageSet = fetchResponse.messageSet(KafkaProperties.TOPIC, partition); for (MessageAndOffset messageAndOffset : messageSet) { Message mess = messageAndOffset.message(); ByteBuffer payload = mess.payload(); byte[] bytes = new byte[payload.limit()]; payload.get(bytes); String msg = new String(bytes); offset = messageAndOffset.offset(); System.out.println("partition : " + 3 + ", offset : " + offset + " mess : " + msg); } // 继续消费下一批 startOffet = offset + 1; } } /** * 找到制定分区的leader broker * * @param brokerHosts broker地址,格式为:“host1:port1,host2:port2,host3:port3” * @param topic topic * @param partition 分区 * @return */ public Broker findLeader(String brokerHosts, String topic, int partition) { Broker leader = findPartitionMetadata(brokerHosts, topic, partition).leader(); System.out.println(String.format("Leader tor topic %s, partition %d is %s:%d", topic, partition, leader.host(), leader.port())); return leader; } /** * 找到指定分区的元数据 * * @param brokerHosts broker地址,格式为:“host1:port1,host2:port2,host3:port3” * @param topic topic * @param partition 分区 * @return 元数据 */ private PartitionMetadata findPartitionMetadata(String brokerHosts, String topic, int partition) { PartitionMetadata returnMetaData = null; for (String brokerHost : brokerHosts.split(",")) { SimpleConsumer consumer = null; String[] splits = brokerHost.split(":"); consumer = new SimpleConsumer(splits[0], Integer.valueOf(splits[1]), 100000, 64 * 1024, "leaderLookup"); List<String> topics = Collections.singletonList(topic); TopicMetadataRequest request = new TopicMetadataRequest(topics); TopicMetadataResponse response = consumer.send(request); List<TopicMetadata> topicMetadatas = response.topicsMetadata(); for (TopicMetadata topicMetadata : topicMetadatas) { for (PartitionMetadata PartitionMetadata : topicMetadata.partitionsMetadata()) { if (PartitionMetadata.partitionId() == partition) { returnMetaData = PartitionMetadata; } } } if (consumer != null) consumer.close(); } return returnMetaData; } /** * 根据时间戳找到某个客户端消费的offset * * @param consumer SimpleConsumer * @param topic topic * @param partition 分区 * @param clientID 客户端的ID * @param whichTime 时间戳 * @return offset */ public long getLastOffset(SimpleConsumer consumer, String topic, int partition, String clientID, long whichTime) { TopicAndPartition topicAndPartition = new TopicAndPartition(topic, partition); Map<TopicAndPartition, PartitionOffsetRequestInfo> requestInfo = new HashMap<TopicAndPartition, PartitionOffsetRequestInfo>(); requestInfo.put(topicAndPartition, new PartitionOffsetRequestInfo(whichTime, 1)); OffsetRequest request = new OffsetRequest(requestInfo, kafka.api.OffsetRequest.CurrentVersion(), clientID); OffsetResponse response = consumer.getOffsetsBefore(request); long[] offsets = response.offsets(topic, partition); return offsets[0]; } } 低级别的API是高级别API实现的基础,也是为了一些对维持消费状态有特殊需求的场景,比如Hadoop consumer这样的离线consumer。
每调用一次createMessageStreams都会将consumer注册到topic上,这样consumer和brokers之间的负载均衡就会进行调整。API鼓励每次调用创建更多的topic流以减少这种调整。createMessageStreamsByFilter方法注册监听可以感知新的符合filter的tipic。