Examples with TransactionCoordinator org.hibernate.resource.transaction.spi.TransactionCoordinator used on opensource projects

Example 1 with TransactionCoordinator

use of org.hibernate.resource.transaction.spi.TransactionCoordinator in project hibernate-orm by hibernate.

the class AbstractRegionAccessStrategyTest method mockedSession.

protected SharedSessionContractImplementor mockedSession() {
    SessionMock session = mock(SessionMock.class);
    when(session.isClosed()).thenReturn(false);
    when(session.getTimestamp()).thenReturn(TIME_SERVICE.wallClockTime());
    if (jtaPlatform == BatchModeJtaPlatform.class) {
        BatchModeTransactionCoordinator txCoord = new BatchModeTransactionCoordinator();
        when(session.getTransactionCoordinator()).thenReturn(txCoord);
        when(session.beginTransaction()).then(invocation -> {
            Transaction tx = txCoord.newTransaction();
            tx.begin();
            return tx;
        });
    } else if (jtaPlatform == null) {
        Connection connection = mock(Connection.class);
        JdbcConnectionAccess jdbcConnectionAccess = mock(JdbcConnectionAccess.class);
        try {
            when(jdbcConnectionAccess.obtainConnection()).thenReturn(connection);
        } catch (SQLException e) {
        // never thrown from mock
        }
        JdbcSessionOwner jdbcSessionOwner = mock(JdbcSessionOwner.class);
        when(jdbcSessionOwner.getJdbcConnectionAccess()).thenReturn(jdbcConnectionAccess);
        SqlExceptionHelper sqlExceptionHelper = mock(SqlExceptionHelper.class);
        JdbcServices jdbcServices = mock(JdbcServices.class);
        when(jdbcServices.getSqlExceptionHelper()).thenReturn(sqlExceptionHelper);
        ServiceRegistry serviceRegistry = mock(ServiceRegistry.class);
        when(serviceRegistry.getService(JdbcServices.class)).thenReturn(jdbcServices);
        JdbcSessionContext jdbcSessionContext = mock(JdbcSessionContext.class);
        when(jdbcSessionContext.getServiceRegistry()).thenReturn(serviceRegistry);
        when(jdbcSessionOwner.getJdbcSessionContext()).thenReturn(jdbcSessionContext);
        NonJtaTransactionCoordinator txOwner = mock(NonJtaTransactionCoordinator.class);
        when(txOwner.getResourceLocalTransaction()).thenReturn(new JdbcResourceTransactionMock());
        when(txOwner.getJdbcSessionOwner()).thenReturn(jdbcSessionOwner);
        when(txOwner.isActive()).thenReturn(true);
        TransactionCoordinator txCoord = JdbcResourceLocalTransactionCoordinatorBuilderImpl.INSTANCE.buildTransactionCoordinator(txOwner, null);
        when(session.getTransactionCoordinator()).thenReturn(txCoord);
        when(session.beginTransaction()).then(invocation -> {
            Transaction tx = new TransactionImpl(txCoord, session.getExceptionConverter());
            tx.begin();
            return tx;
        });
    } else {
        throw new IllegalStateException("Unknown JtaPlatform: " + jtaPlatform);
    }
    return session;
}

Example 2 with TransactionCoordinator

use of org.hibernate.resource.transaction.spi.TransactionCoordinator in project hibernate-orm by hibernate.

the class PutFromLoadValidator method registerRemoteInvalidation.

public boolean registerRemoteInvalidation(Object key, Object lockOwner) {
    SharedSessionContractImplementor session = currentSession.get();
    TransactionCoordinator transactionCoordinator = session == null ? null : session.getTransactionCoordinator();
    if (transactionCoordinator != null) {
        if (trace) {
            log.tracef("Registering synchronization on transaction in %s, cache %s: %s", lockOwnerToString(session), cache.getName(), key);
        }
        InvalidationSynchronization sync = new InvalidationSynchronization(nonTxPutFromLoadInterceptor, key, lockOwner);
        transactionCoordinator.getLocalSynchronizations().registerSynchronization(sync);
        return true;
    }
    // evict() command is not executed in session context
    return false;
}

