feat: add business error code (#199)

Author: MilletBo

src/main/java/io/github/doocs/im/ClientConfiguration.java

@@ -3,7 +3,10 @@
 import io.github.doocs.im.util.VersionInfoUtil;
 import okhttp3.ConnectionPool;
 
+import java.util.Collections;
+import java.util.HashSet;
 import java.util.Objects;
+import java.util.Set;
 
 /**
  * 客户端配置类
@@ -36,6 +39,17 @@ public class ClientConfiguration {
      * 默认业务错误码重试开关关闭
      */
     public static final boolean DEFAULT_ENABLE_BUSINESS_RETRY = false;
+
+    /**
+     * 默认重试错误码
+     */
+    public static final Set<Integer> DEFAULT_BUSINESS_RETRY_CODES =
+            Collections.unmodifiableSet(new HashSet<Integer>() {{
+                add(10002);
+                add(20004);
+                add(20005);
+            }});
+
     /**
      * 默认超时时间（毫秒）
      */
@@ -63,14 +77,15 @@ public class ClientConfiguration {
     private long expireTime = DEFAULT_EXPIRE_TIME;
     private boolean autoRenewSig = DEFAULT_RENEW_SIG;
     private boolean enableBusinessRetry = DEFAULT_ENABLE_BUSINESS_RETRY;
+    private Set<Integer> businessRetryCodes = DEFAULT_BUSINESS_RETRY_CODES;
     private String userAgent = DEFAULT_USER_AGENT;
     private ConnectionPool connectionPool = DEFAULT_CONNECTION_POOL;
 
     public ClientConfiguration() {
     }
 
     public ClientConfiguration(int maxRetries, long retryIntervalMs, long connectTimeout, long readTimeout, long writeTimeout,
-                               long callTimeout, long expireTime, boolean autoRenewSig, boolean enableBusinessRetry,
+                               long callTimeout, long expireTime, boolean autoRenewSig, boolean enableBusinessRetry, Set<Integer> businessRetryCodes,
                                String userAgent, ConnectionPool connectionPool) {
         if (connectionPool == null) {
             connectionPool = DEFAULT_CONNECTION_POOL;
@@ -84,6 +99,7 @@ public ClientConfiguration(int maxRetries, long retryIntervalMs, long connectTim
         this.expireTime = expireTime;
         this.autoRenewSig = autoRenewSig;
         this.enableBusinessRetry = enableBusinessRetry;
+        this.businessRetryCodes = businessRetryCodes;
         this.userAgent = userAgent;
         this.connectionPool = connectionPool;
     }
@@ -98,6 +114,7 @@ private ClientConfiguration(Builder builder) {
         this.expireTime = builder.expireTime;
         this.autoRenewSig = builder.autoRenewSig;
         this.enableBusinessRetry = builder.enableBusinessRetry;
+        this.businessRetryCodes = builder.businessRetryCodes;
         this.userAgent = builder.userAgent;
         this.connectionPool = builder.connectionPool;
     }
@@ -178,6 +195,14 @@ public void setEnableBusinessRetry(boolean enableBusinessRetry) {
         this.enableBusinessRetry = enableBusinessRetry;
     }
 
+    public Set<Integer> getBusinessRetryCodes() {
+        return businessRetryCodes;
+    }
+
+    public void setBusinessRetryCodes(Set<Integer> businessRetryCodes) {
+        this.businessRetryCodes = businessRetryCodes;
+    }
+
     public String getUserAgent() {
         return userAgent;
     }
@@ -234,6 +259,9 @@ public boolean equals(Object o) {
         if (enableBusinessRetry != that.enableBusinessRetry) {
             return false;
         }
+        if (businessRetryCodes != that.businessRetryCodes) {
+            return false;
+        }
         if (!userAgent.equals(that.userAgent)) {
             return false;
         }
@@ -242,7 +270,7 @@ public boolean equals(Object o) {
 
     @Override
     public int hashCode() {
-        return Objects.hash(maxRetries, retryIntervalMs, connectTimeout, readTimeout, writeTimeout, callTimeout, expireTime, autoRenewSig, enableBusinessRetry, userAgent, connectionPool);
+        return Objects.hash(maxRetries, retryIntervalMs, connectTimeout, readTimeout, writeTimeout, callTimeout, expireTime, autoRenewSig, enableBusinessRetry, businessRetryCodes, userAgent, connectionPool);
     }
 
     public static final class Builder {
@@ -255,6 +283,7 @@ public static final class Builder {
         private long expireTime = DEFAULT_EXPIRE_TIME;
         private boolean autoRenewSig = DEFAULT_RENEW_SIG;
         private boolean enableBusinessRetry = DEFAULT_ENABLE_BUSINESS_RETRY;
+        private Set<Integer> businessRetryCodes = DEFAULT_BUSINESS_RETRY_CODES;
         private String userAgent = DEFAULT_USER_AGENT;
         private ConnectionPool connectionPool = DEFAULT_CONNECTION_POOL;
 
@@ -310,6 +339,11 @@ public Builder enableBusinessRetry(boolean enableBusinessRetry) {
             return this;
         }
 
+        public Builder businessRetryCodes(Set<Integer> businessRetryCodes) {
+            this.businessRetryCodes = businessRetryCodes;
+            return this;
+        }
+
         public Builder userAgent(String userAgent) {
             this.userAgent = userAgent;
             return this;

src/main/java/io/github/doocs/im/util/HttpUtil.java

@@ -32,7 +32,7 @@ public class HttpUtil {
             .writeTimeout(DEFAULT_CONFIG.getWriteTimeout(), TimeUnit.MILLISECONDS)
             .callTimeout(DEFAULT_CONFIG.getCallTimeout(), TimeUnit.MILLISECONDS)
             .retryOnConnectionFailure(false)
-            .addInterceptor(new RetryInterceptor(DEFAULT_CONFIG.getMaxRetries(), DEFAULT_CONFIG.getRetryIntervalMs(), null, DEFAULT_CONFIG.isEnableBusinessRetry()))
+            .addInterceptor(new RetryInterceptor(DEFAULT_CONFIG.getMaxRetries(), DEFAULT_CONFIG.getRetryIntervalMs(), DEFAULT_CONFIG.getBusinessRetryCodes(), DEFAULT_CONFIG.isEnableBusinessRetry()))
             .build();
 
     private HttpUtil() {
@@ -59,7 +59,7 @@ private static OkHttpClient getClient(ClientConfiguration config) {
                 .writeTimeout(cfg.getWriteTimeout(), TimeUnit.MILLISECONDS)
                 .callTimeout(cfg.getCallTimeout(), TimeUnit.MILLISECONDS)
                 .retryOnConnectionFailure(false)
-                .addInterceptor(new RetryInterceptor(cfg.getMaxRetries(), cfg.getRetryIntervalMs(), null, DEFAULT_CONFIG.isEnableBusinessRetry()))
+                .addInterceptor(new RetryInterceptor(cfg.getMaxRetries(), cfg.getRetryIntervalMs(), DEFAULT_CONFIG.getBusinessRetryCodes(), DEFAULT_CONFIG.isEnableBusinessRetry()))
                 .build());
     }