Support custom retryable exceptions via @RetryableException and SofaRpcException(#1362)

Author: wunameya

core-impl/client/src/main/java/com/alipay/sofa/rpc/client/FailoverCluster.java

@@ -67,6 +67,15 @@ public SofaResponse doInvoke(SofaRequest request) throws SofaRpcException {
                 providerInfo = select(request, invokedProviderInfos);
                 SofaResponse response = filterChain(providerInfo, request);
                 if (response != null) {
+                    // 检查是否是用户自定义的可重试异常
+                    Object appResponse = response.getAppResponse();
+                    if (appResponse instanceof SofaRpcException) {
+                        if (((SofaRpcException) appResponse).getErrorType() == RpcErrorType.CUSTOMER_RETRY_ERROR) {
+                            time++;
+                            throwable = (SofaRpcException) appResponse;
+                            continue;
+                        }
+                    }
                     if (throwable != null) {
                         if (LOGGER.isWarnEnabled(consumerConfig.getAppName())) {
                             LOGGER.warnWithApp(consumerConfig.getAppName(),

core/api/src/main/java/com/alipay/sofa/rpc/filter/ProviderInvoker.java

@@ -17,6 +17,7 @@
 package com.alipay.sofa.rpc.filter;
 
 import com.alipay.sofa.rpc.common.RpcConstants;
+import com.alipay.sofa.rpc.common.annotation.RetryableException;
 import com.alipay.sofa.rpc.config.ProviderConfig;
 import com.alipay.sofa.rpc.context.RpcInternalContext;
 import com.alipay.sofa.rpc.context.RpcInvokeContext;
@@ -113,7 +114,12 @@ public SofaResponse invoke(SofaRequest request) throws SofaRpcException {
             //            sofaResponse.setErrorMsg(e.getMessage());
         } catch (InvocationTargetException e) { // 业务代码抛出异常
             cutCause(e.getCause());
-            sofaResponse.setAppResponse(e.getCause());
+            // 查看该异常是否被用户声明为可重试异常
+            if (e.getCause().getClass().isAnnotationPresent(RetryableException.class)) {
+                sofaResponse.setAppResponse(new SofaRpcException(RpcErrorType.CUSTOMER_RETRY_ERROR, e.getCause()));
+            } else {
+                sofaResponse.setAppResponse(e.getCause());
+            }
         } finally {
             // R8: Record business processing execution time
             if (RpcInternalContext.isAttachmentEnable()) {

core/exception/src/main/java/com/alipay/sofa/rpc/core/exception/RpcErrorType.java

@@ -107,4 +107,9 @@ public class RpcErrorType {
      * 客户端未定义异常
      */
     public static final int CLIENT_UNDECLARED_ERROR  = 299;
+
+    /**
+     * 用户自定义可重试异常
+     */
+    public static final int CUSTOMER_RETRY_ERROR = 300;
 }