MultiTransactionStatus.java

/*
 * Copyright (c) 2002-2022, City of Paris
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 *
 *  1. Redistributions of source code must retain the above copyright notice
 *     and the following disclaimer.
 *
 *  2. Redistributions in binary form must reproduce the above copyright notice
 *     and the following disclaimer in the documentation and/or other materials
 *     provided with the distribution.
 *
 *  3. Neither the name of 'Mairie de Paris' nor 'Lutece' nor the names of its
 *     contributors may be used to endorse or promote products derived from
 *     this software without specific prior written permission.
 *
 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
 * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
 * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDERS OR CONTRIBUTORS BE
 * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
 * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
 * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
 * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
 * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
 * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
 * POSSIBILITY OF SUCH DAMAGE.
 *
 * License 1.0
 */
package fr.paris.lutece.util.jpa.transaction;

import java.util.HashMap;
import java.util.Map;

import org.springframework.transaction.PlatformTransactionManager;
import org.springframework.transaction.TransactionStatus;

/**
 * This transaction status wraps several {@link TransactionStatus}
 */
public class MultiTransactionStatus implements TransactionStatus
{
    private Map<PlatformTransactionManager, TransactionStatus> _transactionStatuses;
    private PlatformTransactionManager _mainPTM;
    private boolean _bNewSynchonization;

    /**
     * Creates a TransactionStatus that handles several TransactionStatus
     * 
     * @param mainPTM
     *            will be default {@link PlatformTransactionManager} for status informations (is*, has* methods)
     */
    public MultiTransactionStatus( PlatformTransactionManager mainPTM )
    {
        _mainPTM = mainPTM;
        _transactionStatuses = new HashMap<>( );
    }

    /**
     * Sets new synchronization to true
     */
    public void setNewSynchonization( )
    {
        this._bNewSynchonization = true;
    }

    /**
     * true if new synchronization
     * 
     * @return true if new synchronization
     */
    public boolean isNewSynchonization( )
    {
        return this._bNewSynchonization;
    }

    /**
     * Gets the transaction status for the ptm
     * 
     * @param ptm
     *            the {@link PlatformTransactionManager}
     * @return transaction status found
     */
    public TransactionStatus getTransactionStatus( PlatformTransactionManager ptm )
    {
        return _transactionStatuses.get( ptm );
    }

    /**
     *
     * {@inheritDoc}
     */
    public void flush( )
    {
        for ( TransactionStatus ts : _transactionStatuses.values( ) )
        {
            ts.flush( );
        }
    }

    /**
     * Gets the main transaction status
     * 
     * @return the transaction status
     */
    private TransactionStatus getMainTransactionStatus( )
    {
        return _transactionStatuses.get( _mainPTM );
    }

    /**
     *
     * {@inheritDoc}
     */
    public boolean hasSavepoint( )
    {
        return getMainTransactionStatus( ).hasSavepoint( );
    }

    /**
     *
     * {@inheritDoc}
     */
    public boolean isCompleted( )
    {
        return getMainTransactionStatus( ).isCompleted( );
    }

    /**
     *
     * {@inheritDoc}
     */
    public boolean isNewTransaction( )
    {
        return getMainTransactionStatus( ).isNewTransaction( );
    }

    /**
     *
     * {@inheritDoc}
     */
    public boolean isRollbackOnly( )
    {
        return getMainTransactionStatus( ).isRollbackOnly( );
    }

    /**
     *
     * {@inheritDoc}
     */
    public void setRollbackOnly( )
    {
        for ( TransactionStatus ts : _transactionStatuses.values( ) )
        {
            ts.setRollbackOnly( );
        }
    }

    /**
     *
     * {@inheritDoc}
     */
    public Object createSavepoint( )
    {
        return null;
    }

    /**
     *
     * {@inheritDoc}
     */
    public void releaseSavepoint( Object savepoint )
    {
        for ( TransactionStatus ts : _transactionStatuses.values( ) )
        {
            ts.releaseSavepoint( savepoint );
        }
    }

    /**
     *
     * {@inheritDoc}
     */
    public void rollbackToSavepoint( Object savepoint )
    {
        for ( TransactionStatus ts : _transactionStatuses.values( ) )
        {
            ts.rollbackToSavepoint( savepoint );
        }
    }

    /**
     * "Getter method" for {@link #_transactionStatuses}
     * 
     * @return value of {@link #_transactionStatuses}
     */
    public Map<PlatformTransactionManager, TransactionStatus> getTransactionStatuses( )
    {
        return _transactionStatuses;
    }

    /**
     * "Setter method" for {@link #_transactionStatuses}
     * 
     * @param statuses
     *            value of {@link #_transactionStatuses}
     */
    public void setTransactionStatuses( Map<PlatformTransactionManager, TransactionStatus> statuses )
    {
        _transactionStatuses = statuses;
    }
}