/*
 * JBoss, Home of Professional Open Source.
 * Copyright 2010, Red Hat Middleware LLC, and individual contributors
 * as indicated by the @author tags. See the copyright.txt file in the
 * distribution for a full listing of individual contributors.
 *
 * This 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; either version 2.1 of
 * the License, or (at your option) any later version.
 *
 * This software 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 software; if not, write to the Free
 * Software Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA
 * 02110-1301 USA, or see the FSF site: http://www.fsf.org.
 */
package org.jboss.as.clustering.infinispan.invoker;

import static org.jboss.as.clustering.infinispan.InfinispanLogger.ROOT_LOGGER;
import static org.jboss.as.clustering.infinispan.InfinispanMessages.MESSAGES;

import java.util.Arrays;
import java.util.EnumSet;
import java.util.Set;

import javax.transaction.xa.XAException;

import org.infinispan.Cache;
import org.infinispan.CacheException;
import org.infinispan.context.Flag;
import org.infinispan.remoting.transport.jgroups.SuspectException;
import org.infinispan.util.concurrent.TimeoutException;

/**
 * A cache invoker implementation that retries after a specified set of intervals upon timeout or suspect.
 * If the invocation includes Flag.FAIL_SILENTLY, this will only be applied to the last attempt.
 * @author Paul Ferraro
 */
public class RetryingCacheInvoker implements CacheInvoker {

    private final CacheInvoker invoker;
    private final int[] backOffIntervals;

    /**
     * Creates a new RetryingCacheInvoker.
     *
     * @param backOffIntervals specifies the sleep intervals between retries, and implicitly, the number of retries
     */
    public RetryingCacheInvoker(int... backOffIntervals) {
        this(new SimpleCacheInvoker(), backOffIntervals);
    }

    public RetryingCacheInvoker(CacheInvoker invoker, int... backOffIntervals) {
        this.invoker = invoker;
        this.backOffIntervals = backOffIntervals;
    }

    /**
     * {@inheritDoc}
     *
     * @see org.jboss.as.clustering.infinispan.invoker.CacheInvoker#invoke(org.infinispan.Cache, org.jboss.as.clustering.infinispan.invoker.CacheInvoker.Operation)
     */
    @Override
    public <K, V, R> R invoke(Cache<K, V> cache, Operation<K, V, R> operation, Flag... allFlags) {
        Flag[] attemptFlags = null;
        // attemptFlags = allFlags - Flag.FAIL_SILENTLY
        if ((allFlags != null) && (allFlags.length > 0)) {
            Set<Flag> flags = EnumSet.noneOf(Flag.class);
            flags.addAll(Arrays.asList(allFlags));
            flags.remove(Flag.FAIL_SILENTLY);
            attemptFlags = flags.toArray(new Flag[flags.size()]);
        }

        Exception exception = null;

        for (int i = 0; i <= this.backOffIntervals.length; ++i) {
            // Make sure Flag.FAIL_SILENTLY, if specified, is applied to the last try only
            boolean retry = (i < this.backOffIntervals.length);
            try {
                return this.invoker.invoke(cache, operation, retry ? attemptFlags : allFlags);
            } catch (TimeoutException e) {
                exception = e;
            } catch (SuspectException e) {
                exception = e;
            } catch (CacheException e) {
                // If the tx commit (e.g. Cache.endBatch(true)) was rolled back due to failure on prepare, then retry
                Throwable cause = e.getCause();
                if (cause instanceof XAException) {
                    XAException xaCause = (XAException) cause;
                    // TimeoutException/SuspectException on prepare throws XAException.XA_RBROLLBACK
                    // See org.infinispan.transaction.TransactionCoordinator.prepare(...)
                    if (xaCause.errorCode == XAException.XA_RBROLLBACK) {
                        exception = e;
                    } else {
                        throw e;
                    }
                } else {
                    throw e;
                }
            }

            if (retry) {
                int delay = this.backOffIntervals[i];

                if (ROOT_LOGGER.isTraceEnabled()) {
                    ROOT_LOGGER.tracef(exception, "Cache operation failed.  Retrying in %d ms", delay);
                }

                if (delay > 0) {
                    try {
                        Thread.sleep(delay);
                    } catch (InterruptedException e) {
                        Thread.currentThread().interrupt();
                    }
                } else {
                    Thread.yield();
                }
                if (Thread.currentThread().isInterrupted()) break;
            }
        }

        throw MESSAGES.abortingCacheOperation(exception, this.backOffIntervals.length);
    }
}

@@ -1321,108 +1321,28 @@
 ort 
-javax.transaction.xa.XAException;%0A%0Aimport org.infinispan.Cache;%0Aimport org.infinispan.CacheException
+org.infinispan.Cache
 ;%0Aim
@@ -3595,790 +3595,8 @@
    %7D
- catch (CacheException e) %7B%0A                // If the tx commit (e.g. Cache.endBatch(true)) was rolled back due to failure on prepare, then retry%0A                Throwable cause = e.getCause();%0A                if (cause instanceof XAException) %7B%0A                    XAException xaCause = (XAException) cause;%0A                    // TimeoutException/SuspectException on prepare throws XAException.XA_RBROLLBACK%0A                    // See org.infinispan.transaction.TransactionCoordinator.prepare(...)%0A                    if (xaCause.errorCode == XAException.XA_RBROLLBACK) %7B%0A                        exception = e;%0A                    %7D else %7B%0A                        throw e;%0A                    %7D%0A                %7D else %7B%0A                    throw e;%0A                %7D%0A            %7D
 %0A%0A  

Remove logic to retry on tx commit on SuspectException. TX commit no longer throws SuspectException as of Infinispan 5.2.0.CR2.