com.github.danielwegener.logback.kafka.encoding.LayoutKafkaMessageEncoder Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of logback-kafka-appender Show documentation

There is a newer version: 0.2.0-RC2

Show newest version

package com.github.danielwegener.logback.kafka.encoding;

import ch.qos.logback.core.Layout;

import java.nio.charset.Charset;

/**
 * A KafkaMessageEncoder that can be configured with a {@link Layout} and a {@link Charset} and creates
 * a serialized string for each event using the given layout.
 * @since 0.1.0
 */
public class LayoutKafkaMessageEncoder extends KafkaMessageEncoderBase {

    public LayoutKafkaMessageEncoder() {
    }

    public LayoutKafkaMessageEncoder(Layout layout, Charset charset) {
        this.layout = layout;
        this.charset = charset;
    }

    private Layout layout;
    private Charset charset;
    private static final Charset UTF8 = Charset.forName("UTF-8");

    @Override
    public void start() {
        if (charset == null) {
            addInfo("No charset specified for PatternLayoutKafkaEncoder. Using default UTF8 encoding.");
            charset = UTF8;
        }
        super.start();
    }

    @Override
    public byte[] doEncode(E event) {
        final String message = layout.doLayout(event);
        return message.getBytes(charset);
    }

    public void setLayout(Layout layout) {
        this.layout = layout;
    }

    public void setCharset(Charset charset) {
        this.charset = charset;
    }

    public Layout getLayout() {
        return layout;
    }

    public Charset getCharset() {
        return charset;
    }
}