lv.ctco.cukesrest.internal.logging.LoggerPrintStream Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of cukes-rest Show documentation

The newest version!

package lv.ctco.cukesrest.internal.logging;

import org.apache.commons.lang3.StringUtils;
import org.slf4j.Logger;
import org.slf4j.event.Level;

import java.io.IOException;
import java.io.OutputStream;
import java.io.PrintStream;

public class LoggerPrintStream extends PrintStream {

    public LoggerPrintStream(Logger logger, Level level) {
        super(new LoggerOutputStream(logger, level), true);
    }

    private static class LoggerOutputStream extends OutputStream {

        private final Logger logger;
        private final Level level;
        private StringBuilder stringBuilder = new StringBuilder();

        LoggerOutputStream(Logger logger, Level level) {
            this.logger = logger;
            this.level = level;
        }

        @Override
        public void write(int b) throws IOException {
            stringBuilder.append((char) b);
        }

        @Override
        public void flush() {
            log(logger, level, stringBuilder.toString());
            stringBuilder = new StringBuilder();
        }

        private void log(Logger logger, Level level, String message) {

            if (StringUtils.isBlank(message)) return;

            switch (level) {
                case TRACE:
                    logger.trace(message);
                    break;
                case DEBUG:
                    logger.debug(message);
                    break;
                case INFO:
                    logger.info(message);
                    break;
                case WARN:
                    logger.warn(message);
                    break;
                case ERROR:
                    logger.error(message);
                    break;
            }
        }
    }

}