Example usage for org.apache.commons.dbcp BasicDataSource getConnection

Introduction

In this page you can find the example usage for org.apache.commons.dbcp BasicDataSource getConnection.

Prototype

public Connection getConnection() throws SQLException 

Source Link

Document

Create (if necessary) and return a connection to the database.

Usage

From source file:org.sonar.db.profiling.NullConnectionInterceptor.java

@Override
public Connection getConnection(BasicDataSource dataSource) throws SQLException {
    return dataSource.getConnection();
}

---

From source file:org.sonar.db.profiling.ProfiledConnectionInterceptor.java

@Override
public Connection getConnection(BasicDataSource dataSource) throws SQLException {
    return buildConnectionProxy(new ProfilingConnectionHandler(dataSource.getConnection()));
}

---

From source file:org.tdmx.lib.control.datasource.DynamicDataSource.java

@Override
public Connection getConnection() throws SQLException {
    if (getPartitionIdProvider() == null) {
        throw new SQLException("No partitionIdProvider.");
    }//from w  w w  .j a va  2  s.  com
    String partitionId = getPartitionIdProvider().getPartitionId();
    if (partitionId == null) {
        log.info("Partition defaulting to " + VALIDATION_PARTITION_ID);
        partitionId = VALIDATION_PARTITION_ID;
    }
    // must be fast. Caching at the DatabasePartitionServiceRepositoryImpl supports this.
    DatabaseConnectionInfo ci = getConfigurationProvider().getPartitionInfo(partitionId);
    if (ci == null) {
        throw new SQLException("No DatabaseConnectionInfo provided for partitionId " + partitionId);
    }
    DatabaseConnectionInfo existingCi = partitionConnectionInfoMap.get(partitionId);
    if (existingCi == null) {
        partitionConnectionInfoMap.put(partitionId, ci);
        log.info("First connection use for partitionId " + partitionId);
    } else if (!existingCi.equals(ci)) {
        // there's been a change of DB connection for a partition
        partitionConnectionInfoMap.put(partitionId, ci);
        log.warn("DatabasePartition change for partitionId " + partitionId);
    }
    BasicDataSource bds = connectionDataSourceMap.get(ci);
    if (bds == null) {
        createDataSource(ci);
        bds = connectionDataSourceMap.get(ci);
    }
    if (bds == null) {
        throw new SQLException("Unable to create BasicDataSource for " + ci);
    }
    return bds.getConnection();
}

---

From source file:org.wso2.carbon.identity.common.testng.MockInitialContextFactory.java

private static BasicDataSource createDb(String dbName, Class clazz, String[] files)
        throws TestCreationException {

    BasicDataSource dataSource = new BasicDataSource();
    dataSource.setDriverClassName("org.h2.Driver");
    dataSource.setUsername("username");
    dataSource.setPassword("password");
    dataSource.setUrl("jdbc:h2:mem:test" + dbName);
    try (Connection connection = dataSource.getConnection()) {
        for (String f : files) {
            File fileFromClasspathResource = getClasspathAccessibleFile(f, clazz);
            String scriptPath = null;
            File tempFile = null;
            if (fileFromClasspathResource != null && fileFromClasspathResource.exists()) {
                scriptPath = fileFromClasspathResource.getAbsolutePath();
            } else {
                //This may be from jar.
                tempFile = copyTempFile(f, clazz);
                scriptPath = tempFile.getAbsolutePath();
            }//  www .  j  ava2s.co  m
            if (scriptPath != null) {
                try (Statement statement = connection.createStatement()) {
                    statement.executeUpdate("RUNSCRIPT FROM '" + scriptPath + "'"); //NOSONAR
                } catch (SQLException e) {
                    throw new TestCreationException(
                            "Error while loading data to the in-memory H2 Database located from resource : "
                                    + fileFromClasspathResource + "\nabsolute path : " + scriptPath,
                            e);
                }
            }
            if (tempFile != null) {
                tempFile.delete();
            }
        }
        return dataSource;
    } catch (SQLException e) {
        throw new TestCreationException("Error while creating the in-memory H2 Database : ", e);
    }
}

---

From source file:org.wso2.carbon.identity.oauth.dao.TestOAuthDAOBase.java

protected void initiateH2Base(String databaseName, String scriptPath) throws Exception {

    BasicDataSource dataSource = new BasicDataSource();
    dataSource.setDriverClassName("org.h2.Driver");
    dataSource.setUsername("username");
    dataSource.setPassword("password");
    dataSource.setUrl("jdbc:h2:mem:test" + databaseName);
    try (Connection connection = dataSource.getConnection()) {
        connection.createStatement().executeUpdate("RUNSCRIPT FROM '" + scriptPath + "'");
    }//from w ww . j ava2  s  . c  o  m
    dataSourceMap.put(databaseName, dataSource);
}

---

From source file:org.wso2.carbon.identity.oauth.endpoint.user.impl.TestUtils.java

public static void initiateH2Base() throws Exception {

    BasicDataSource dataSource = new BasicDataSource();
    dataSource.setDriverClassName("org.h2.Driver");
    dataSource.setUsername("username");
    dataSource.setPassword("password");
    dataSource.setUrl("jdbc:h2:mem:test" + DB_NAME);
    try (Connection connection = dataSource.getConnection()) {
        connection.createStatement().executeUpdate("RUNSCRIPT FROM '" + getFilePath(H2_SCRIPT_NAME) + "'");
    }//from ww w .j  a v  a  2s . c  om
    dataSourceMap.put(DB_NAME, dataSource);
}

---

From source file:org.wso2.carbon.identity.oauth.endpoint.util.TestOAthEndpointBase.java

