SpringBoot Kafka工具类封装

bootstrap需要自己配置一下，其他的直接用就可以。 

package com.oal.microservice.util;

import com.alibaba.fastjson.JSONObject;
import com.oal.microservice.config.Bootstrap;
import org.apache.kafka.clients.producer.KafkaProducer;
import org.apache.kafka.clients.producer.ProducerRecord;
import org.springframework.beans.factory.annotation.Autowired;

import java.util.List;
import java.util.Map;
import java.util.Properties;

public class KafkaProducerUtils {

    @Autowired
    private Bootstrap bootstrap;

    /**
     *
     * 私有静态方法，创建Kafka生产者
     *
     * @author IG
     * @Date 2017年4月14日 上午10:32:32
     * @version 1.0.0
     * @return KafkaProducer
     */
    private KafkaProducer createProducer() {
        Properties props = new Properties();
        props.put("bootstrap.servers", bootstrap.getServers());
        props.put("acks", "all");
        props.put("retries", 0);
        props.put("batch.size", 0);
        props.put("linger.ms", 1);
        props.put("buffer.memory", 33554432);
        props.put("key.serializer", "org.apache.kafka.common.serialization.StringSerializer");
        props.put("value.serializer", "org.apache.kafka.common.serialization.StringSerializer");
        return new KafkaProducer((props));
    }

    /**
     *
     * 传入kafka约定的topicName,json格式字符串，发送给kafka集群
     *
     * @author IG
     * @Date 2017年4月14日 下午1:29:09
     * @version 1.0.0
     * @param topicName
     * @param jsonMessage
     */
    public void sendMessage(String topicName, String jsonMessage) {
        KafkaProducer producer = createProducer();
        producer.send(new ProducerRecord<>(topicName, jsonMessage));
        producer.close();
    }

    /**
     *
     * 传入kafka约定的topicName,json格式字符串数组，发送给kafka集群

     * 用于批量发送消息，性能较高。
     *
     * @author IG
     * @Date 2017年4月14日 下午2:00:12
     * @version 1.0.0
     * @param topicName
     * @param jsonMessages
     * @throws InterruptedException
     */
    public void sendMessage(String topicName, String... jsonMessages) {
        KafkaProducer producer = createProducer();
        for (String jsonMessage : jsonMessages) {
            producer.send(new ProducerRecord<>(topicName, jsonMessage));
        }
        producer.close();
    }

    /**
     *
     * 传入kafka约定的topicName,Map集合，内部转为json发送给kafka集群 

     * 用于批量发送消息，性能较高。
     *
     * @author IG
     * @Date 2017年4月14日 下午2:01:18
     * @version 1.0.0
     * @param topicName
     * @param mapMessageToJSONForArray
     */
    public void sendMessage(String topicName, List> mapMessageToJSONForArray) {
        KafkaProducer producer = createProducer();
        for (Map mapMessageToJSON : mapMessageToJSONForArray) {
            String array = JSONObject.toJSONString(mapMessageToJSON);
            producer.send(new ProducerRecord<>(topicName, array));
        }
        producer.close();
    }

    /**
     *
     * 传入kafka约定的topicName,Map，内部转为json发送给kafka集群
     *
     * @author IG
     * @Date 2017年4月14日 下午1:30:10
     * @version 1.0.0
     * @param topicName
     * @param mapMessageToJSON
     */
    public void sendMessage(String topicName, Map mapMessageToJSON) {
        KafkaProducer producer = createProducer();
        String array = JSONObject.toJSONString(mapMessageToJSON);
        producer.send(new ProducerRecord<>(topicName, array));
        producer.close();
    }
}