|
package com.paile.kafka.service.impl;
import java.util.Properties;
import java.util.Random;
import kafka.javaapi.producer.Producer;
import kafka.producer.KeyedMessage;
import kafka.producer.ProducerConfig;
import org.apache.log4j.Logger;
import com.paile.kafka.bean.MessageBean;
import com.paile.kafka.service.IKafkaService;
import com.paile.utils.others.Const;
/***
* kafka消息服务类
*
* @author libo
*
*/
public class KafkaServiceImpl implements IKafkaService {
private Logger logger = Logger.getLogger(KafkaServiceImpl.class);
/***
* 发送一条消息
*/
public void sendSinglePartitionMessage(String broke_list, String topic, Object message)
throws Exception {
Producer<Integer, Object> producer = null;
try {
Properties props = new Properties();
props.put("metadata.broker.list", broke_list);
props.put("serializer.class", "com.paile.kafka.CustomEncoder");
props.put("key.serializer.class", "kafka.serializer.StringEncoder");
props.put("producer.type", "sync");//是否同步 sync:同步 async:异步
props.put("request.required.acks", "1");
producer = new Producer<Integer, Object>(new ProducerConfig(props));
KeyedMessage<Integer, Object> data = new KeyedMessage<Integer, Object>(topic, Const.defaultPartitionKey, message);
producer.send(data);
} catch (Exception e) {
logger.error("发送消息到Kafka失败,", e);
throw e;
}finally{
if(producer!=null)
producer.close();
}
}
/***
* 发送多分区消息
*/
public void sendMutilPartitionMessage(String broke_list,
String topic,MessageBean message)
throws Exception {
Producer<String, MessageBean> producer = null;
try {
Properties props = new Properties();
props.put("metadata.broker.list", broke_list);
props.put("serializer.class", "com.paile.kafka.productor.CustomEncoder");
props.put("key.serializer.class", "kafka.serializer.StringEncoder");
props.put("producer.type", "async");//是否同步 sync:同步 async:异步
props.put("partitioner.class", "com.paile.kafka.productor.SimplePartitioner");//分区算法类
props.put("request.required.acks", "1");
producer = new Producer<String, MessageBean>(new ProducerConfig(props));
Random random = new Random();
int partitionKey = random.nextInt(255);
KeyedMessage<String, MessageBean> data = new KeyedMessage<String, MessageBean>(topic,
String.valueOf(partitionKey), message);
producer.send(data);
} catch (Throwable e) {
logger.error("发送消息到Kafka失败,", e);
System.out.println(e.getMessage());
throw new Exception(e.getMessage());
}finally{
if(producer!=null)
producer.close();
}
}
/***
* 接收消息
*/
@Override
public void startConsumer(String zookeeperConnect, String groupId, String topic,int threads)
throws Exception {
GroupConsumerManager example = new GroupConsumerManager(zookeeperConnect, groupId, topic);
try {
example.run(threads);
} catch (Exception e) {
e.printStackTrace();
}
try {
Thread.sleep(10000);
} catch (InterruptedException ie) {
}
example.shutdown();
}
public static void main(String[] args){
KafkaServiceImpl service = new KafkaServiceImpl();
try {
for(int i=0;i<10;i++){
MessageBean bean = new MessageBean();
bean.setId("00"+i);
bean.setData("111111111111111");
bean.setImg(new byte[0]);
service.sendMutilPartitionMessage("192.168.1.101:9092", "paile01",bean);
}
System.out.println("");
} catch (Exception e) {
e.printStackTrace();
}
}
} |
|
|