pl.tkowalcz.tjahzi.log4j2.TjahziByteBufferDestination Maven / Gradle / Ivy

Go to download

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

There is a newer version: 0.9.32

package pl.tkowalcz.tjahzi.log4j2;

import org.apache.logging.log4j.core.LogEvent;
import org.apache.logging.log4j.core.layout.ByteBufferDestination;
import org.apache.logging.log4j.core.layout.ByteBufferDestinationHelper;

import java.nio.ByteBuffer;
import java.util.function.BiConsumer;

public class TjahziByteBufferDestination implements ByteBufferDestination {

    private final ByteBuffer buffer;

    private BiConsumer drain;
    private LogEvent context;

    public TjahziByteBufferDestination(int maxLogLineSizeBytes) {
        buffer = ByteBuffer.allocate(maxLogLineSizeBytes);
    }

    public void drainRemaining() {
        if (buffer.hasRemaining()) {
            drain(buffer);
        }
    }

    @Override
    public ByteBuffer getByteBuffer() {
        return buffer;
    }

    @Override
    public ByteBuffer drain(ByteBuffer buf) {
        buf.flip();
        drain.accept(context, buf);

        buf.clear();
        return buf;
    }

    @Override
    public void writeBytes(ByteBuffer data) {
        ByteBufferDestinationHelper.writeToUnsynchronized(data, this);
    }

    @Override
    public void writeBytes(byte[] data, int offset, int length) {
        ByteBufferDestinationHelper.writeToUnsynchronized(data, offset, length, this);
    }

    public void initialize(BiConsumer drain, LogEvent context) {
        this.context = context;
        this.drain = drain;

        buffer.clear();
    }
}