ExecContextLogger.java

/*
 * Copyright (c) 2001-2017, Zoltan Farkas All Rights Reserved.
 *
 * 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 2.1 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 General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307, USA.
 *
 * Additionally licensed with:
 *
 * Licensed 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.spf4j.log;

import edu.umd.cs.findbugs.annotations.SuppressFBWarnings;
import javax.annotation.Nullable;
import org.slf4j.Logger;
import org.slf4j.Marker;
import org.spf4j.base.Arrays;
import org.spf4j.base.ExecutionContext;
import org.spf4j.base.ExecutionContexts;
import org.spf4j.base.Wrapper;
import org.spf4j.base.avro.LogRecord;

/**
 * A Execution context aware logger.
 *
 * does the following:
 *
 * <li>
 * 1) if Execution context is present, it logs the context id. (relies on Logging back-ends supporting overflow args)
 * </li>
 * <li>
 * 2) if Execution context is present, it allows for context aware log level, and upgrades log messages to be logged by
 * backend.
 * </li>
 *
 * @author Zoltan Farkas
 */
@SuppressFBWarnings("LO_SUSPECT_LOG_PARAMETER")
public final class ExecContextLogger implements Logger, Wrapper<Logger> {

  private final XLog logger;

  public static ExecContextLogger from(final Logger wrapped) {
    if (wrapped instanceof ExecContextLogger) {
      return (ExecContextLogger) wrapped;
    } else {
      return new ExecContextLogger(wrapped);
    }
  }

  public ExecContextLogger(final Logger wrapped) {
    this(new SLf4jXLogAdapter(wrapped));
  }

  public ExecContextLogger(final XLog traceLogger) {
    this.logger = traceLogger;
  }

  @Override
  public Logger getWrapped() {
    return this.logger.getWrapped();
  }

  @Override
  public String getName() {
    return this.logger.getWrapped().getName();
  }

  public boolean isEnabled(final Level level, @Nullable final Marker marker) {
    ExecutionContext ctx = ExecutionContexts.current();
    if (ctx == null) {
      return logger.isEnabled(level, marker);
    }
    String name = getName();
    Level backendOverwrite = ctx.getBackendMinLogLevel(name);
    if (backendOverwrite == null) {
      return logger.isEnabled(level, marker) || level.ordinal() >= ctx.getContextMinLogLevel(name).ordinal();
    } else {
      return logger.isEnabled(level, marker)
              || level.ordinal()
              >= Math.min(ctx.getContextMinLogLevel(name).ordinal(), backendOverwrite.ordinal());
    }
  }


  public void log(final ExecutionContext ctx, final Level level, final LogRecord log) {
    if (ctx == null) {
      logger.log(null, level, "RemoteLog", log);
      return;
    }
    String name = getName();
    boolean logged = false;
    if (logger.isEnabled(level, null)) {
      logger.log(null, level, "RemoteLog", LogAttribute.traceId(ctx.getId()), log);
      logged = true;
    } else {
      Level backendOverwrite = ctx.getBackendMinLogLevel(name);
      if (backendOverwrite != null && backendOverwrite.ordinal() <= level.ordinal()) {
       logger.logUpgrade(null, level, "RemoteLog", LogAttribute.traceId(ctx.getId()), log);
       logged = true;
      }
    }
    if (ctx.getContextMinLogLevel(name).ordinal() <= level.ordinal()) {
      ctx.addLog(new AvroLogRecordImpl(log, logged));
    }
  }

  public void log(@Nullable final Marker marker, final Level level, final String msg, final Object... args) {
    log(ExecutionContexts.current(), marker, level, msg, args);
  }

