container/openejb-core/src/main/java/org/apache/openejb/util/JuliLogStream.java

/*
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You under the Apache License, Version 2.0
 * (the "License"); you may not use this file except in compliance with
 * the License.  You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package org.apache.openejb.util;

import org.apache.openejb.log.LoggerCreator;

import java.util.concurrent.atomic.AtomicBoolean;
import java.util.logging.Level;
import java.util.logging.LogRecord;
import java.util.logging.Logger;

public class JuliLogStream implements LogStream {
    protected final LoggerCreator logger;
    protected final AtomicBoolean debug = new AtomicBoolean(false);
    protected final AtomicBoolean info = new AtomicBoolean(false);

    public JuliLogStream(final LogCategory logCategory) {
        logger = new LoggerCreator(logCategory.getName());
    }

    public boolean isFatalEnabled() {
        return LoggerCreator.Get.exec(logger, debug, info).isLoggable(Level.SEVERE);
    }

    public void fatal(final String message) {
        log(Level.SEVERE, message, null);
    }

    public void fatal(final String message, final Throwable t) {
        log(Level.SEVERE, message, t);
    }

    public boolean isErrorEnabled() {
        return LoggerCreator.Get.exec(logger, debug, info).isLoggable(Level.SEVERE);
    }

    public void error(final String message) {
        log(Level.SEVERE, message, null);
    }

    public void error(final String message, final Throwable t) {
        log(Level.SEVERE, message, t);
    }

    public boolean isWarnEnabled() {
        return LoggerCreator.Get.exec(logger, debug, info).isLoggable(Level.WARNING);
    }

    public void warn(final String message) {
        log(Level.WARNING, message, null);
    }

    public void warn(final String message, final Throwable t) {
        log(Level.WARNING, message, t);
    }

    public boolean isInfoEnabled() {
        LoggerCreator.Get.exec(logger, debug, info);
        return info.get();
    }

    public void info(final String message) {
        log(Level.INFO, message, null);
    }

    public void info(final String message, final Throwable t) {
        log(Level.INFO, message, t);
    }

    public boolean isDebugEnabled() {
        LoggerCreator.Get.levels(logger, debug, info); // cost nothing is init is done
        return debug.get();
    }

    public void debug(final String message) {
        log(Level.FINE, message, null);
    }

    public void debug(final String message, final Throwable t) {
        log(Level.FINE, message, t);
    }

    private void log(final Level level, final String message, final Throwable t) {
        final Logger log = LoggerCreator.Get.exec(logger, debug, info);
        if (log.isLoggable(level)) {
            final LogRecord logRecord = new OpenEJBLogRecord(level, message);
            if (t != null) {
                logRecord.setThrown(t);
            }
            logRecord.setLoggerName(log.getName());
            log.log(logRecord);
        }
    }

    public static class OpenEJBLogRecord extends LogRecord {
        private static final String LOGGER = org.apache.openejb.util.Logger.class.getName();
        private static final String ASYNC_LOG_NAME = "org.apache.openejb.util.LogStreamAsync";
        private static final String ASYNC_LOG_THREAD = ASYNC_LOG_NAME + "$Consumer";

        /**
         * The name of the class that issued the logging call.
         *
         * @serial
         */
        private String sourceClassName;

        /**
         * The name of the method that issued the logging call.
         *
         * @serial
         */
        private String sourceMethodName;

        // If the source method and source class has been inited
        private transient boolean sourceInited;

        public OpenEJBLogRecord(final Level level, final String message) {
            super(level, message);
            sourceInited = false;
        }

        /**
         * Gets the name of the class that issued the logging call.
         *
         * @return the name of the class that issued the logging call
         */
        public String getSourceClassName() {
            initSource();
            return sourceClassName;
        }

        /**
         * Sets the name of the class that issued the logging call.
         *
         * @param sourceClassName the name of the class that issued the logging call
         */
        public void setSourceClassName(final String sourceClassName) {
            sourceInited = true;
            this.sourceClassName = sourceClassName;
        }

        /**
         * Gets the name of the method that issued the logging call.
         *
         * @return the name of the method that issued the logging call
         */
        public String getSourceMethodName() {
            initSource();
            return sourceMethodName;
        }

        /**
         * Sets the name of the method that issued the logging call.
         *
         * @param sourceMethodName the name of the method that issued the logging call
         */
        public void setSourceMethodName(final String sourceMethodName) {
            sourceInited = true;
            this.sourceMethodName = sourceMethodName;
        }

        /**
         * Init the sourceClass and sourceMethod fields.
         */
        private void initSource() {
            if (!this.sourceInited) {
                // search back up the stack for the first use of the OpenEJB Logger
                final StackTraceElement[] elements = new Throwable().getStackTrace();
                int i = 0;
                String current = null;
                for (; i < elements.length; i++) {
                    current = elements[i].getClassName();
                    if (current.equals(LOGGER)) {
                        break;
                    }
                    if (current.equals(ASYNC_LOG_THREAD)) { // kind of mock since we lost the info
                        this.sourceClassName = ASYNC_LOG_NAME;
                        this.sourceMethodName = elements[i].getMethodName();
                        this.sourceInited = true;
                        return;
                    }
                }

                // Skip any internal OpenEJB Logger call
                while (++i < elements.length && elements[i].getClassName().equals(current)) { //NOPMD
                    // do nothing
                }

                // If we didn't run out of elements, set the source
                if (i < elements.length) {
                    this.sourceClassName = elements[i].getClassName();
                    this.sourceMethodName = elements[i].getMethodName();
                } else if (current != null && current.endsWith("LoggerThread")) {
                    this.sourceClassName = ASYNC_LOG_NAME;
                    this.sourceMethodName = "run";
                }

                this.sourceInited = true;
            }
        }
    }
}