net.sf.hajdbc.sql.DriverDatabase Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of ha-jdbc2 Show documentation
High-Availability JDBC
There is a newer version: 3.6.61
/*
 * HA-JDBC: High-Availability JDBC
 * Copyright (C) 2012  Paul Ferraro
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * 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 Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with this program.  If not, see .
 */
package net.sf.hajdbc.sql;

import java.sql.Connection;
import java.sql.Driver;
import java.sql.DriverManager;
import java.sql.SQLException;
import java.util.Map;
import java.util.Properties;

import javax.xml.bind.annotation.XmlType;

import net.sf.hajdbc.Messages;
import net.sf.hajdbc.management.Description;
import net.sf.hajdbc.management.MBean;
import net.sf.hajdbc.management.ManagedAttribute;

/**
 * @author  Paul Ferraro
 * @version $Revision: 1948 $
 * @since   1.0
 */
@MBean
@Description("Database accessed via DriverManager")
@XmlType(name = "database")
public class DriverDatabase extends AbstractDatabase
{
	private static final String USER = "user"; //$NON-NLS-1$
	private static final String PASSWORD = "password"; //$NON-NLS-1$

	/**
	 * {@inheritDoc}
	 * @see net.sf.hajdbc.sql.AbstractDatabase#setLocation(java.lang.String)
	 */
	@ManagedAttribute
	@Description("JDBC url")
	@Override
	public void setLocation(String location)
	{
		getDriver(location);

		super.setLocation(location);
	}
	
	/**
	 * {@inheritDoc}
	 * @see net.sf.hajdbc.Database#connect(java.lang.Object, java.lang.String)
	 */
	@Override
	public Connection connect(Driver driver, String password) throws SQLException
	{
		Properties properties = new Properties();
		
		for (Map.Entry entry: this.getProperties().entrySet())
		{
			properties.setProperty(entry.getKey(), entry.getValue());
		}
		
		if (this.requiresAuthentication())
		{
			properties.setProperty(USER, this.getUser());
			if (password != null)
			{
				properties.setProperty(PASSWORD, password);
			}
		}
		
		return driver.connect(this.getLocation(), properties);
	}

	/**
	 * @see net.sf.hajdbc.Database#getConnectionSource()
	 */
	@Override
	public Driver getConnectionSource()
	{
		return getDriver(this.getLocation());
	}

	public String parseVendor()
	{
		String url = this.getLocation();
		return url.substring(5, url.indexOf(":", 5));
	}

	private static Driver getDriver(String url)
	{
		try
		{
			return DriverManager.getDriver(url);
		}
		catch (SQLException e)
		{
			throw new IllegalArgumentException(Messages.JDBC_URL_REJECTED.getMessage(url), e);
		}
	}
}