com.blazemeter.taurus.grinder.TaurusAppender Maven / Gradle / Ivy

Go to download
package com.blazemeter.taurus.grinder;

import ch.qos.logback.classic.spi.LoggingEvent;
import ch.qos.logback.core.FileAppender;
import net.grinder.engine.process.TestRegistryAccessor;

import java.io.IOException;

public class TaurusAppender extends FileAppender {

    private static boolean isInit = false;

    @Override
    protected void writeOut(Object event) throws IOException {
        if (event instanceof LoggingEvent) {
            LoggingEvent proto = (LoggingEvent) event;
            if (!isInit) {
                TestRegistryAccessor.changeTestStatisticsMap();
                isInit = true;
            }
//            for (Test t : TestRegistryAccessor.getNewTests()) {
//                LoggingEvent le = new LoggingEvent();
//                le.setLoggerName(proto.getLoggerName());
//                le.setMessage("Test name for ID " + t.getNumber() + ": " + t.getDescription());
//                le.setLevel(Level.INFO);
//                super.writeOut(le);
//            }

            LoggingEvent wrapped = new LoggingEvent();
            if (proto.getLoggerName().equals("data")) {
                wrapped.setLoggerName(proto.getLoggerName() + "." + proto.getLoggerContextVO().getPropertyMap().get("WORKER_NAME"));
            } else {
                wrapped.setLoggerName(proto.getLoggerName());
            }
            wrapped.setMarker(proto.getMarker());
            wrapped.setMessage(proto.getMessage());
            wrapped.setLevel(proto.getLevel());
            wrapped.setArgumentArray(proto.getArgumentArray());
            super.writeOut(wrapped);

        } else {
            super.writeOut(event);
        }
    }
}