/*
* 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 <http://www.gnu.org/licenses/>.
*/
package net.sf.hajdbc.cache.eager;
import java.sql.Connection;
import java.sql.DatabaseMetaData;
import java.sql.SQLException;
import java.util.Map;
import java.util.TreeMap;
import net.sf.hajdbc.Database;
import net.sf.hajdbc.DatabaseCluster;
import net.sf.hajdbc.DatabaseProperties;
import net.sf.hajdbc.cache.DatabaseMetaDataCache;
import net.sf.hajdbc.dialect.Dialect;
import net.sf.hajdbc.util.Resources;
/**
* Per-database {@link DatabaseMetaDataCache} implementation that populates itself eagerly.
* @author Paul Ferraro
*/
public class EagerDatabaseMetaDataCache<Z, D extends Database<Z>> implements DatabaseMetaDataCache<Z, D>
{
private final Map<D, DatabaseProperties> map = new TreeMap<D, DatabaseProperties>();
private final DatabaseCluster<Z, D> cluster;
public EagerDatabaseMetaDataCache(DatabaseCluster<Z, D> cluster)
{
this.cluster = cluster;
}
/**
* {@inheritDoc}
* @see net.sf.hajdbc.cache.DatabaseMetaDataCache#flush()
*/
@Override
public void flush() throws SQLException
{
Map<D, DatabaseProperties> map = new TreeMap<D, DatabaseProperties>();
for (D database: this.cluster.getBalancer())
{
Connection connection = database.connect(database.createConnectionSource(), database.decodePassword(this.cluster.getDecoder()));
try
{
map.put(database, this.createDatabaseProperties(connection));
}
finally
{
Resources.close(connection);
}
}
synchronized (this.map)
{
this.map.clear();
this.map.putAll(map);
}
}
/**
* {@inheritDoc}
* @see net.sf.hajdbc.cache.DatabaseMetaDataCache#getDatabaseProperties(net.sf.hajdbc.Database, java.sql.Connection)
*/
@Override
public DatabaseProperties getDatabaseProperties(D database, Connection connection) throws SQLException
{
synchronized (this.map)
{
DatabaseProperties properties = this.map.get(database);
if (properties == null)
{
properties = this.createDatabaseProperties(connection);
this.map.put(database, properties);
}
return properties;
}
}
private DatabaseProperties createDatabaseProperties(Connection connection) throws SQLException
{
DatabaseMetaData metaData = connection.getMetaData();
Dialect dialect = this.cluster.getDialect();
return new EagerDatabaseProperties(metaData, dialect);
}
}