com.github.lianjiatech.retrofit.spring.boot.retry.RetryInterceptor Maven / Gradle / Ivy
package com.github.lianjiatech.retrofit.spring.boot.retry;
import com.github.lianjiatech.retrofit.spring.boot.exception.RetryFailedException;
import com.github.lianjiatech.retrofit.spring.boot.util.AnnotationExtendUtils;
import lombok.extern.slf4j.Slf4j;
import okhttp3.Interceptor;
import okhttp3.Request;
import okhttp3.Response;
import retrofit2.Invocation;
import java.io.IOException;
import java.util.Arrays;
import java.util.HashSet;
import java.util.stream.Collectors;
/**
* @author 陈添明
*/
@Slf4j
public class RetryInterceptor implements Interceptor {
protected final GlobalRetryProperty globalRetryProperty;
public RetryInterceptor(GlobalRetryProperty globalRetryProperty) {
this.globalRetryProperty = globalRetryProperty;
}
@Override
public Response intercept(Chain chain) throws IOException {
Request request = chain.request();
Invocation invocation = request.tag(Invocation.class);
if (invocation == null) {
return chain.proceed(request);
}
// 获取重试配置
Retry retry = AnnotationExtendUtils.findMergedAnnotation(invocation.method(), invocation.service(), Retry.class);
if (!needRetry(retry)) {
return chain.proceed(request);
}
// 重试
int maxRetries = retry == null ? globalRetryProperty.getMaxRetries() : retry.maxRetries();
int intervalMs = retry == null ? globalRetryProperty.getIntervalMs() : retry.intervalMs();
RetryRule[] retryRules = retry == null ? globalRetryProperty.getRetryRules() : retry.retryRules();
return retryIntercept(maxRetries, intervalMs, retryRules, chain);
}
protected boolean needRetry(Retry retry) {
if (globalRetryProperty.isEnable()) {
if (retry == null) {
return true;
}
return retry.enable();
} else {
return retry != null && retry.enable();
}
}
protected Response retryIntercept(int maxRetries, int intervalMs, RetryRule[] retryRules, Chain chain) {
HashSet retryRuleSet = (HashSet)Arrays.stream(retryRules).collect(Collectors.toSet());
RetryStrategy retryStrategy = new RetryStrategy(maxRetries, intervalMs);
Request request = chain.request();
while (true) {
try {
Response response = chain.proceed(request);
// 如果响应状态码是2xx就不用重试，直接返回 response
if (!retryRuleSet.contains(RetryRule.RESPONSE_STATUS_NOT_2XX) || response.isSuccessful()) {
return response;
} else {
if (!retryStrategy.shouldRetry()) {
// 最后一次还没成功，返回最后一次response
return response;
}
// 执行重试
retryStrategy.retry();
log.warn("The response fails, retry is performed! The request is {}, Response is {}", request,
response);
response.close();
}
} catch (Exception e) {
if (shouldThrowEx(retryRuleSet, e)) {
throw new RuntimeException(e);
} else {
if (!retryStrategy.shouldRetry()) {
// 最后一次还没成功，抛出异常
throw new RetryFailedException(
"Retry Failed: Total " + maxRetries + " attempts made at interval " + intervalMs + "ms",
e);
}
retryStrategy.retry();
log.warn("The response fails, retry is performed! The request is {} ", request, e);
}
}
}
}
protected boolean shouldThrowEx(HashSet retryRuleSet, Exception e) {
if (retryRuleSet.contains(RetryRule.OCCUR_EXCEPTION)) {
return false;
}
if (retryRuleSet.contains(RetryRule.OCCUR_IO_EXCEPTION)) {
return !(e instanceof IOException);
}
return true;
}
}