com.sun.tools.xjc.addon.xew.LoggingOutputStream.java Source code

Introduction

Here is the source code for com.sun.tools.xjc.addon.xew.LoggingOutputStream.java

Source

/*
 * LoggingOutputStream.java
 * 
 * Copyright (C) 2009, Dmitry Katsubo
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 3 of the License, or (at your option) any later version.
 *
 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
 * MA 02110-1301  USA
 */
package com.sun.tools.xjc.addon.xew;

import java.io.OutputStream;

import org.apache.commons.logging.Log;

/**
 * Class will redirect everything printed to this {@link OutputStream} to logger.
 * 
 * @author Dmitry Katsubo
 */
class LoggingOutputStream extends OutputStream {

    public enum LogLevel {
        TRACE, DEBUG, INFO, WARN, ERROR, FATAL
    }

    private final StringBuilder sb = new StringBuilder();

    private final Log logger;
    private final LoggingOutputStream.LogLevel logLevel;
    private final String messagePrefix;

    public LoggingOutputStream(Log logger, LoggingOutputStream.LogLevel logLevel) {
        this(logger, logLevel, null);
    }

    public LoggingOutputStream(Log logger, LoggingOutputStream.LogLevel logLevel, String messagePrefix) {
        this.logger = logger;
        this.logLevel = logLevel;
        this.messagePrefix = messagePrefix;
    }

    @Override
    public void write(byte[] buf, int off, int len) {
        for (int i = 0; i < len; i++) {
            write(buf[off + i]);
        }
    }

    @Override
    public void write(int b) {
        // Scan all input bytes and log a message on newline:
        switch (b) {
        case '\n':
            logMessage();

        case '\r':
            break;

        default:
            sb.append((char) b);
        }
    }

    @Override
    public void close() {
        if (sb.length() > 0) {
            logMessage();
        }
    }

    private void logMessage() {
        if (messagePrefix != null) {
            sb.insert(0, messagePrefix);
        }

        String message = sb.toString();

        switch (logLevel) {
        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;

        case FATAL:
            logger.fatal(message);
            break;
        }

        sb.setLength(0);
    }
}