/*
 * Copyright (c) MuleSoft, Inc.  All rights reserved.  http://www.mulesoft.com
 * The software in this package is published under the terms of the CPAL v1.0
 * license, a copy of which has been included with this distribution in the
 * LICENSE.txt file.
 */
package org.mule.runtime.core.execution;

import org.mule.runtime.core.api.execution.ExecutionCallback;
import org.mule.runtime.core.api.transaction.Transaction;
import org.mule.runtime.core.api.transaction.TransactionConfig;
import org.mule.runtime.core.transaction.TransactionCoordination;

public class IsolateCurrentTransactionInterceptor<T> implements ExecutionInterceptor<T> {

  private ExecutionInterceptor<T> next;
  private TransactionConfig transactionConfig;

  public IsolateCurrentTransactionInterceptor(ExecutionInterceptor<T> nextProcessingInterceptor,
                                              TransactionConfig transactionConfig) {
    this.next = nextProcessingInterceptor;
    this.transactionConfig = transactionConfig;
  }


  @Override
  public T execute(ExecutionCallback<T> muleEventProcessingCallback, ExecutionContext executionContext) throws Exception {
    boolean transactionIsolated = false;
    try {
      if (transactionConfig.getAction() == TransactionConfig.ACTION_NOT_SUPPORTED) {
        Transaction transaction = TransactionCoordination.getInstance().getTransaction();
        if (transaction != null) {
          TransactionCoordination.getInstance().isolateTransaction();
          transactionIsolated = true;
        }
      }
      return next.execute(muleEventProcessingCallback, executionContext);
    } finally {
      if (transactionIsolated) {
        TransactionCoordination.getInstance().restoreIsolatedTransaction();
      }
    }
  }
}