EDIT

/**

* Copyright (C) 2007 EDIT

* European Distributed Institute of Taxonomy

* http://www.e-taxonomy.eu

*

* The contents of this file are subject to the Mozilla Public License Version 1.1

* See LICENSE.TXT at the top of this package for the full license terms.

*/

package eu.etaxonomy.cdm.database;

import java.io.File;

import java.io.IOException;

import java.sql.Connection;

import java.sql.Driver;

import java.sql.DriverManager;

import java.sql.SQLException;

import java.util.Properties;

import javax.sql.DataSource;

import org.apache.commons.dbcp.BasicDataSource;

import org.apache.commons.pool.impl.GenericObjectPool;

import org.apache.logging.log4j.LogManager;

import org.apache.logging.log4j.Logger;

import org.h2.tools.Server;

import org.springframework.jdbc.CannotGetJdbcConnectionException;

import eu.etaxonomy.cdm.persistence.utils.CdmPersistenceUtils;

/**

 * @author a.mueller

 *

 * IN WORK

 *

 */

public class LocalH2 extends BasicDataSource {

	private static final Logger logger = LogManager.getLogger(LocalH2.class);

	private final String sep = System.getProperty("file.separator");

	/** url without database name */

	protected String pureUrl = "jdbc:h2:";

	/** database name */

	protected String dbName = "cdm";

	/** path, where database should be stored in the file system */

	protected String databasePath = getDefaultPath();

	/** Server instance */

	protected Server h2Server;

	/** if true starts server on init() */

	protected boolean isStartServer = true;

	/** makes the Server silent (no messages) */

	protected boolean isSilent = true;

	/** default driver class name */

	protected String DEFAULT_DRIVER_CLASS_NAME = "org.h2.Driver";

	String mode = H2Mode.EMBEDDED.toString();

	public LocalH2() {

		setDriverClassName(DEFAULT_DRIVER_CLASS_NAME);

		setLocalUrl();

	}

	/**

	 * @param url

	 * @throws CannotGetJdbcConnectionException

	 */

	public LocalH2(String url) throws CannotGetJdbcConnectionException {

		super();

		this.setUrl(url);

		setDriverClassName(DEFAULT_DRIVER_CLASS_NAME);

	}

	/* FIXME This is a workaround to solve a problem with dbcp connection pooling.

	 * Remove this when dbcp connection pool gets configured correctly

	 *

	 * (non-Javadoc)

	 * @see org.apache.commons.dbcp.BasicDataSource#createDataSource()

	 */

	@Override

	protected synchronized DataSource createDataSource() throws SQLException {

		super.createDataSource();

		connectionPool.setWhenExhaustedAction(GenericObjectPool.WHEN_EXHAUSTED_GROW);

		return dataSource;

	}

	/**

	 * @param url

	 * @param username

	 * @param password

	 * @throws CannotGetJdbcConnectionException

	 */

	public LocalH2(String url, String username, String password)

			throws CannotGetJdbcConnectionException {

		this(url);

		this.setUsername(username);

		this.setPassword(password);

	}

	/**

	 * @param driverClassName

	 * @param url

	 * @param username

	 * @param password

	 * @throws CannotGetJdbcConnectionException

	 */

	public LocalH2(String driverClassName, String url, String username,

			String password) throws CannotGetJdbcConnectionException {

		this(url, username, password);

		this.setDriverClassName(driverClassName);

	}

//** ********************************************************************************/

	public void init(){

		logger.info("LocalH2init");

		if (true){   //starting sever is not necessary for H2

			return;

		}

		if (isStartServer){

			this.startH2Server();

		}

	}

	public void destroy(){

		this.stopH2Server();

	}

	//checks if h2-server is started, if not it will be started	(taken over from hsqldb, maybe not necessary for H2

	private void startH2Server(){

		try {

			Driver driver = DriverManager.getDriver(getUrl());

			Properties prop = new Properties();

			prop.setProperty("user", this.getUsername());

			prop.setProperty("password", this.getPassword());

			Connection con = driver.connect(getUrl(),  prop);

			if (con == null) {

				logger.warn("Connection to URL " +  getUrl() +  " could not be established");

				throw new SQLException();

			}

		} catch (SQLException e) {

			try {

				//server is probably not running on the url (or login is wrong !!)

				logger.info("Start H2Server");

				String[] args = new String[] { "-trace" };

				h2Server = Server.createTcpServer(args).start();

//				h2Server.setDatabaseName(0, getDbName());

//				h2Server.setDatabasePath(0,  getDatabasePath());

				h2Server.start();

			} catch (SQLException sqle1) {

				logger.error("SQL Exception when starting Local H2Server: "+ sqle1);

			} catch (RuntimeException e1) {

				logger.error("Local H2Server could not be started or connection to existing server could not be established.");

			}

		}

	}

	/**

	 * stops the H2 Server

	 */

	private void stopH2Server(){

		if (h2Server != null){

			logger.info("stop H2Server");

			h2Server.stop();

		}

	}

	private static final String getDefaultPath(){

		File path;

		try {

			path = CdmPersistenceUtils.getWritableResourceDir();

		} catch (IOException e) {

			logger.error(e);

			throw new RuntimeException(e);

		}

		String subPath = File.separator + "h2" + File.separator + "LocalH2";

		return  path + subPath;

	}

	public String getDatabasePath() {

		return databasePath;

	}

	public void setDatabasePath(String databasePath) {

		if (databasePath.endsWith(sep)){

			databasePath = databasePath + "localCdm";

		}

		this.databasePath = databasePath;

	}

	public boolean isStartServer() {

		return isStartServer;

	}

	public void setStartServer(boolean isStartServer) {

		this.isStartServer = isStartServer;

	}

	public void setLocalUrl(){

		String dbName = "cdmLocal";

		String localUrlString = pureUrl + "file:" + getDefaultPath() + "/" + dbName;

		logger.info("setLocalUrl: " + localUrlString);

		setUrl(localUrlString);

	}

	public void setMode(String mode){

		this.mode = mode;

	}

	public String getMode(){

		return mode;

	}

}