Examples with SQLProvider org.apache.activemq.artemis.jdbc.store.sql.SQLProvider used on opensource projects

Example 1 with SQLProvider

use of org.apache.activemq.artemis.jdbc.store.sql.SQLProvider in project activemq-artemis by apache.

the class PagingStoreFactoryDatabase method newFileFactory.

private synchronized SequentialFileFactory newFileFactory(final String directoryName, boolean writeToDirectory) throws Exception {
    JDBCSequentialFile directoryList = (JDBCSequentialFile) pagingFactoryFileFactory.createSequentialFile(DIRECTORY_NAME);
    directoryList.open();
    SimpleString simpleString = SimpleString.toSimpleString(directoryName);
    ActiveMQBuffer buffer = ActiveMQBuffers.fixedBuffer(simpleString.sizeof());
    buffer.writeSimpleString(simpleString);
    if (writeToDirectory)
        directoryList.write(buffer, true);
    directoryList.close();
    final SQLProvider sqlProvider;
    if (dbConf.getDataSource() != null) {
        final SQLProvider.Factory sqlProviderFactory;
        if (dbConf.getSqlProviderFactory() != null) {
            sqlProviderFactory = dbConf.getSqlProviderFactory();
        } else {
            sqlProviderFactory = new PropertySQLProvider.Factory(dbConf.getDataSource());
        }
        sqlProvider = sqlProviderFactory.create(getTableNameForGUID(directoryName), SQLProvider.DatabaseStoreType.PAGE);
    } else {
        sqlProvider = JDBCUtils.getSQLProvider(dbConf.getJdbcDriverClassName(), getTableNameForGUID(directoryName), SQLProvider.DatabaseStoreType.PAGE);
    }
    final JDBCSequentialFileFactory fileFactory = new JDBCSequentialFileFactory(pagingFactoryFileFactory.getDbDriver().getConnection(), sqlProvider, executorFactory.getExecutor(), criticalErrorListener);
    final int jdbcNetworkTimeout = dbConf.getJdbcNetworkTimeout();
    if (jdbcNetworkTimeout >= 0) {
        fileFactory.setNetworkTimeout(this.executorFactory.getExecutor(), jdbcNetworkTimeout);
    }
    return fileFactory;
}

Example 2 with SQLProvider

use of org.apache.activemq.artemis.jdbc.store.sql.SQLProvider in project activemq-artemis by apache.

the class JdbcNodeManager method with.

public static JdbcNodeManager with(DatabaseStorageConfiguration configuration, ScheduledExecutorService scheduledExecutorService, ExecutorFactory executorFactory, IOCriticalErrorListener ioCriticalErrorListener) {
    if (configuration.getDataSource() != null) {
        final SQLProvider.Factory sqlProviderFactory;
        if (configuration.getSqlProviderFactory() != null) {
            sqlProviderFactory = configuration.getSqlProviderFactory();
        } else {
            sqlProviderFactory = new PropertySQLProvider.Factory(configuration.getDataSource());
        }
        final String brokerId = java.util.UUID.randomUUID().toString();
        return usingDataSource(brokerId, configuration.getJdbcLockExpirationMillis(), configuration.getJdbcLockRenewPeriodMillis(), configuration.getJdbcLockAcquisitionTimeoutMillis(), configuration.getJdbcMaxAllowedMillisFromDbTime(), configuration.getDataSource(), sqlProviderFactory.create(configuration.getNodeManagerStoreTableName(), SQLProvider.DatabaseStoreType.NODE_MANAGER), scheduledExecutorService, executorFactory, ioCriticalErrorListener);
    } else {
        final SQLProvider sqlProvider = JDBCUtils.getSQLProvider(configuration.getJdbcDriverClassName(), configuration.getNodeManagerStoreTableName(), SQLProvider.DatabaseStoreType.NODE_MANAGER);
        final String brokerId = java.util.UUID.randomUUID().toString();
        return usingConnectionUrl(brokerId, configuration.getJdbcLockExpirationMillis(), configuration.getJdbcLockRenewPeriodMillis(), configuration.getJdbcLockAcquisitionTimeoutMillis(), configuration.getJdbcMaxAllowedMillisFromDbTime(), configuration.getJdbcConnectionUrl(), configuration.getJdbcDriverClassName(), sqlProvider, scheduledExecutorService, executorFactory, ioCriticalErrorListener);
    }
}

Example 3 with SQLProvider

use of org.apache.activemq.artemis.jdbc.store.sql.SQLProvider in project activemq-artemis by apache.

the class ActiveMQTestBase method destroyTables.

public void destroyTables(List<String> tableNames) throws Exception {
    Driver driver = getDriver(getJDBCClassName());
    Connection connection = driver.connect(getTestJDBCConnectionUrl(), null);
    Statement statement = connection.createStatement();
    try {
        for (String tableName : tableNames) {
            connection.setAutoCommit(false);
            SQLProvider sqlProvider = JDBCUtils.getSQLProvider(getJDBCClassName(), tableName, SQLProvider.DatabaseStoreType.LARGE_MESSAGE);
            try (ResultSet rs = connection.getMetaData().getTables(null, null, sqlProvider.getTableName(), null)) {
                if (rs.next()) {
                    statement.execute("DROP TABLE " + sqlProvider.getTableName());
                }
                connection.commit();
            } catch (SQLException e) {
                connection.rollback();
            }
        }
        connection.setAutoCommit(true);
    } catch (Throwable e) {
        e.printStackTrace();
    } finally {
        connection.close();
    }
}