fr.ms.log4jdbc.Driver Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of log4jdbc-file Show documentation
The newest version!
/*
 * This file is part of Log4Jdbc.
 *
 * Log4Jdbc is free software: you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * Log4Jdbc 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 General Public License
 * along with Log4Jdbc.  If not, see .
 *
 */
package fr.ms.log4jdbc;

import java.io.PrintWriter;
import java.sql.Connection;
import java.sql.DriverPropertyInfo;
import java.sql.SQLException;
import java.util.Enumeration;
import java.util.Iterator;
import java.util.Properties;
import java.util.logging.Logger;

import fr.ms.log4jdbc.context.JdbcContext;
import fr.ms.log4jdbc.proxy.Handlers;
import fr.ms.log4jdbc.utils.Service;
import fr.ms.log4jdbc.utils.SystemPropertyUtils;
import fr.ms.log4jdbc.utils.drivermanager.Log4JdbcDriverManager;
import fr.ms.log4jdbc.utils.drivermanager.Log4JdbcDriverManagerFactory;

/**
 * 
 * @see Marco4J
 * 
 * 
 * @author Marco Semiao
 * 
 */
public class Driver implements java.sql.Driver {

  private static final String LOG4JDBC_PREFIX = "jdbc:log4";

  private final static boolean logDriverManager = SystemPropertyUtils.getProperty("log4jdbc.driverManager.log", false);

  private final static Log4JdbcDriverManager driverManager = Log4JdbcDriverManagerFactory.getInstance();

  private java.sql.Driver driver;

  static {
    if (logDriverManager) {
      driverManager.setLogWriter(new PrintWriter(System.out));
    }
    loadAdditionalDrivers();
    loadDrivers();
  }

  public Connection connect(String url, final Properties info) throws SQLException {
    final java.sql.Driver d = getRealDriver(url);
    if (d == null) {
      return null;
    }

    url = getRealUrl(url);

    final Connection c = d.connect(url, info);

    if (c == null) {
      throw new SQLException("invalid or unknown driver url: " + url);
    }

    final JdbcContext jdbcContext = new JdbcContext(d, url);
    final Connection wrap = Handlers.getConnection(c, jdbcContext);

    driver = d;
    return wrap;
  }

  public boolean acceptsURL(final String url) throws SQLException {
    final java.sql.Driver d = getRealDriver(url);
    if (d != null) {
      driver = d;
      return true;
    }
    return false;
  }

  public DriverPropertyInfo[] getPropertyInfo(final String url, final Properties info) throws SQLException {
    final java.sql.Driver d = getRealDriver(url);
    if (d == null) {
      return new DriverPropertyInfo[0];
    }

    driver = d;
    return d.getPropertyInfo(url, info);
  }

  public int getMajorVersion() {
    if (driver == null) {
      return 1;
    }
    return driver.getMajorVersion();
  }

  public int getMinorVersion() {
    if (driver == null) {
      return 0;
    }
    return driver.getMinorVersion();
  }

  public boolean jdbcCompliant() {
    return driver != null && driver.jdbcCompliant();
  }

  public Logger getParentLogger() {
    try {
      return driver.getParentLogger();
    } catch (final Exception e) {
      throw new RuntimeException(e.getMessage());
    }
  }

  private static java.sql.Driver getRealDriver(String url) throws SQLException {
    if (url.startsWith(LOG4JDBC_PREFIX)) {
      url = getRealUrl(url);

      final Enumeration e = driverManager.getDrivers();

      java.sql.Driver d;
      while (e.hasMoreElements()) {
        d = (java.sql.Driver) e.nextElement();

        if (d.acceptsURL(url)) {
          return d;
        }
      }
    }
    return null;
  }

  private static String getRealUrl(final String url) {
    return url.substring(LOG4JDBC_PREFIX.length());
  }

  private static void loadAdditionalDrivers() {
    String drivers = System.getProperty("log4jdbc.drivers");

    if (logDriverManager) {
      System.out.println("Log4Jdbc DriverManager.Initialize: log4jdbc.drivers = " + drivers);
    }

    if (drivers != null) {
      while (drivers.length() != 0) {
        final int x = drivers.indexOf(':');
        String driver;
        if (x < 0) {
          driver = drivers;
          drivers = "";
        } else {
          driver = drivers.substring(0, x);
          drivers = drivers.substring(x + 1);
        }
        if (driver.length() == 0) {
          continue;
        }
        try {
          driver = driver.trim();
          if (logDriverManager) {
            System.out.println("Log4jdbc DriverManager.Initialize: loading " + driver);
          }
          final Class clazz = Class.forName(driver);
          final java.sql.Driver d = (java.sql.Driver) clazz.newInstance();
          driverManager.registerDriver(d);
        } catch (final Exception ex) {
          throw new RuntimeException("Log4jdbc DriverManager.Initialize: load " + driver + " failed", ex);
        }
      }
    }
  }

  private static void loadDrivers() {
    final Iterator providers = Service.providers(java.sql.Driver.class);

    while (providers.hasNext()) {
      final java.sql.Driver d = (java.sql.Driver) providers.next();
      try {
        final Class clazz = d.getClass();
        if (!Driver.class.equals(clazz)) {
          driverManager.registerDriver(d);
        }
      } catch (final Exception ex) {
        throw new RuntimeException("Log4jdbc DriverManager.Initialize: load " + d + " failed", ex);
      }
    }

    try {
      driverManager.registerDriver(new Driver());
    } catch (final Exception ex) {
      throw new RuntimeException("Log4jdbc DriverManager.Initialize: load " + Driver.class + " failed", ex);
    }
  }
}