Examples with H2Connection org.dbunit.ext.h2.H2Connection used on opensource projects

Example 1 with H2Connection

use of org.dbunit.ext.h2.H2Connection in project sharding-jdbc by shardingjdbc.

the class DBUnitUtil method getConnection.

/**
 * Get connection.
 * @param dbEnv Database environment
 * @param connection connection
 * @return database connection
 * @throws DatabaseUnitException DatabaseUnitException
 */
public static IDatabaseConnection getConnection(final DatabaseEnvironment dbEnv, final Connection connection) throws DatabaseUnitException {
    switch(dbEnv.getDatabaseType()) {
        case H2:
            return new H2Connection(connection, "PUBLIC");
        case MySQL:
            return new MySqlConnection(connection, null);
        case PostgreSQL:
            DatabaseConnection databaseConnection = new DatabaseConnection(connection);
            databaseConnection.getConfig().setProperty("http://www.dbunit.org/properties/datatypeFactory", new PostgresqlDataTypeFactory());
            return databaseConnection;
        case Oracle:
            return new OracleConnection(connection, "JDBC");
        case SQLServer:
            return new MsSqlConnection(connection);
        default:
            throw new UnsupportedOperationException(dbEnv.getDatabaseType().name());
    }
}

Example 2 with H2Connection

use of org.dbunit.ext.h2.H2Connection in project ignite by apache.

the class JdbcThinTransactionsLeaksMvccTest method usedConnectionCount.

/**
 * @param igx Node.
 * @return Count of detached connections.
 */
private int usedConnectionCount(IgniteEx igx) {
    ConnectionManager connMgr = ((IgniteH2Indexing) igx.context().query().getIndexing()).connections();
    Set<H2Connection> usedConns = GridTestUtils.getFieldValue(connMgr, "usedConns");
    return usedConns.size();
}

Example 3 with H2Connection

use of org.dbunit.ext.h2.H2Connection in project ignite by apache.

the class IgniteCacheQueryH2IndexingLeakTest method getStatementCacheSize.

/**
 * @param qryProcessor Query processor.
 * @return size of statement cache.
 */
private static int getStatementCacheSize(GridQueryProcessor qryProcessor) {
    IgniteH2Indexing h2Idx = (IgniteH2Indexing) qryProcessor.getIndexing();
    ConcurrentStripedPool<H2Connection> conns = GridTestUtils.getFieldValue(h2Idx.connections(), "connPool");
    return conns.stream().mapToInt(H2Connection::statementCacheSize).sum();
}