All Downloads are FREE. Search and download functionalities are using the official Maven repository.

com.darwinsys.sql.ConnectionUtil Maven / Gradle / Ivy

There is a newer version: 1.8.0
Show newest version
package com.darwinsys.sql;

import java.io.File;
import java.io.FileInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.sql.Connection;
import java.sql.DriverManager;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.Enumeration;
import java.util.List;
import java.util.Properties;
import java.util.Set;
import java.util.TreeSet;

import com.darwinsys.database.DataBaseException;
import com.darwinsys.util.Verbosity;

/** Encapsulate the Connection-related operations that every
 * JDBC program seems to use.
 */
public class ConnectionUtil {
	/** The default config filename, relative to CLASSPATH and/or ${user.home} */
	public static final String DEFAULT_NAME = ".db.properties";
	/** The current config filename */
	private static String configFileName =
		System.getProperty("user.home") + File.separator + DEFAULT_NAME;
	private static Verbosity verbosity = Verbosity.QUIET;
	
	/** Sets the full path of the config file to read.
	 * @param configFileName The FileName of the configuration file to use.
	 */
	public static void setConfigFileName(String configFileName) {
		File file = new File(configFileName);
		if (!file.canRead()) {
			throw new IllegalArgumentException("Unreadable: " + configFileName);
		}
		try { // to set saved filename to canonical path
			ConnectionUtil.configFileName = file.getCanonicalPath();
		} catch (IOException ex) {
			System.err.println("Warning: IO error checking path: " + configFileName);
			ConnectionUtil.configFileName = configFileName;
		}
	}

	/** Returns the full path of the configuration file being used.
	 * @return Returns the configFileName.
	 */
	public static String getConfigFileName() {
		return configFileName;
	}

	/** Get a SimpleSQLConfiguration for the given config using the default or set property file name */
	public static Configuration getConfiguration(String config) throws DataBaseException {
		InputStream inputStream = null;
		try {
			Properties p = new Properties();
			inputStream = ConnectionUtil.class.getResourceAsStream("/" + DEFAULT_NAME);
			if (inputStream == null) {
				inputStream = new FileInputStream(configFileName);
			}
			p.load(new FileInputStream(configFileName));
			return getConfiguration(p, config);
		} catch (IOException ex) {
			throw new DataBaseException(ex.toString());
		} finally {
			if (inputStream != null) {
				try {
					inputStream.close();
				} catch (IOException stupidException) {
					// empty
				}
			}
		}
	}

	/**
	 * @param p The Properties file
	 * @param config The name of the wanted configuration
	 * @return The matching configuration
	 */
	static SimpleSQLConfiguration getConfiguration(final Properties p, final String config) {
		final String db_url = p.getProperty(config  + "." + "DBURL");
		final String db_driver = p.getProperty(config  + "." + "DBDriver");
		final String db_user = p.getProperty(config  + "." + "DBUser");
		final String db_password = p.getProperty(config  + "." + "DBPassword");
		if (db_driver == null || db_url == null) {
			throw new DataBaseException("Driver or URL null: " + config);
		}
		return new SimpleSQLConfiguration(config, db_url, db_driver, db_user, db_password);
	}

	/** Get a Connection for the given config using the default or set property file name */
	public static Connection getConnection(final String configName) throws DataBaseException {
		try {
			final Properties p = new Properties();
			p.load(new FileInputStream(configFileName));
			return getConnection(p, configName);
		} catch (IOException ex) {
			throw new DataBaseException(ex.toString());
		}
	}

	/** Get a Connection for the given config name from a provided Properties */
	public static Connection getConnection(Properties p,  String configName) throws DataBaseException {
		try {
			String db_url = p.getProperty(configName  + "." + "DBURL");
			String db_driver = p.getProperty(configName  + "." + "DBDriver");
			String db_user = p.getProperty(configName  + "." + "DBUser");
			String db_password = p.getProperty(configName  + "." + "DBPassword");
			if (db_driver == null || db_url == null) {
				throw new DataBaseException("Driver or URL null: " + configName);
			}
			return getConnection(db_url, db_driver, db_user, db_password);
		} catch (ClassNotFoundException ex) {
			throw new DataBaseException(ex.toString());

		} catch (SQLException ex) {
			throw new DataBaseException(ex.toString());
		}
	}

	public static Connection getConnection(String dbUrl, String dbDriver,
					String dbUserName, String dbPassword)
			throws ClassNotFoundException, SQLException {

		// Load the database driver
		if (verbosity != Verbosity.QUIET) {
			System.out.println("Loading driver " + dbDriver);
		}
		Class.forName(dbDriver);

		if (verbosity != Verbosity.QUIET) {
			System.out.println("Connecting to DB " + dbUrl);
		}
		return DriverManager.getConnection(
			dbUrl, dbUserName, dbPassword);
	}

	public static Connection getConnection(Configuration c) throws ClassNotFoundException, SQLException {
		return getConnection(c.getDbURL(), c.getDriverName(), c.getUserName(), c.getPassword());
	}

	/** Generate a Set<String> of the config names available
	 * from the current configuration file.
	 * @return Set<String> of the configurations
	 */
	public static Set getConfigurationNames() {
		Set configNames = new TreeSet();
		try {
			Properties p = new Properties();
			FileInputStream is = new FileInputStream(configFileName);
			p.load(is);
			is.close();
			Enumeration enumeration = p.keys();
			while (enumeration.hasMoreElements()) {
				String element = (String) enumeration.nextElement();
				int offset;
				if ((offset= element.indexOf('.')) == -1)
					continue;
				String configName = element.substring(0, offset);
				configNames.add(configName);
			}
		} catch (IOException ex) {
			throw new DataBaseException(ex.toString());
		}
		return configNames;
	}

	/** Return all the configurations as SimpleSQLConfiguration objects
	 */
	public static List getConfigurations() {
		List configs = new ArrayList();
		for (String name : getConfigurationNames()) {
			configs.add(getConfiguration(name));
		}
		return configs;
	}
	
	/** Convert a TransactionIsolation int (defined in java.sql.Connection)
	 * to the corresponding printable string.
	 */
	public static String transactionIsolationToString(int txisolation) {
		switch(txisolation) {
			case Connection.TRANSACTION_NONE: 
				// transactions not supported.
				return "TRANSACTION_NONE";
			case Connection.TRANSACTION_READ_UNCOMMITTED: 
				// All three phenomena can occur
				return "TRANSACTION_NONE";
			case Connection.TRANSACTION_READ_COMMITTED: 
			// Dirty reads are prevented; non-repeatable reads and 
			// phantom reads can occur.
				return "TRANSACTION_READ_COMMITTED";
			case Connection.TRANSACTION_REPEATABLE_READ: 
				// Dirty reads and non-repeatable reads are prevented;
				// phantom reads can occur.
				return "TRANSACTION_REPEATABLE_READ";
			case Connection.TRANSACTION_SERIALIZABLE:
				// All three phenomena prvented; slowest!
				return "TRANSACTION_SERIALIZABLE";
			default:
				throw new IllegalArgumentException(
					txisolation + " not a valid TX_ISOLATION");
		}
	}

	public static Verbosity getVerbosity() {
		return verbosity;
	}

	public static void setVerbosity(Verbosity verbosity) {
		ConnectionUtil.verbosity = verbosity;
	}
}




© 2015 - 2024 Weber Informatics LLC | Privacy Policy