fr.figarocms.flume.formatter.JsonObjectFormatter Maven / Gradle / Ivy

Go to download

package fr.figarocms.flume.formatter;

import com.cloudera.flume.core.Event;
import com.cloudera.flume.handlers.text.output.AbstractOutputFormat;
import fr.figarocms.flume.formatter.config.Formatter;
import org.codehaus.jackson.map.ObjectMapper;

import java.io.IOException;
import java.io.OutputStream;

public class JsonObjectFormatter extends AbstractOutputFormat {

  private ObjectMapper mapper;
  private Formatter formatter;

  public JsonObjectFormatter(Formatter formatter, ObjectMapper mapper) {
    this.formatter = formatter;
    this.mapper = mapper;
  }

  @Override
  public void format(OutputStream o, Event e) throws IOException {
    Object obj = formatter.format(e);
    mapper.writeValue(o, obj);
  }

}