com.awilton.junit.kafka.embedded.EmbeddedProducer Maven / Gradle / Ivy

Go to download
package com.awilton.junit.kafka.embedded;

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

import kafka.javaapi.producer.Producer;
import kafka.producer.KeyedMessage;
import kafka.producer.ProducerConfig;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

public class EmbeddedProducer {
  private static Logger log = LoggerFactory.getLogger(EmbeddedProducer.class);
  
  private Producer producer;
  
  public EmbeddedProducer(int kafkaPort) {
    init(initProps(kafkaPort));
  }
  
  public EmbeddedProducer(Properties kafkaProps) {
    log.info("Initializing");
    init(kafkaProps);
  }
  
  public Producer getProducer() {
    return this.producer;
  }
  
  public void send(String topic, Y msg) {
    KeyedMessage kMsg = new KeyedMessage(topic,null,msg);
    this.sendMessage(kMsg);
  }
  
  public void send(List> messages) {
    this.producer.send(messages);
  }
  
  public void sendMessage(KeyedMessage msg) {
    try {
      this.producer.send(msg);
      log.debug(">>>>>>>>>>>>  Message sent");
    } catch(Exception e) {
      log.error("Failed to send",e);
      throw new RuntimeException(e);
    }
  }

  public void close() {
    if (null != producer) {
      producer.close();
    }
  }
  private void init(Properties p) {
    ProducerConfig pc = new ProducerConfig(p);
    this.producer = new Producer(pc);
    
  }
  
  private Properties initProps(int kfkaPort) {
    Properties p = new Properties();
    String kHost = "localhost:"+kfkaPort;
    p.put("metadata.broker.list",kHost);
    p.put("serializer.class", "kafka.serializer.StringEncoder");
    p.put("request.required.acks", "1");
    
    return p;
  }
}