source/mircoders/global/DatabaseEngine.java

   1 /*
   2  * Copyright (C) 2001, 2002 The Mir-coders group
   3  *
   4  * This file is part of Mir.
   5  *
   6  * Mir is free software; you can redistribute it and/or modify
   7  * it under the terms of the GNU General Public License as published by
   8  * the Free Software Foundation; either version 2 of the License, or
   9  * (at your option) any later version.
  10  *
  11  * Mir is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  14  * GNU General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU General Public License
  17  * along with Mir; if not, write to the Free Software
  18  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  19  *
  20  * In addition, as a special exception, The Mir-coders gives permission to link
  21  * the code of this program with  any library licensed under the Apache Software License,
  22  * The Sun (tm) Java Advanced Imaging library (JAI), The Sun JIMI library
  23  * (or with modified versions of the above that use the same license as the above),
  24  * and distribute linked combinations including the two.  You must obey the
  25  * GNU General Public License in all respects for all of the code used other than
  26  * the above mentioned libraries.  If you modify this file, you may extend this
  27  * exception to your version of the file, but you are not obligated to do so.
  28  * If you do not wish to do so, delete this exception statement from your version.
  29  */
  30 package mircoders.global;
  31
  32 import java.sql.Connection;
  33 import java.sql.Driver;
  34 import java.util.Properties;
  35
  36 import mir.config.MirPropertiesConfiguration;
  37
  38 import org.apache.commons.dbcp.ConnectionFactory;
  39 import org.apache.commons.dbcp.DriverConnectionFactory;
  40 import org.apache.commons.dbcp.PoolableConnection;
  41 import org.apache.commons.dbcp.PoolableConnectionFactory;
  42 import org.apache.commons.pool.ObjectPool;
  43 import org.apache.commons.pool.impl.GenericObjectPool;
  44
  45 public class DatabaseEngine {
  46   private ObjectPool connectionPool;
  47
  48   public DatabaseEngine() {
  49     MirPropertiesConfiguration configuration = MirPropertiesConfiguration.instance();
  50
  51     Properties connectionProperties = new Properties();
  52     connectionProperties.put("user", configuration.getString("Database.Username"));
  53     connectionProperties.put("password", configuration.getString("Database.Password"));
  54     String dbHost = configuration.getString("Database.Host");
  55     String dbPort = configuration.getString("Database.Port");
  56     String dbName = configuration.getString("Database.Name");
  57     String dbUrl = "jdbc:postgresql://"+dbHost+":"+dbPort+"/"+dbName;
  58     String dbDriver = configuration.getString("Database.Driver");
  59     Driver driver;
  60     try {
  61       driver = (Driver) (Class.forName(dbDriver)).newInstance();
  62     }
  63     catch (ClassNotFoundException e) {
  64       throw new Error("JDBC Driver not found: " + e.toString());
  65     }
  66     catch (InstantiationException e) {
  67       throw new Error("Can't instantiate JDBC Driver: " + e.toString());
  68     }
  69     catch (IllegalAccessException e) {
  70       throw new Error("No access to JDBC Driver: " + e.toString());
  71     }
  72
  73     connectionPool = new GenericObjectPool(null);
  74     ConnectionFactory rawConnectionFactory =
  75         new DriverConnectionFactory(driver, dbUrl, connectionProperties);
  76     new PoolableConnectionFactory(rawConnectionFactory, connectionPool,
  77         null, "select now()", false, true);
  78
  79     // test the connection right before someone borrows it
  80     ((GenericObjectPool) connectionPool).setTestOnBorrow(true);
  81
  82
  83   }
  84
  85   /**
  86    * Obtain a database connection
  87    */
  88   public Connection obtainConnection() throws Exception {
  89     return (Connection) connectionPool.borrowObject();
  90   }
  91
  92   /**
  93    * release a previously obtained database connection
  94    */
  95   public void releaseConnection(Connection aConnection) throws Exception {
  96     connectionPool.returnObject(aConnection);
  97   }
  98
  99   /**
 100    * Returns the native connection of a previously obtained connection
 101    */
 102   public Connection getNativeConnection(Connection aConnection) {
 103     return ((PoolableConnection) aConnection).getDelegate();
 104   }
 105
 106   public String getStatus() {
 107     return
 108         connectionPool.getNumActive() + " / " +
 109         (connectionPool.getNumIdle()+connectionPool.getNumActive()) +
 110         " db conn.";
 111   }
 112 }