Examples with RetryException cn.taketoday.retry.RetryException used on opensource projects

Example 1 with RetryException

use of cn.taketoday.retry.RetryException in project today-infrastructure by TAKETODAY.

the class StatefulRecoveryRetryTests method testCacheCapacity.

@Test
public void testCacheCapacity() throws Throwable {
    this.retryTemplate.setRetryPolicy(new SimpleRetryPolicy(1));
    this.retryTemplate.setRetryContextCache(new MapRetryContextCache(1));
    RetryCallback<Object, Exception> callback = context -> {
        StatefulRecoveryRetryTests.this.count++;
        throw new RuntimeException("Barf!");
    };
    try {
        this.retryTemplate.execute(callback, new DefaultRetryState("foo"));
        fail("Expected RuntimeException");
    } catch (RuntimeException e) {
        assertEquals("Barf!", e.getMessage());
    }
    try {
        this.retryTemplate.execute(callback, new DefaultRetryState("bar"));
        fail("Expected RetryException");
    } catch (RetryException e) {
        String message = e.getNestedMessage();
        assertTrue("Message does not contain 'capacity': " + message, message.contains("capacity"));
    }
}

Example 2 with RetryException

use of cn.taketoday.retry.RetryException in project today-framework by TAKETODAY.

the class RetryTemplate method open.

/**
 * Delegate to the {@link RetryPolicy} having checked in the cache for an existing
 * value if the state is not null.
 *
 * @param state a {@link RetryState}
 * @param retryPolicy a {@link RetryPolicy} to delegate the context creation
 * @return a retry context, either a new one or the one used last time the same state
 * was encountered
 */
protected RetryContext open(RetryPolicy retryPolicy, RetryState state) {
    if (state == null) {
        return doOpenInternal(retryPolicy);
    }
    Object key = state.getKey();
    if (state.isForceRefresh()) {
        return doOpenInternal(retryPolicy, state);
    }
    // cache re-hydration.
    if (!retryContextCache.containsKey(key)) {
        // The cache is only used if there is a failure.
        return doOpenInternal(retryPolicy, state);
    }
    RetryContext context = retryContextCache.get(key);
    if (context == null) {
        if (retryContextCache.containsKey(key)) {
            throw new RetryException("Inconsistent state for failed item: no history found. " + "Consider whether equals() or hashCode() for the item might be inconsistent, " + "or if you need to supply a better ItemKeyGenerator");
        }
        // containsKey(), so we have to live with this:
        return doOpenInternal(retryPolicy, state);
    }
    // Start with a clean slate for state that others may be inspecting
    context.removeAttribute(RetryContext.CLOSED);
    context.removeAttribute(RetryContext.EXHAUSTED);
    context.removeAttribute(RetryContext.RECOVERED);
    return context;
}

Example 3 with RetryException

use of cn.taketoday.retry.RetryException in project today-framework by TAKETODAY.

the class StatefulRecoveryRetryTests method testKeyGeneratorNotConsistentAfterFailure.

@Test
public void testKeyGeneratorNotConsistentAfterFailure() throws Throwable {
    RetryPolicy retryPolicy = new SimpleRetryPolicy(3);
    this.retryTemplate.setRetryPolicy(retryPolicy);
    final StringHolder item = new StringHolder("bar");
    RetryState state = new DefaultRetryState(item);
    RetryCallback<StringHolder, Exception> callback = new RetryCallback<StringHolder, Exception>() {

        @Override
        public StringHolder doWithRetry(RetryContext context) throws Exception {
            // This simulates what happens if someone uses a primary key
            // for hashCode and equals and then relies on default key
            // generator
            item.string = item.string + (StatefulRecoveryRetryTests.this.count++);
            throw new RuntimeException("Barf!");
        }
    };
    try {
        this.retryTemplate.execute(callback, state);
        fail("Expected RuntimeException");
    } catch (RuntimeException ex) {
        String message = ex.getMessage();
        assertEquals("Barf!", message);
    }
    // item already...
    try {
        this.retryTemplate.execute(callback, state);
        fail("Expected RetryException");
    } catch (RetryException ex) {
        String message = ex.getNestedMessage();
        assertTrue("Message doesn't contain 'inconsistent': " + message, message.contains("inconsistent"));
    }
    RetryContext context = this.retryTemplate.open(retryPolicy, state);
    // True after exhausted - the history is reset...
    assertEquals(0, context.getRetryCount());
}

Example 4 with RetryException

use of cn.taketoday.retry.RetryException in project today-framework by TAKETODAY.

the class StatefulRecoveryRetryTests method testCacheCapacity.

@Test
public void testCacheCapacity() throws Throwable {
    this.retryTemplate.setRetryPolicy(new SimpleRetryPolicy(1));
    this.retryTemplate.setRetryContextCache(new MapRetryContextCache(1));
    RetryCallback<Object, Exception> callback = context -> {
        StatefulRecoveryRetryTests.this.count++;
        throw new RuntimeException("Barf!");
    };
    try {
        this.retryTemplate.execute(callback, new DefaultRetryState("foo"));
        fail("Expected RuntimeException");
    } catch (RuntimeException e) {
        assertEquals("Barf!", e.getMessage());
    }
    try {
        this.retryTemplate.execute(callback, new DefaultRetryState("bar"));
        fail("Expected RetryException");
    } catch (RetryException e) {
        String message = e.getNestedMessage();
        assertTrue("Message does not contain 'capacity': " + message, message.contains("capacity"));
    }
}

Example 5 with RetryException

use of cn.taketoday.retry.RetryException in project today-infrastructure by TAKETODAY.

the class RetryTemplate method open.

/**
 * Delegate to the {@link RetryPolicy} having checked in the cache for an existing
 * value if the state is not null.
 *
 * @param state a {@link RetryState}
 * @param retryPolicy a {@link RetryPolicy} to delegate the context creation
 * @return a retry context, either a new one or the one used last time the same state
 * was encountered
 */
protected RetryContext open(RetryPolicy retryPolicy, RetryState state) {
    if (state == null) {
        return doOpenInternal(retryPolicy);
    }
    Object key = state.getKey();
    if (state.isForceRefresh()) {
        return doOpenInternal(retryPolicy, state);
    }
    // cache re-hydration.
    if (!retryContextCache.containsKey(key)) {
        // The cache is only used if there is a failure.
        return doOpenInternal(retryPolicy, state);
    }
    RetryContext context = retryContextCache.get(key);
    if (context == null) {
        if (retryContextCache.containsKey(key)) {
            throw new RetryException("Inconsistent state for failed item: no history found. " + "Consider whether equals() or hashCode() for the item might be inconsistent, " + "or if you need to supply a better ItemKeyGenerator");
        }
        // containsKey(), so we have to live with this:
        return doOpenInternal(retryPolicy, state);
    }
    // Start with a clean slate for state that others may be inspecting
    context.removeAttribute(RetryContext.CLOSED);
    context.removeAttribute(RetryContext.EXHAUSTED);
    context.removeAttribute(RetryContext.RECOVERED);
    return context;
}