org.bonitasoft.engine.transaction.JTATransactionWrapper Maven / Gradle / Ivy

Go to download
/**
 * Copyright (C) 2013 BonitaSoft S.A.
 * BonitaSoft, 32 rue Gustave Eiffel - 38000 Grenoble
 * This library is free software; you can redistribute it and/or modify it under the terms
 * of the GNU Lesser General Public License as published by the Free Software Foundation
 * version 2.1 of the License.
 * This library is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY;
 * without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.
 * See the GNU Lesser General Public License for more details.
 * You should have received a copy of the GNU Lesser General Public License along with this
 * program; if not, write to the Free Software Foundation, Inc., 51 Franklin Street, Fifth
 * Floor, Boston, MA 02110-1301, USA.
 **/
package org.bonitasoft.engine.transaction;

import javax.transaction.Status;
import javax.transaction.Synchronization;

public class JTATransactionWrapper implements Synchronization {

    private final BonitaTransactionSynchronization bonitaTx;

    public JTATransactionWrapper(final BonitaTransactionSynchronization bonitaTx) {
        this.bonitaTx = bonitaTx;
    }

    @Override
    public void beforeCompletion() {
        bonitaTx.beforeCommit();
    }

    @Override
    public void afterCompletion(final int status) {
        bonitaTx.afterCompletion(convert(status));
    }

    private TransactionState convert(final int status) {
        switch (status) {
            case Status.STATUS_COMMITTED:
                return TransactionState.COMMITTED;
            case Status.STATUS_ROLLEDBACK:
                return TransactionState.ROLLEDBACK;
        }
        return null;
    }

}