Skip to content

Commit

Permalink
spring-projectsGH-299: Fix @recover with Class Level @retryable
Browse files Browse the repository at this point in the history
Resolves spring-projects#299

Caused by the fix for spring-projects#244, recover methods should not be considered
as retryable, with class level annotation.

**cherry-pick to 1.3.x**
  • Loading branch information
garyrussell committed May 18, 2022
1 parent 47c1e52 commit 4343501
Show file tree
Hide file tree
Showing 2 changed files with 95 additions and 2 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -174,7 +174,7 @@ private MethodInterceptor getDelegate(Object target, Method method) {
MethodInterceptor interceptor = NULL_INTERCEPTOR;
Retryable retryable = AnnotatedElementUtils.findMergedAnnotation(method, Retryable.class);
if (retryable == null) {
retryable = AnnotatedElementUtils.findMergedAnnotation(method.getDeclaringClass(), Retryable.class);
retryable = classLevelAnnotation(method, Retryable.class);
}
if (retryable == null) {
retryable = findAnnotationOnTarget(target, method, Retryable.class);
Expand Down Expand Up @@ -203,7 +203,7 @@ private <A extends Annotation> A findAnnotationOnTarget(Object target, Method me
Method targetMethod = target.getClass().getMethod(method.getName(), method.getParameterTypes());
A retryable = AnnotatedElementUtils.findMergedAnnotation(targetMethod, annotation);
if (retryable == null) {
retryable = AnnotatedElementUtils.findMergedAnnotation(targetMethod.getDeclaringClass(), annotation);
retryable = classLevelAnnotation(targetMethod, annotation);
}

return retryable;
Expand All @@ -213,6 +213,17 @@ private <A extends Annotation> A findAnnotationOnTarget(Object target, Method me
}
}

/*
* With a class level annotation, exclude @Recover methods.
*/
private <A extends Annotation> A classLevelAnnotation(Method method, Class<A> annotation) {
A ann = AnnotatedElementUtils.findMergedAnnotation(method.getDeclaringClass(), annotation);
if (ann != null && AnnotatedElementUtils.findMergedAnnotation(method, Recover.class) != null) {
ann = null;
}
return ann;
}

private MethodInterceptor getStatelessInterceptor(Object target, Method method, Retryable retryable) {
RetryTemplate template = createTemplate(retryable.listeners());
template.setRetryPolicy(getRetryPolicy(retryable, true));
Expand Down
Original file line number Diff line number Diff line change
@@ -0,0 +1,82 @@
/*
* Copyright 2019 the original author or authors.
*
* 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
*
* https://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.springframework.retry.annotation;

import org.junit.jupiter.api.Test;

import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.test.context.junit.jupiter.SpringJUnitConfig;

import static org.assertj.core.api.Assertions.assertThat;
import static org.assertj.core.api.Assertions.assertThatExceptionOfType;

/**
* @author Gary Russell
* @since 1.3.4
*/
@SpringJUnitConfig
public class DontRetryRecovererTests {

@Test
void dontRetry(@Autowired Service service) {
assertThatExceptionOfType(RuntimeException.class).isThrownBy(() -> service.foo("x")).withMessage("test");
assertThat(service.getCallCount()).isEqualTo(3);
assertThat(service.getRecoverCount()).isEqualTo(1);
}

@Configuration
@EnableRetry
public static class Config {

@Bean
Service service() {
return new Service();
}

}

@Retryable
public static class Service {

int callCount;

int recoverCount;

public void foo(String in) {
callCount++;
throw new RuntimeException();
}

@Recover
public void recover(Exception ex, String in) {
this.recoverCount++;
throw new RuntimeException("test");
}

public int getCallCount() {
return callCount;
}

public int getRecoverCount() {
return recoverCount;
}

}

}

0 comments on commit 4343501

Please sign in to comment.