Java tutorial
/******************************************************************************* * This file is part of OpenNMS(R). * * Copyright (C) 2007-2014 The OpenNMS Group, Inc. * OpenNMS(R) is Copyright (C) 1999-2014 The OpenNMS Group, Inc. * * OpenNMS(R) is a registered trademark of The OpenNMS Group, Inc. * * OpenNMS(R) is free software: you can redistribute it and/or modify * it under the terms of the GNU Affero General Public License as published * by the Free Software Foundation, either version 3 of the License, * or (at your option) any later version. * * OpenNMS(R) 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 Affero General Public License for more details. * * You should have received a copy of the GNU Affero General Public License * along with OpenNMS(R). If not, see: * http://www.gnu.org/licenses/ * * For more information contact: * OpenNMS(R) Licensing <license@opennms.org> * http://www.opennms.org/ * http://www.opennms.com/ *******************************************************************************/ package org.opennms.core.db; import java.beans.PropertyVetoException; import java.io.ByteArrayInputStream; import java.io.IOException; import java.io.InputStream; import java.sql.Connection; import java.sql.SQLException; import java.sql.Statement; import junit.framework.TestCase; import org.apache.commons.io.IOUtils; import org.exolab.castor.xml.MarshalException; import org.exolab.castor.xml.ValidationException; import org.opennms.core.test.ConfigurationTestUtils; import org.opennms.core.utils.ConfigFileConstants; import org.opennms.test.DaoTestConfigBean; /** * * @author <a href="mailto:david@opennms.org">David Hustace</a> */ public class ConnectionFactoryIT extends TestCase { public void testMarshalDataSourceFromConfig() throws Exception { DaoTestConfigBean bean = new DaoTestConfigBean(); bean.afterPropertiesSet(); AtomikosDataSourceFactory factory1 = null; AtomikosDataSourceFactory factory2 = null; try { factory1 = makeFactory("opennms"); factory1.setUniqueResourceName("opennms"); factory2 = makeFactory("opennms2"); factory2.setUniqueResourceName("opennms2"); Connection conn = null; Statement s = null; try { conn = factory2.getConnection(); s = conn.createStatement(); s.execute("select * from pg_proc"); } finally { if (s != null) { s.close(); } if (conn != null) { conn.close(); } } } finally { Throwable t1 = null; Throwable t2 = null; if (factory1 != null) { try { factory1.close(); factory1 = null; } catch (Throwable e1) { t1 = e1; } } if (factory2 != null) { try { factory2.close(); factory2 = null; } catch (Throwable e2) { t2 = e2; } } if (t1 != null || t2 != null) { StringBuffer message = new StringBuffer(); message.append("Could not successfully close both C3P0 factories. Future tests might fail."); Throwable choice; if (t1 != null) { message.append(" First factory failed with: " + t1.getMessage() + "; see stack back trace."); choice = t1; if (t2 != null) { System.err.println( " Both factories failed to close. See stderr for second stack back trace."); t2.printStackTrace(System.err); } } else { choice = t2; } AssertionError e = new AssertionError(message.toString()); e.initCause(choice); throw e; } } } public void testPoolWithSqlExceptions() throws Exception { DaoTestConfigBean bean = new DaoTestConfigBean(); bean.afterPropertiesSet(); AtomikosDataSourceFactory factory = makeFactory("opennms"); factory.afterPropertiesSet(); // Verify the default values assertEquals(30, factory.poolAvailableSize()); assertEquals(30, factory.poolTotalSize()); // Close the factory so that we can reregister another factory with the same name factory.close(); final AtomikosDataSourceFactory factory2 = makeFactory("opennms"); factory2.setPoolSize(50); factory2.afterPropertiesSet(); // Verify the altered values assertEquals(50, factory2.poolAvailableSize()); assertEquals(50, factory2.poolTotalSize()); // Spawn a bunch of threads that generate continuous SQLExceptions for (int i = 0; i < 2000; i++) { new Thread() { public void run() { Connection conn = null; try { assertEquals(50, factory2.poolTotalSize()); conn = factory2.getConnection(); // Make sure that the total size of the pool stays at 50 assertEquals(50, factory2.poolTotalSize()); assertTrue(factory2.poolAvailableSize() > 0); // Fetching the current connection will push the available connections below 50 assertTrue(factory2.poolAvailableSize() < 50); Statement stmt = conn.createStatement(); stmt.execute("BEGIN"); stmt.execute("SELECT * FROM doesnt_exist_in_the_database"); } catch (SQLException e) { e.printStackTrace(); } finally { /* if (conn != null) { try { Statement stmt = conn.createStatement(); stmt.execute("ROLLBACK"); conn.close(); } catch (SQLException e) { fail("Exception thrown when trying to close connection"); } } */ } } }.start(); // Only sleep for a bit after spawning 10 threads so that we force some // contention. if (i % 10 == 0) Thread.sleep(200); } } private AtomikosDataSourceFactory makeFactory(String database) throws MarshalException, ValidationException, PropertyVetoException, SQLException, IOException, ClassNotFoundException { InputStream stream1 = new ByteArrayInputStream( ConfigurationTestUtils .getConfigForResourceWithReplacements(this, ConfigFileConstants .getFileName(ConfigFileConstants.OPENNMS_DATASOURCE_CONFIG_FILE_NAME)) .getBytes()); DataSourceFactory.setDataSourceConfigurationFactory(new DataSourceConfigurationFactory(stream1)); InputStream stream2 = new ByteArrayInputStream( ConfigurationTestUtils .getConfigForResourceWithReplacements(this, ConfigFileConstants .getFileName(ConfigFileConstants.OPENNMS_DATASOURCE_CONFIG_FILE_NAME)) .getBytes()); XADataSourceFactory.setDataSourceConfigurationFactory(new DataSourceConfigurationFactory(stream2)); try { return new AtomikosDataSourceFactory(); } finally { IOUtils.closeQuietly(stream1); IOUtils.closeQuietly(stream2); } } }