org.postgresql.util.LogWriterHandler Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of jdbc-yugabytedb Show documentation
Java JDBC 4.2 (JRE 8+) driver for YugaByte SQL database
The newest version!
/*
 * Copyright (c) 2004, PostgreSQL Global Development Group
 * See the LICENSE file in the project root for more information.
 */

package org.postgresql.util;

import org.postgresql.jdbc.ResourceLock;

import org.checkerframework.checker.nullness.qual.Nullable;

import java.io.Writer;
import java.util.logging.ErrorManager;
import java.util.logging.Formatter;
import java.util.logging.Handler;
import java.util.logging.Level;
import java.util.logging.LogRecord;
import java.util.logging.SimpleFormatter;

public class LogWriterHandler extends Handler {

  private @Nullable Writer writer;
  private final ResourceLock lock = new ResourceLock();

  @SuppressWarnings("method.invocation")
  public LogWriterHandler(Writer inWriter) {
    super();
    setLevel(Level.INFO);
    setFilter(null);
    setFormatter(new SimpleFormatter());
    setWriter(inWriter);
  }

  @Override
  public void publish(LogRecord record) {
    final String formatted;
    final Formatter formatter = getFormatter();

    try {
      formatted = formatter.format(record);
    } catch (Exception ex) {
      reportError("Error Formatting record", ex, ErrorManager.FORMAT_FAILURE);
      return;
    }

    if (formatted.length() == 0) {
      return;
    }
    try {
      try (ResourceLock ignore = lock.obtain()) {
        Writer writer = this.writer;
        if (writer != null) {
          writer.write(formatted);
        }
      }
    } catch (Exception ex) {
      reportError("Error writing message", ex, ErrorManager.WRITE_FAILURE);
    }
  }

  @Override
  public void flush() {
    try (ResourceLock ignore = lock.obtain()) {
      Writer writer = this.writer;
      if (writer != null) {
        writer.flush();
      }
    } catch ( Exception ex ) {
      reportError("Error on flush", ex, ErrorManager.WRITE_FAILURE);
    }
  }

  @Override
  public void close() throws SecurityException {
    try (ResourceLock ignore = lock.obtain()) {
      Writer writer = this.writer;
      if (writer != null) {
        writer.close();
      }
    } catch ( Exception ex ) {
      reportError("Error closing writer", ex, ErrorManager.WRITE_FAILURE);
    }
  }

  private void setWriter(Writer writer) throws IllegalArgumentException {
    try (ResourceLock ignore = lock.obtain()) {
      if (writer == null) {
        throw new IllegalArgumentException("Writer cannot be null");
      }
      this.writer = writer;

      try {
        writer.write(getFormatter().getHead(this));
      } catch (Exception ex) {
        reportError("Error writing head section", ex, ErrorManager.WRITE_FAILURE);
      }
    }
  }
}