io.quarkiverse.mailpit.deployment.JbossContainerLogConsumer Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of quarkus-mailpit-deployment Show documentation
There is a newer version: 1.3.0
Show newest version
package io.quarkiverse.mailpit.deployment;

import java.util.HashMap;
import java.util.Map;

import org.jboss.logging.Logger;
import org.jboss.logging.MDC;
import org.testcontainers.containers.output.BaseConsumer;
import org.testcontainers.containers.output.OutputFrame;

/**
 * A TestContainers consumer for container output that logs output to an JBoss logger.
 */
public class JbossContainerLogConsumer extends BaseConsumer {

    private final Logger logger;

    private final Map mdc = new HashMap<>();

    private boolean separateOutputStreams;

    private String prefix = "";

    public JbossContainerLogConsumer(Logger logger) {
        this(logger, false);
    }

    public JbossContainerLogConsumer(Logger logger, boolean separateOutputStreams) {
        this.logger = logger;
        this.separateOutputStreams = separateOutputStreams;
    }

    public JbossContainerLogConsumer withPrefix(String prefix) {
        this.prefix = "[" + prefix + "] ";
        return this;
    }

    public JbossContainerLogConsumer withMdc(String key, String value) {
        mdc.put(key, value);
        return this;
    }

    public JbossContainerLogConsumer withMdc(Map mdc) {
        this.mdc.putAll(mdc);
        return this;
    }

    public JbossContainerLogConsumer withSeparateOutputStreams() {
        this.separateOutputStreams = true;
        return this;
    }

    @Override
    public void accept(OutputFrame outputFrame) {
        final OutputFrame.OutputType outputType = outputFrame.getType();
        final String utf8String = outputFrame.getUtf8StringWithoutLineEnding();

        final Map originalMdc = MDC.getMap();
        MDC.clear();
        MDC.getMap().putAll(mdc);
        try {
            switch (outputType) {
                case END:
                    break;
                case STDOUT:
                    if (separateOutputStreams) {
                        logger.infof("%s%s", prefix.isEmpty() ? "" : (prefix + ": "), utf8String);
                    } else {
                        logger.infof("%s%s: %s", prefix, outputType, utf8String);
                    }
                    break;
                case STDERR:
                    if (separateOutputStreams) {
                        logger.errorf("%s%s", prefix.isEmpty() ? "" : (prefix + ": "), utf8String);
                    } else {
                        logger.infof("%s%s: %s", prefix, outputType, utf8String);
                    }
                    break;
                default:
                    throw new IllegalArgumentException("Unexpected outputType " + outputType);
            }
        } finally {
            MDC.clear();
            if (originalMdc != null) {
                MDC.getMap().putAll(originalMdc);
            }
        }
    }
}