protected void initiateInMemoryH2() throws Exception {
    BasicDataSource dataSource = new BasicDataSource();

    dataSource.setDriverClassName("org.h2.Driver");
    dataSource.setUsername("username");
    dataSource.setPassword("password");
    dataSource.setUrl("jdbc:h2:mem:test");

    connection = dataSource.getConnection();
    connection.createStatement().executeUpdate("RUNSCRIPT FROM 'src/test/resources/dbscripts/h2.sql'");
}

---

From source file:org.wso2.carbon.identity.oauth2.dao.util.DAOUtils.java

public static void initializeDataSource(String databaseName, String scriptPath) throws Exception {
    BasicDataSource dataSource = new BasicDataSource();
    dataSource.setDriverClassName("org.h2.Driver");
    dataSource.setUsername("username");
    dataSource.setPassword("password");
    dataSource.setUrl("jdbc:h2:mem:" + databaseName);

    try (Connection connection = dataSource.getConnection()) {
        connection.createStatement().executeUpdate("RUNSCRIPT FROM '" + scriptPath + "'");
    }/* w w w  .j  a v a 2  s. co  m*/
    dataSourceMap.put(databaseName, dataSource);
}

---

From source file:org.wso2.carbon.identity.openidconnect.DefaultOIDCClaimsCallbackHandlerTest.java

@BeforeClass
public void setUp() throws Exception {

    System.setProperty(CarbonBaseConstants.CARBON_HOME, CARBON_HOME);
    BasicDataSource dataSource1 = new BasicDataSource();
    dataSource1.setDriverClassName("org.h2.Driver");
    dataSource1.setUsername("username");
    dataSource1.setPassword("password");
    dataSource1.setUrl("jdbc:h2:mem:test" + DB_NAME);
    connection = dataSource1.getConnection();
    connection.createStatement().executeUpdate("RUNSCRIPT FROM '" + getFilePath(H2_SCRIPT_NAME) + "'");

    mockStatic(FrameworkUtils.class);
    when(FrameworkUtils.getMultiAttributeSeparator()).thenReturn(MULTI_ATTRIBUTE_SEPARATOR_DEFAULT);

    RequestObjectService requestObjectService = Mockito.mock(RequestObjectService.class);
    List<RequestedClaim> requestedClaims = Collections.emptyList();
    when(requestObjectService.getRequestedClaimsForIDToken(anyString())).thenReturn(requestedClaims);
    when(requestObjectService.getRequestedClaimsForUserInfo(anyString())).thenReturn(requestedClaims);

    // Skipping filtering with user consent.
    // TODO: Remove mocking claims filtering based on consent when fixing https://github.com/wso2/product-is/issues/2676
    OpenIDConnectClaimFilterImpl openIDConnectClaimFilter = spy(new OpenIDConnectClaimFilterImpl());
    when(openIDConnectClaimFilter.getClaimsFilteredByUserConsent(anyMap(), any(AuthenticatedUser.class),
            anyString(), anyString())).thenAnswer(invocation -> invocation.getArguments()[0]);
    OpenIDConnectServiceComponentHolder.getInstance().getOpenIDConnectClaimFilters()
            .add(openIDConnectClaimFilter);

    OpenIDConnectServiceComponentHolder.setRequestObjectService(requestObjectService);
    defaultOIDCClaimsCallbackHandler = new DefaultOIDCClaimsCallbackHandler();

}

---

From source file:org.wso2.carbon.identity.openidconnect.DefaultOIDCClaimsCallbackHandlerTest.java

private JWTClaimsSet getJwtClaimSet(JWTClaimsSet.Builder jwtClaimsSetBuilder,
        OAuthTokenReqMessageContext requestMsgCtx) {

    OAuthServerConfiguration mockOAuthServerConfiguration = PowerMockito.mock(OAuthServerConfiguration.class);
    DataSource dataSource = mock(DataSource.class);
    mockStatic(JDBCPersistenceManager.class);
    JDBCPersistenceManager jdbcPersistenceManager = mock(JDBCPersistenceManager.class);
    mockStatic(OAuthServerConfiguration.class);
    when(OAuthServerConfiguration.getInstance()).thenReturn(mockOAuthServerConfiguration);
    when(mockOAuthServerConfiguration.isConvertOriginalClaimsFromAssertionsToOIDCDialect()).thenReturn(true);
    JWTClaimsSet jwtClaimsSet = null;/*from  w  w  w  . j a va2s  .  co  m*/

    try {
        if (connection.isClosed()) {

            BasicDataSource dataSource1 = new BasicDataSource();
            dataSource1.setDriverClassName("org.h2.Driver");
            dataSource1.setUsername("username");
            dataSource1.setPassword("password");
            dataSource1.setUrl("jdbc:h2:mem:test" + DB_NAME);
            Connection connection1 = null;
            connection1 = dataSource1.getConnection();
            Mockito.when(dataSource.getConnection()).thenReturn(connection1);

        } else {
            Mockito.when(dataSource.getConnection()).thenReturn(connection);
        }
    } catch (Exception e) {
        log.error("Error while obtaining the datasource. ");
    }

    Mockito.when(jdbcPersistenceManager.getInstance()).thenReturn(jdbcPersistenceManager);
    Mockito.when(jdbcPersistenceManager.getDataSource()).thenReturn(dataSource);
    jwtClaimsSet = defaultOIDCClaimsCallbackHandler.handleCustomClaims(jwtClaimsSetBuilder, requestMsgCtx);

    //return jwtClaimsSet;

    return jwtClaimsSet;
}

---

• «
• 1
• 2
• 3
• 4
• 5
• 6
• 7
• 8
• 9
• 10
• »