Example 3 with TransactionCoordinator

use of org.hibernate.resource.transaction.spi.TransactionCoordinator in project hibernate-orm by hibernate.

the class NonStrictAccessDelegate method remove.

@Override
public void remove(SharedSessionContractImplementor session, Object key) throws CacheException {
    // there's no 'afterRemove', so we have to use our own synchronization
    // the API does not provide version of removed item but we can't load it from the cache
    // as that would be prone to race conditions - if the entry was updated in the meantime
    // the remove could be discarded and we would end up with stale record
    // See VersionedTest#testCollectionUpdate for such situation
    TransactionCoordinator transactionCoordinator = session.getTransactionCoordinator();
    RemovalSynchronization sync = new RemovalSynchronization(transactionCoordinator, writeCache, false, region, key);
    transactionCoordinator.getLocalSynchronizations().registerSynchronization(sync);
}

Example 4 with TransactionCoordinator

use of org.hibernate.resource.transaction.spi.TransactionCoordinator in project hibernate-orm by hibernate.

the class BasicJdbcTransactionTests method basicUsageTest.

@Test
public void basicUsageTest() throws Exception {
    final TransactionCoordinatorOwnerTestingImpl owner = new TransactionCoordinatorOwnerTestingImpl();
    final JdbcResourceLocalTransactionCoordinatorBuilderImpl transactionCoordinatorBuilder = new JdbcResourceLocalTransactionCoordinatorBuilderImpl();
    final TransactionCoordinator transactionCoordinator = transactionCoordinatorBuilder.buildTransactionCoordinator(owner, new TransactionCoordinatorBuilder.Options() {

        @Override
        public boolean shouldAutoJoinTransaction() {
            return false;
        }
    });
    SynchronizationCollectorImpl sync = new SynchronizationCollectorImpl();
    transactionCoordinator.getLocalSynchronizations().registerSynchronization(sync);
    transactionCoordinator.getTransactionDriverControl().begin();
    assertEquals(0, sync.getBeforeCompletionCount());
    assertEquals(0, sync.getSuccessfulCompletionCount());
    assertEquals(0, sync.getFailedCompletionCount());
    transactionCoordinator.getTransactionDriverControl().commit();
    assertEquals(1, sync.getBeforeCompletionCount());
    assertEquals(1, sync.getSuccessfulCompletionCount());
    assertEquals(0, sync.getFailedCompletionCount());
}

Example 5 with TransactionCoordinator

use of org.hibernate.resource.transaction.spi.TransactionCoordinator in project hibernate-orm by hibernate.

the class BasicJdbcTransactionTests method testSynchronizationFailure.

@Test
@SuppressWarnings("EmptyCatchBlock")
public void testSynchronizationFailure() {
    final TransactionCoordinatorOwnerTestingImpl owner = new TransactionCoordinatorOwnerTestingImpl();
    final JdbcResourceLocalTransactionCoordinatorBuilderImpl transactionCoordinatorBuilder = new JdbcResourceLocalTransactionCoordinatorBuilderImpl();
    final TransactionCoordinator transactionCoordinator = transactionCoordinatorBuilder.buildTransactionCoordinator(owner, new TransactionCoordinatorBuilder.Options() {

        @Override
        public boolean shouldAutoJoinTransaction() {
            return false;
        }
    });
    assertEquals(TransactionStatus.NOT_ACTIVE, transactionCoordinator.getTransactionDriverControl().getStatus());
    transactionCoordinator.getLocalSynchronizations().registerSynchronization(SynchronizationErrorImpl.forBefore());
    transactionCoordinator.getTransactionDriverControl().begin();
    assertEquals(TransactionStatus.ACTIVE, transactionCoordinator.getTransactionDriverControl().getStatus());
    try {
        transactionCoordinator.getTransactionDriverControl().commit();
    } catch (Exception expected) {
    } finally {
        assertEquals(TransactionStatus.NOT_ACTIVE, transactionCoordinator.getTransactionDriverControl().getStatus());
    }
}