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

There is a newer version: 42.7.3-yb-1

/*
 * Copyright (c) 2004, PostgreSQL Global Development Group
 * See the LICENSE file in the project root for more information.
 */

package org.postgresql.util;

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 Object lock = new Object();

  @SuppressWarnings("method.invocation.invalid")
  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 {
      synchronized (lock) {
        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 {
      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 {
      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 {
    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);
    }
  }
}