BigQueryJdbcCustomLogger.java

/*
 * Copyright 2023 Google LLC
 *
 * 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 com.google.cloud.bigquery.jdbc;

import java.util.function.Supplier;
import java.util.logging.Level;
import java.util.logging.LogRecord;
import java.util.logging.Logger;

public class BigQueryJdbcCustomLogger extends Logger {

  protected BigQueryJdbcCustomLogger(String name, String resourceBundleName) {
    super(name, resourceBundleName);
    this.setParent(BigQueryJdbcRootLogger.getRootLogger());
  }

  public BigQueryJdbcCustomLogger(String name) {
    this(name, null);
    this.setParent(BigQueryJdbcRootLogger.getRootLogger());
  }

  private void logWithCaller(Level level, Supplier<String> msgSupplier) {
    logWithCaller(level, null, msgSupplier);
  }

  private void logWithCaller(Level level, Throwable thrown, Supplier<String> msgSupplier) {
    if (!isLoggable(level)) {
      return;
    }

    StackTraceElement[] stackTrace = new Throwable().getStackTrace();
    String sourceClass = "unknown";
    String sourceMethod = "unknown";

    for (StackTraceElement element : stackTrace) {
      String className = element.getClassName();
      if (!className.equals(BigQueryJdbcCustomLogger.class.getName())
          && !className.startsWith("com.google.cloud.bigquery.exception.")) {
        sourceClass = className;
        sourceMethod = element.getMethodName();
        break;
      }
    }

    if (thrown == null) {
      logp(level, sourceClass, sourceMethod, msgSupplier);
    } else {
      LogRecord record = new LogRecord(level, msgSupplier.get());
      record.setSourceClassName(sourceClass);
      record.setSourceMethodName(sourceMethod);
      record.setThrown(thrown);
      log(record);
    }
  }

  @Override
  public void finest(String msg) {
    logWithCaller(Level.FINEST, () -> msg);
  }

  @Override
  public void finer(String msg) {
    logWithCaller(Level.FINER, () -> msg);
  }

  @Override
  public void fine(String msg) {
    logWithCaller(Level.FINE, () -> msg);
  }

  void finest(String format, Object... args) {
    logWithCaller(Level.FINEST, () -> String.format(format, args));
  }

  void finer(String format, Object... args) {
    logWithCaller(Level.FINER, () -> String.format(format, args));
  }

  void fine(String format, Object... args) {
    logWithCaller(Level.FINE, () -> String.format(format, args));
  }

  void config(String format, Object... args) {
    logWithCaller(Level.CONFIG, () -> String.format(format, args));
  }

  void info(String format, Object... args) {
    logWithCaller(Level.INFO, () -> String.format(format, args));
  }

  void warning(String format, Object... args) {
    logWithCaller(Level.WARNING, () -> String.format(format, args));
  }

  void warning(Throwable thrown, String msg) {
    logWithCaller(Level.WARNING, thrown, () -> msg);
  }

  void warning(Throwable thrown, String format, Object... args) {
    logWithCaller(Level.WARNING, thrown, () -> String.format(format, args));
  }

  void severe(String format, Object... args) {
    logWithCaller(Level.SEVERE, () -> String.format(format, args));
  }

  public void severe(String msg, Throwable thrown) {
    logWithCaller(Level.SEVERE, thrown, () -> msg);
  }

  public void severe(String format, Throwable thrown, Object... args) {
    logWithCaller(Level.SEVERE, thrown, () -> String.format(format, args));
  }

  @Override
  public void finest(Supplier<String> msgSupplier) {
    logWithCaller(Level.FINEST, msgSupplier);
  }

  @Override
  public void fine(Supplier<String> msgSupplier) {
    logWithCaller(Level.FINE, msgSupplier);
  }
}