  public void log(final ExecutionContext ctx, @Nullable final Marker marker,
          final Level level, final String msg, final Object... args) {
    if (ctx == null) {
      logger.log(marker, level, msg, args);
      return;
    }
    String name = getName();
    boolean logged;
    if (logger.isEnabled(level, marker)) {
      logger.log(null, level, msg, Arrays.append(args, LogAttribute.traceId(ctx.getId())));
      logged = true;
    } else {
      Level backendOverwrite = ctx.getBackendMinLogLevel(name);
      if (backendOverwrite == null) {
        logged = false;
      } else if (backendOverwrite.ordinal() <= level.ordinal()) {
        logger.logUpgrade(null, level, msg, Arrays.append(args, LogAttribute.traceId(ctx.getId())));
        logged = true;
      } else {
        logged = false;
      }
    }
    if (ctx.getContextMinLogLevel(name).ordinal() <= level.ordinal()) {
      ctx.addLog(new Slf4jLogRecordImpl(logged, name, level, marker, msg, args));
    }
  }

  @Override
  public boolean isTraceEnabled() {
    return isEnabled(Level.TRACE, null);
  }

  @Override
  public void trace(final String msg) {
    log(null, Level.TRACE, msg);
  }

  @Override
  public void trace(final String format, final Object arg) {
    log(null, Level.TRACE, format, arg);
  }

  @Override
  public void trace(final String format, final Object arg1, final Object arg2) {
    log(null, Level.TRACE, format, arg1, arg2);
  }

  @Override
  public void trace(final String format, final Object... arguments) {
    log(null, Level.TRACE, format, arguments);
  }

  @Override
  public void trace(final String msg, final Throwable t) {
    log(null, Level.TRACE, msg, t);
  }

  @Override
  public boolean isTraceEnabled(final Marker marker) {
    return isEnabled(Level.TRACE, marker);
  }

  @Override
  public void trace(final Marker marker, final String msg) {
    log(marker, Level.TRACE, msg);
  }

  @Override
  public void trace(final Marker marker, final String format, final Object arg) {
    log(marker, Level.TRACE, format, arg);
  }

  @Override
  public void trace(final Marker marker, final String format, final Object arg1, final Object arg2) {
    log(marker, Level.TRACE, format, arg1, arg2);
  }

  @Override
  public void trace(final Marker marker, final String format, final Object... argArray) {
    log(marker, Level.TRACE, format, argArray);
  }

  @Override
  public void trace(final Marker marker, final String msg, final Throwable t) {
    log(marker, Level.TRACE, msg, t);
  }

  @Override
  public boolean isDebugEnabled() {
    return isEnabled(Level.DEBUG, null);
  }

  @Override
  public void debug(final String msg) {
    log(null, Level.DEBUG, msg);
  }

  @Override
  public void debug(final String format, final Object arg) {
    log(null, Level.DEBUG, format, arg);
  }

  @Override
  public void debug(final String format, final Object arg1, final Object arg2) {
    log(null, Level.DEBUG, format, arg1, arg2);
  }

  @Override
  public void debug(final String format, final Object... arguments) {
    log(null, Level.DEBUG, format, arguments);
  }

  @Override
  public void debug(final String msg, final Throwable t) {
    log(null, Level.DEBUG, msg, t);
  }

  @Override
  public boolean isDebugEnabled(final Marker marker) {
    return isEnabled(Level.DEBUG, marker);
  }

  @Override
  public void debug(final Marker marker, final String msg) {
    log(marker, Level.DEBUG, msg);
  }

  @Override
  public void debug(final Marker marker, final String format, final Object arg) {
    log(marker, Level.DEBUG, format, arg);
  }

  @Override
  public void debug(final Marker marker, final String format, final Object arg1, final Object arg2) {
    log(marker, Level.DEBUG, format, arg1, arg2);
  }

  @Override
  public void debug(final Marker marker, final String format, final Object... arguments) {
    log(marker, Level.DEBUG, format, arguments);
  }

  @Override
  public void debug(final Marker marker, final String msg, final Throwable t) {
    log(marker, Level.DEBUG, msg, t);
  }

  @Override
  public boolean isInfoEnabled() {
    return isEnabled(Level.INFO, null);
  }

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

  @Override
  public void info(final String format, final Object arg) {
    log(null, Level.INFO, format, arg);
  }

  @Override
  public void info(final String format, final Object arg1, final Object arg2) {
    log(null, Level.INFO, format, arg1, arg2);
  }

  @Override
  public void info(final String format, final Object... arguments) {
    log(null, Level.INFO, format, arguments);
  }

