Examples with StatementMethod org.vibur.dbcp.stcache.StatementMethod used on opensource projects

Example 1 with StatementMethod

use of org.vibur.dbcp.stcache.StatementMethod in project vibur-dbcp by vibur.

the class ViburDBCPDataSourceTest method testTwoPreparedSelectStatementsWithStatementsCache.

@Test
public void testTwoPreparedSelectStatementsWithStatementsCache() throws SQLException {
    ViburDBCPDataSource ds = createDataSourceWithStatementsCache();
    ConcurrentMap<StatementMethod, StatementHolder> mockedStatementCache = mockStatementCache(ds);
    try (Connection connection = ds.getConnection()) {
        executeAndVerifyPreparedSelectStatement(connection);
        executeAndVerifyPreparedSelectStatementByLastName(connection);
        InOrder inOrder = inOrder(mockedStatementCache);
        inOrder.verify(mockedStatementCache).get(key1.capture());
        inOrder.verify(mockedStatementCache).putIfAbsent(same(key1.getValue()), val1.capture());
        inOrder.verify(mockedStatementCache).get(key2.capture());
        inOrder.verify(mockedStatementCache).putIfAbsent(same(key2.getValue()), val2.capture());
        // key1 will be evicted from the StatementCache because its capacity is set to 1.
        assertEquals(1, mockedStatementCache.size());
        assertTrue(mockedStatementCache.containsKey(key2.getValue()));
        assertNotEquals(key1.getValue(), key2.getValue());
        assertEquals(EVICTED, val1.getValue().state().get());
        assertEquals(AVAILABLE, val2.getValue().state().get());
    }
}

Example 2 with StatementMethod

use of org.vibur.dbcp.stcache.StatementMethod in project vibur-dbcp by vibur.

the class ViburDBCPDataSourceTest method testPreparedSelectStatementWithStatementsCache.

@Test
public void testPreparedSelectStatementWithStatementsCache() throws SQLException {
    ViburDBCPDataSource ds = createDataSourceWithStatementsCache();
    ConcurrentMap<StatementMethod, StatementHolder> mockedStatementCache = mockStatementCache(ds);
    try (Connection connection = ds.getConnection()) {
        executeAndVerifyPreparedSelectStatement(connection);
        executeAndVerifyPreparedSelectStatement(connection);
        InOrder inOrder = inOrder(mockedStatementCache);
        inOrder.verify(mockedStatementCache).get(key1.capture());
        inOrder.verify(mockedStatementCache).putIfAbsent(same(key1.getValue()), val1.capture());
        inOrder.verify(mockedStatementCache).get(key2.capture());
        assertEquals(1, mockedStatementCache.size());
        assertTrue(mockedStatementCache.containsKey(key1.getValue()));
        assertEquals(key1.getValue(), key2.getValue());
        assertEquals(AVAILABLE, val1.getValue().state().get());
    }
}

Example 3 with StatementMethod

use of org.vibur.dbcp.stcache.StatementMethod in project hibernate-orm by hibernate.

the class ViburDBCPConnectionProviderTest method testSelectStatementWithStatementsCache.

@Test
public void testSelectStatementWithStatementsCache() {
    setUpPoolAndDatabase(1, 10);
    ConnectionProvider cp = sessionFactory().getServiceRegistry().getService(ConnectionProvider.class);
    ViburDBCPDataSource ds = ((ViburDBCPConnectionProvider) cp).getDataSource();
    ConcurrentMap<StatementMethod, StatementHolder> mockedStatementCache = mockStatementCache(ds);
    doInHibernate(this::sessionFactory, ViburDBCPConnectionProviderTest::executeAndVerifySelect);
    // We set above the poolMaxSize = 1, that's why the second session will get and use the same underlying connection.
    doInHibernate(this::sessionFactory, ViburDBCPConnectionProviderTest::executeAndVerifySelect);
    InOrder inOrder = inOrder(mockedStatementCache);
    inOrder.verify(mockedStatementCache).get(key1.capture());
    inOrder.verify(mockedStatementCache).putIfAbsent(same(key1.getValue()), val1.capture());
    inOrder.verify(mockedStatementCache).get(key2.capture());
    assertEquals(1, mockedStatementCache.size());
    assertTrue(mockedStatementCache.containsKey(key1.getValue()));
    assertEquals(key1.getValue(), key2.getValue());
    assertEquals(AVAILABLE, val1.getValue().state().get());
}

Example 4 with StatementMethod

use of org.vibur.dbcp.stcache.StatementMethod in project vibur-dbcp by vibur.

the class ViburDBCPDataSourceTest method testSelectStatementWithStatementsCache.

@Test
public void testSelectStatementWithStatementsCache() throws SQLException {
    ViburDBCPDataSource ds = createDataSourceWithStatementsCache();
    ConcurrentMap<StatementMethod, StatementHolder> mockedStatementCache = mockStatementCache(ds);
    try (Connection connection = ds.getConnection()) {
        executeAndVerifySelectStatement(connection);
        executeAndVerifySelectStatement(connection);
        verifyZeroInteractions(mockedStatementCache);
    }
}