PooledOrJndiDatasourceService.java

/*
 * This program is free software; you can redistribute it and/or modify it under the 
 * terms of the GNU General Public License, version 2 as published by the Free Software 
 * Foundation.
 *
 * You should have received a copy of the GNU General Public License along with this 
 * program; if not, you can obtain a copy at http://www.gnu.org/licenses/gpl-2.0.html 
 * or from the Free Software Foundation, Inc., 
 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA.
 *
 * This program 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.
 * 
 * Copyright 2005 - 2008 Pentaho Corporation.  All rights reserved.
 *  
 * @created Jul 07, 2008 
 * @author rmansoor
 */
package org.pentaho.platform.engine.services.connection.datasource.dbcp;

import javax.naming.NamingException;
import javax.sql.DataSource;

import org.pentaho.platform.api.data.DatasourceServiceException;
import org.pentaho.platform.api.data.IDatasourceService;
import org.pentaho.platform.api.engine.ObjectFactoryException;
import org.pentaho.platform.api.repository.datasource.DatasourceMgmtServiceException;
import org.pentaho.platform.api.repository.datasource.IDatasource;
import org.pentaho.platform.api.repository.datasource.IDatasourceMgmtService;
import org.pentaho.platform.engine.core.system.PentahoSystem;
import org.pentaho.platform.engine.services.messages.Messages;

public class PooledOrJndiDatasourceService extends BaseDatasourceService {

  public PooledOrJndiDatasourceService() {
	}

	DataSource retrieve(String datasource) throws DatasourceServiceException {
		DataSource ds = null;
		try {
      IDatasourceMgmtService datasourceMgmtSvc = (IDatasourceMgmtService) PentahoSystem.getObjectFactory().get(IDatasourceMgmtService.class,null);
			IDatasource dataSource = datasourceMgmtSvc.getDatasource(datasource);
			// Look in the database for the datasource
			if(dataSource != null) {
			  ds = PooledDatasourceHelper.setupPooledDataSource(dataSource);
			// Database does not have the datasource, look in jndi now
			} else {
			  ds = getJndiDataSource(datasource);
			}
			// if the resulting datasource is not null then store it in the cache
			if(ds != null) {
			  cacheManager.putInRegionCache(IDatasourceService.JDBC_DATASOURCE, datasource, ds);  
			}

    } catch (ObjectFactoryException objface) {
      try {
        return getJndiDataSource(datasource);  
      }
      catch(DatasourceServiceException dse) {
       throw new DatasourceServiceException(Messages.getInstance().getErrorString("PooledOrJndiDatasourceService.ERROR_0003_UNABLE_TO_GET_JNDI_DATASOURCE") ,dse);  //$NON-NLS-1$
      }
		} catch (DatasourceMgmtServiceException daoe) {
      try {
        return getJndiDataSource(datasource);  
      }
      catch(DatasourceServiceException dse) {
        throw new DatasourceServiceException(Messages.getInstance().getErrorString("PooledOrJndiDatasourceService.ERROR_0003_UNABLE_TO_GET_JNDI_DATASOURCE") ,dse);         //$NON-NLS-1$
      }
		}
		return ds;
	}

	/**
	 * This method clears the JNDI DS cache.  The need exists because after a JNDI
	 * connection edit the old DS must be removed from the cache.
	 *
	 */
	public void clearCache() {
		cacheManager.removeRegionCache(IDatasourceService.JDBC_DATASOURCE);
	}

	/**
	 * This method clears the JNDI DS cache.  The need exists because after a JNDI
	 * connection edit the old DS must be removed from the cache.
	 *
	 */
	public void clearDataSource(String dsName) {
		cacheManager.removeFromRegionCache(IDatasourceService.JDBC_DATASOURCE, dsName);
	}

	/**
	 * Since JNDI is supported different ways in different app servers, it's
	 * nearly impossible to have a ubiquitous way to look up a datasource. This
	 * method is intended to hide all the lookups that may be required to find a
	 * jndi name.
	 * 
	 * @param dsName
	 *            The Datasource name
	 * @return DataSource if there is one bound in JNDI
	 * @throws NamingException
	 */
	public DataSource getDataSource(String dsName)
			throws DatasourceServiceException {
		DataSource dataSource = null;
		if (cacheManager != null) {
      if(!cacheManager.cacheEnabled(IDatasourceService.JDBC_DATASOURCE)) {
        cacheManager.addCacheRegion(IDatasourceService.JDBC_DATASOURCE);
      } 
			Object foundDs = cacheManager.getFromRegionCache(
					IDatasourceService.JDBC_DATASOURCE, dsName);
			if (foundDs != null) {
				dataSource = (DataSource) foundDs;
			} else {
				dataSource = retrieve(dsName);
			}
		}
		return dataSource;
	}

}