/*
 *******************************************************************************
 * Copyright (c) 2017 Contributors to the Eclipse Foundation
 *
 * See the NOTICE file(s) distributed with this work for additional
 * information regarding copyright ownership.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * You may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 *******************************************************************************/
package org.eclipse.microprofile.fault.tolerance.tck.retrytimeout.clientserver;

import static org.testng.Assert.fail;

import javax.enterprise.context.RequestScoped;

import org.eclipse.microprofile.fault.tolerance.tck.util.TCKConfig;
import org.eclipse.microprofile.faulttolerance.Retry;
import org.eclipse.microprofile.faulttolerance.Timeout;
import org.eclipse.microprofile.faulttolerance.exceptions.BulkheadException;
import org.eclipse.microprofile.faulttolerance.exceptions.TimeoutException;

/**
 * A client to demonstrate the combination of the @Retry and @Timeout annotations.
 * 
 * @author <a href="mailto:[email protected]">Neil Young</a>
 *
 */
@RequestScoped
public class RetryTimeoutClient {

    private int counterForInvokingServiceA = 0;
    private int counterForInvokingServiceWithoutRetryOn = 0;
    private int counterForInvokingServiceWithAbortOn = 0;
    private TCKConfig config = TCKConfig.getConfig();

    public int getCounterForInvokingServiceA() {
        return counterForInvokingServiceA;
    }

    public int getCounterForInvokingServiceWithoutRetryOn() {
        return counterForInvokingServiceWithoutRetryOn;
    }

    public int getCounterForInvokingServiceWithAbortOn() {
        return counterForInvokingServiceWithAbortOn;
    }

    /**
     * Times out after 500ms, retries once
     * 
     * @param timeToSleep time this method should sleep for in ms 
     * @return {@code null}
     */
    @Timeout(500)
    @Retry(maxRetries = 1)
    public String serviceA(long timeToSleep) {
        try {
            counterForInvokingServiceA++;
            Thread.sleep(timeToSleep);
            throw new RuntimeException("Timeout did not interrupt");
        } 
        catch (InterruptedException e) {
            //expected
        }
        return null;
    }
    
    /**
     * Sleeps for 1000ms, times out after 500ms, retries once on BulkheadException
     * <p>
     * Method will never throw a BulkheadException so the Retry annotation should have no effect
     * 
     * @return {@code null}
     */
    @Timeout(500)
    @Retry(maxRetries = 1, retryOn = BulkheadException.class)
    public String serviceWithoutRetryOn() {
        try {
            counterForInvokingServiceWithoutRetryOn++;
            Thread.sleep(config.getTimeoutInMillis(1000));
            fail("Timeout did not interrupt");
        }
        catch (InterruptedException e) {
            // expected
        }
        return null;
    }
    
    /**
     * Sleeps for 1000ms, times out after 500ms, retries once on anything but TimeoutException
     * 
     * @return {@code null}
     */
    @Timeout(500)
    @Retry(maxRetries = 1, abortOn = TimeoutException.class)
    public String serviceWithAbortOn() {
        try {
            counterForInvokingServiceWithAbortOn++;
            Thread.sleep(config.getTimeoutInMillis(1000));
            fail("Timeout did not interrupt");
        }
        catch (InterruptedException e) {
            // expected
        }
        return null;
    }
}