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 import org.apache.commons.dbcp.ConnectionFactory;
  38 import org.apache.commons.dbcp.DriverConnectionFactory;
  39 import org.apache.commons.dbcp.PoolableConnection;
  40 import org.apache.commons.dbcp.PoolableConnectionFactory;
  41 import org.apache.commons.pool.ObjectPool;
  42 import org.apache.commons.pool.impl.GenericObjectPool;
  43
  44 public class DatabaseEngine {
  45   private ObjectPool connectionPool;
  46
  47   public DatabaseEngine() {
  48     MirPropertiesConfiguration configuration = MirPropertiesConfiguration.instance();
  49
  50     Properties connectionProperties = new Properties();
  51     connectionProperties.put("user", configuration.getString("Database.Username"));
  52     connectionProperties.put("password", configuration.getString("Database.Password"));
  53     String dbHost = configuration.getString("Database.Host");
  54     String dbPort = configuration.getString("Database.Port");
  55     String dbName = configuration.getString("Database.Name");
  56     String dbUrl = "jdbc:postgresql://"+dbHost+":"+dbPort+"/"+dbName+"?compatible=7.1";
  57     String dbDriver = configuration.getString("Database.Driver");
  58     Driver driver;
  59     try {
  60       driver = (Driver) (Class.forName(dbDriver)).newInstance();
  61     }
  62     catch (ClassNotFoundException e) {
  63       throw new Error("JDBC Driver not found: " + e.toString());
  64     }
  65     catch (InstantiationException e) {
  66       throw new Error("Can't instantiate JDBC Driver: " + e.toString());
  67     }
  68     catch (IllegalAccessException e) {
  69       throw new Error("No access to JDBC Driver: " + e.toString());
  70     }
  71
  72     connectionPool = new GenericObjectPool(null);
  73     ConnectionFactory rawConnectionFactory =
  74         new DriverConnectionFactory(driver, dbUrl, connectionProperties);
  75     new PoolableConnectionFactory(rawConnectionFactory, connectionPool,
  76         null, "select now()", false, true);
  77
  78     // test the connection right before someone borrows it
  79     ((GenericObjectPool) connectionPool).setTestOnBorrow(true);
  80
  81
  82   }
  83
  84   /**
  85    * Obtain a database connection
  86    */
  87   public Connection obtainConnection() throws Exception {
  88     return (Connection) connectionPool.borrowObject();
  89   }
  90
  91   /**
  92    * release a previously obtained database connection
  93    */
  94   public void releaseConnection(Connection aConnection) throws Exception {
  95     connectionPool.returnObject(aConnection);
  96   }
  97
  98   /**
  99    * Returns the native connection of a previously obtained connection
 100    */
 101   public Connection getNativeConnection(Connection aConnection) {
 102     return ((PoolableConnection) aConnection).getDelegate();
 103   }
 104
 105   public String getStatus() {
 106     return
 107         connectionPool.getNumActive() + " / " +
 108         (connectionPool.getNumIdle()+connectionPool.getNumActive()) +
 109         " db conn.";
 110   }
 111 }