  @Override
  public void info(final String msg, final Throwable t) {
    log(null, Level.INFO, msg, t);
  }

  @Override
  public boolean isInfoEnabled(final Marker marker) {
    return isEnabled(Level.INFO, marker);
  }

  @Override
  public void info(final Marker marker, final String msg) {
    log(marker, Level.INFO, msg);
  }

  @Override
  public void info(final Marker marker, final String format, final Object arg) {
    log(marker, Level.INFO, format, arg);
  }

  @Override
  public void info(final Marker marker, final String format, final Object arg1, final Object arg2) {
    log(marker, Level.INFO, format, arg1, arg2);
  }

  @Override
  public void info(final Marker marker, final String format, final Object... arguments) {
    log(marker, Level.INFO, format, arguments);
  }

  @Override
  public void info(final Marker marker, final String msg, final Throwable t) {
    log(marker, Level.INFO, msg, t);
  }

  @Override
  public boolean isWarnEnabled() {
    return isEnabled(Level.WARN, null);
  }

  @Override
  public void warn(final String msg) {
    log(null, Level.WARN, msg);
  }

  @Override
  public void warn(final String format, final Object arg) {
    log(null, Level.WARN, format, arg);
  }

  @Override
  public void warn(final String format, final Object... arguments) {
    log(null, Level.WARN, format, arguments);
  }

  @Override
  public void warn(final String format, final Object arg1, final Object arg2) {
    log(null, Level.WARN, format, arg1, arg2);
  }

  @Override
  public void warn(final String msg, final Throwable t) {
    log(null, Level.WARN, msg, t);
  }

  @Override
  public boolean isWarnEnabled(final Marker marker) {
    return isEnabled(Level.WARN, marker);
  }

  @Override
  public void warn(final Marker marker, final String msg) {
    log(marker, Level.WARN, msg);
  }

  @Override
  public void warn(final Marker marker, final String format, final Object arg) {
    log(marker, Level.WARN, format, arg);
  }

  @Override
  public void warn(final Marker marker, final String format, final Object arg1, final Object arg2) {
    log(marker, Level.WARN, format, arg1, arg2);
  }

  @Override
  public void warn(final Marker marker, final String format, final Object... arguments) {
    log(marker, Level.WARN, format, arguments);
  }

  @Override
  public void warn(final Marker marker, final String msg, final Throwable t) {
    log(marker, Level.WARN, msg, t);
  }

  @Override
  public boolean isErrorEnabled() {
    return isEnabled(Level.ERROR, null);
  }

  @Override
  public void error(final String msg) {
    log(null, Level.ERROR, msg);
  }

  @Override
  public void error(final String format, final Object arg) {
    log(null, Level.ERROR, format, arg);
  }

  @Override
  public void error(final String format, final Object arg1, final Object arg2) {
    log(null, Level.ERROR, format, arg1, arg2);
  }

  @Override
  public void error(final String format, final Object... arguments) {
    log(null, Level.ERROR, format, arguments);
  }

  @Override
  public void error(final String msg, final Throwable t) {
    log(null, Level.ERROR, msg, t);
  }

  @Override
  public boolean isErrorEnabled(final Marker marker) {
    return isEnabled(Level.ERROR, marker);
  }

  @Override
  public void error(final Marker marker, final String msg) {
    log(marker, Level.ERROR, msg);
  }

  @Override
  public void error(final Marker marker, final String format, final Object arg) {
    log(marker, Level.ERROR, format, arg);
  }

  @Override
  public void error(final Marker marker, final String format, final Object arg1, final Object arg2) {
    log(marker, Level.ERROR, format, arg1, arg2);
  }

  @Override
  public void error(final Marker marker, final String format, final Object... arguments) {
    log(marker, Level.ERROR, format, arguments);
  }

  @Override
  public void error(final Marker marker, final String msg, final Throwable t) {
    log(marker, Level.ERROR, msg, t);
  }

  @Override
  public String toString() {
    return "ExecContextLogger{" + "traceLogger=" + this.logger + '}';
  }

}