pl.tkowalcz.tjahzi.log4j2.AppenderLogic 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.Logger;
import org.apache.logging.log4j.core.Layout;
import org.apache.logging.log4j.core.LogEvent;
import org.apache.logging.log4j.status.StatusLogger;
import pl.tkowalcz.tjahzi.LabelSerializer;
import pl.tkowalcz.tjahzi.LabelSerializers;
import pl.tkowalcz.tjahzi.LoggingSystem;
import pl.tkowalcz.tjahzi.TjahziLogger;
import pl.tkowalcz.tjahzi.log4j2.labels.LabelPrinter;

import java.io.Serializable;
import java.nio.ByteBuffer;
import java.util.Map;
import java.util.concurrent.TimeUnit;
import java.util.function.BiConsumer;

public class AppenderLogic implements BiConsumer {

    private static final Logger LOGGER = StatusLogger.getLogger();

    private final LoggingSystem loggingSystem;

    private final String logLevelLabel;
    private final TjahziLogger logger;

    private final Map lokiLabels;
    private final ByteBufferDestinationRepository destinationRepository;

    public AppenderLogic(
            LoggingSystem loggingSystem,
            String logLevelLabel,
            Map lokiLabels,
            int maxLogLineSizeBytes
    ) {
        this.loggingSystem = loggingSystem;
        this.logLevelLabel = logLevelLabel;

        this.logger = loggingSystem.createLogger();
        this.lokiLabels = lokiLabels;
        this.destinationRepository = new ByteBufferDestinationRepository(maxLogLineSizeBytes);
    }

    public void append(
            Layout layout,
            LogEvent event
    ) {
        TjahziByteBufferDestination destination = destinationRepository.threadLocalDestination(
                this,
                event
        );

        layout.encode(event, destination);
        destination.drainRemaining();
    }

    public void close(long timeout, TimeUnit timeUnit) {
        loggingSystem.close(
                (int) timeUnit.toMillis(timeout),
                thread -> LOGGER.error("Loki appender was unable to stop thread on shutdown: " + thread)
        );
    }

    @Override
    public void accept(LogEvent event, ByteBuffer byteBuffer) {
        LabelSerializer labelSerializer = LabelSerializers.threadLocal();
        processDynamicLabelsIfAny(labelSerializer, event);

        logger.log(
                event.getTimeMillis(),
                labelSerializer,
                byteBuffer
        );
    }

    private void processDynamicLabelsIfAny(LabelSerializer labelSerializer, LogEvent event) {
        if (logLevelLabel != null) {
            labelSerializer.appendLabel(logLevelLabel, event.getLevel().toString());
        }

        for (Map.Entry labelPrinter : lokiLabels.entrySet()) {
            labelSerializer.appendLabelName(labelPrinter.getKey());

            labelSerializer.startAppendingLabelValue();
            labelPrinter.getValue().append(event, labelSerializer::appendPartialLabelValue);
            labelSerializer.finishAppendingLabelValue();
        }
    }
}