databasemanager.java

来自「RESIN 3.2 最新源码」· Java 代码 · 共 265 行

/*
 * Copyright (c) 1998-2008 Caucho Technology -- all rights reserved
 *
 * This file is part of Resin(R) Open Source
 *
 * Each copy or derived work must preserve the copyright notice and this
 * notice unmodified.
 *
 * Resin Open Source 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 2 of the License, or
 * (at your option) any later version.
 *
 * Resin Open Source 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, or any warranty
 * of NON-INFRINGEMENT.  See the GNU General Public License for more
 * details.
 *
 * You should have received a copy of the GNU General Public License
 * along with Resin Open Source; if not, write to the
 *
 *   Free Software Foundation, Inc.
 *   59 Temple Place, Suite 330
 *   Boston, MA 02111-1307  USA
 *
 * @author Scott Ferguson
 */

package com.caucho.sql;

import com.caucho.loader.EnvironmentLocal;
import com.caucho.util.L10N;
import com.caucho.config.ConfigException;
import com.caucho.vfs.*;

import java.net.*;
import java.sql.*;
import javax.sql.*;
import java.util.*;
import java.util.logging.*;

/**
 * Manages databases in a local environment, e.g. for PHP dynamic
 * database lookup.
 */
public class DatabaseManager {
  protected static final Logger log
    = Logger.getLogger(DatabaseManager.class.getName());
  private static final L10N L = new L10N(DatabaseManager.class);

  private static final EnvironmentLocal<DatabaseManager> _localManager
    = new EnvironmentLocal<DatabaseManager>();

  private final HashMap<String,DBPool> _databaseMap
    = new HashMap<String,DBPool>();

  private final ArrayList<Driver> _driverList
    = new ArrayList<Driver>();

  private static int _gId;

  /**
   * The manager is never instantiated.
   */
  private DatabaseManager()
  {
    initDriverList();
  }

  /**
   * Returns the database manager for the local environment.
   */
  private static DatabaseManager getLocalManager()
  {
    synchronized (_localManager) {
      DatabaseManager manager = _localManager.getLevel();

      if (manager == null) {
	manager = new DatabaseManager();

	_localManager.set(manager);
      }

      return manager;
    }
  }

  /**
   * Returns a matching dbpool.
   */
  public static DataSource findDatabase(String url)
    throws SQLException
  {
    String driver = findDriverByUrl(url);
    
    return getLocalManager().findDatabaseImpl(url, driver);
  }

  /**
   * Returns a matching dbpool.
   */
  public static DataSource findDatabase(String url, String driver)
    throws SQLException
  {
    return getLocalManager().findDatabaseImpl(url, driver);
  }

  /**
   * Looks up the local database, creating if necessary.
   */
  private DataSource findDatabaseImpl(String url,
				      String driverName)
    throws SQLException
  {
    try {
      synchronized (_databaseMap) {
	DBPool db = _databaseMap.get(url);

	if (db == null) {
	  db = new DBPool();

	  db.setVar(url + "-" + _gId++);
	
	  DriverConfig driver = db.createDriver();

	  ClassLoader loader = Thread.currentThread().getContextClassLoader();
	
	  Class driverClass = Class.forName(driverName, false, loader);

	  driver.setType(driverClass);
	  driver.setURL(url);

	  db.init();

	  _databaseMap.put(url, db);
	}

	return db;
      }
    } catch (RuntimeException e) {
      throw e;
    } catch (SQLException e) {
      throw e;
    } catch (Exception e) {
      throw ConfigException.create(e);
    }
  }

  public static String findDriverByUrl(String url)
  {
    return getLocalManager().findDriverByUrlImpl(url);
  }
  
  private String findDriverByUrlImpl(String url)
  {
    for (int i = 0; i < _driverList.size(); i++) {
      try {
	Driver driver = (Driver) _driverList.get(i);
	
	if (driver.acceptsURL(url))
	  return driver.getClass().getName();
      } catch (Exception e) {
	log.log(Level.FINE, e.toString(), e);
      }
    }

    return null;
  }

  private void initDriverList()
  {
    try {
      Thread thread = Thread.currentThread();
      ClassLoader loader = thread.getContextClassLoader();
      
      Enumeration iter
	= loader.getResources("META-INF/services/java.sql.Driver");
      while (iter.hasMoreElements()) {
	URL url = (URL) iter.nextElement();

	ReadStream is = null;
	try {
	  is = Vfs.lookup(url.toString()).openRead();

	  String filename;

	  while ((filename = is.readLine()) != null) {
	    int p = filename.indexOf('#');
	    
	    if (p >= 0)
	      filename = filename.substring(0, p);

	    filename = filename.trim();
	    if (filename.length() == 0)
	      continue;

	    try {
	      Class cl = Class.forName(filename, false, loader);
	      Driver driver = null;
	      
	      if (Driver.class.isAssignableFrom(cl))
		driver = (Driver) cl.newInstance();

	      if (driver != null) {
		log.fine(L.l("DatabaseManager adding driver '{0}'",
			     driver.getClass().getName()));

		_driverList.add(driver);
	      }
	    } catch (Exception e) {
	      log.log(Level.FINE, e.toString(), e);
	    }
	  }
	} catch (Exception e) {
	  log.log(Level.FINE, e.toString(), e);
	} finally {
	  if (is != null)
	    is.close();
	}
      }
    } catch (Exception e) {
      log.log(Level.FINE, e.toString(), e);
    }
  }

  static class DatabaseKey {
    private String _url;
    private String _catalog;

    DatabaseKey(String url, String catalog)
    {
      _url = url;
      _catalog = catalog;
    }

    public int hashCode()
    {
      int hash = 37;

      hash = 65521 * hash + _url.hashCode();

      if (_catalog != null)
	hash = 65521 * hash + _catalog.hashCode();

      return hash;
    }

    public boolean equals(Object o)
    {
      if (! (o instanceof DatabaseKey))
	return false;

      DatabaseKey key = (DatabaseKey) o;

      if (! _url.equals(key._url))
	return false;

      return (_catalog == key._catalog
	      || _catalog != null && _catalog.equals(key._catalog));
    }
  }
}