support consumer routers config

sofastack · May 16, 2024 · 27603ea · 27603ea
1 parent d6d5c85
commit 27603ea
Show file tree

Hide file tree

Showing 6 changed files with 46 additions and 4 deletions.
diff --git a/core/api/src/main/java/com/alipay/sofa/rpc/common/RpcOptions.java b/core/api/src/main/java/com/alipay/sofa/rpc/common/RpcOptions.java
@@ -380,6 +380,13 @@ public class RpcOptions {
      */
     public static final String CONSUMER_REJECTED_EXECUTION_POLICY       = "consumer.rejected.execution.policy";
 
+    /**
+     * 需要解析的 routers
+     *
+     * @since 5.13.0
+     */
+    public static final String CONSUMER_ROUTERS                         = "consumer.routers";
+
     /**
      * 默认回调线程池最小
      */

diff --git a/core/api/src/main/java/com/alipay/sofa/rpc/config/ConsumerConfig.java b/core/api/src/main/java/com/alipay/sofa/rpc/config/ConsumerConfig.java
@@ -33,10 +33,12 @@
 
 import java.io.Serializable;
 import java.lang.reflect.Method;
+import java.util.ArrayList;
 import java.util.List;
 
 import static com.alipay.sofa.rpc.common.RpcConfigs.getBooleanValue;
 import static com.alipay.sofa.rpc.common.RpcConfigs.getIntValue;
+import static com.alipay.sofa.rpc.common.RpcConfigs.getListValue;
 import static com.alipay.sofa.rpc.common.RpcConfigs.getStringValue;
 import static com.alipay.sofa.rpc.common.RpcOptions.CONSUMER_REJECTED_EXECUTION_POLICY;
 import static com.alipay.sofa.rpc.common.RpcOptions.CONSUMER_ADDRESS_HOLDER;
@@ -56,6 +58,7 @@
 import static com.alipay.sofa.rpc.common.RpcOptions.CONSUMER_RECONNECT_PERIOD;
 import static com.alipay.sofa.rpc.common.RpcOptions.CONSUMER_REPEATED_REFERENCE_LIMIT;
 import static com.alipay.sofa.rpc.common.RpcOptions.CONSUMER_RETRIES;
+import static com.alipay.sofa.rpc.common.RpcOptions.CONSUMER_ROUTERS;
 import static com.alipay.sofa.rpc.common.RpcOptions.CONSUMER_STICKY;
 import static com.alipay.sofa.rpc.common.RpcOptions.DEFAULT_PROTOCOL;
 
@@ -173,7 +176,8 @@ public class ConsumerConfig<T> extends AbstractInterfaceConfig<T, ConsumerConfig
     /**
      * 路由配置别名
      */
-    protected List<String>                          router;
+    protected List<String>                          router                  = new ArrayList<String>(
+                                                                                getListValue(CONSUMER_ROUTERS));
 
     /**
      * 路由规则引用，多个用英文逗号隔开。List<Router>
@@ -682,6 +686,18 @@ public ConsumerConfig<T> setRouter(List<String> router) {
         return this;
     }
 
+    /**
+     * Add router.
+     *
+     * @param router the add router
+     */
+    public void addRouter(List<String> router) {
+        if (this.router == null) {
+            this.router = new ArrayList<>();
+        }
+        this.router.addAll(router);
+    }
+
     /**
      * Gets routerRef.
      *

diff --git a/core/api/src/test/java/com/alipay/sofa/rpc/client/RouterChainTest.java b/core/api/src/test/java/com/alipay/sofa/rpc/client/RouterChainTest.java
@@ -48,7 +48,6 @@ public void buildProviderChain() {
         ConsumerConfig config = new ConsumerConfig();
         config.setBootstrap("test");
         ArrayList<Router> list = new ArrayList<Router>();
-        config.setRouter(Arrays.asList("testChainRouter0", "-testChainRouter8", "notExistChainRouter"));
         list.add(new TestChainRouter1());
         list.add(new TestChainRouter2());
         list.add(new TestChainRouter3());

diff --git a/core/api/src/test/java/com/alipay/sofa/rpc/std/config/ConsumerConfigTest.java b/core/api/src/test/java/com/alipay/sofa/rpc/std/config/ConsumerConfigTest.java
@@ -24,9 +24,13 @@
 import com.alipay.sofa.rpc.core.exception.SofaRpcRuntimeException;
 import com.alipay.sofa.rpc.std.sample.SampleService;
 import com.alipay.sofa.rpc.std.sample.SampleServiceImpl;
+import org.junit.Assert;
 import org.junit.Before;
 import org.junit.Test;
 
+import java.util.ArrayList;
+import java.util.List;
+
 /**
  * @author zhaowang
  * @version : ConsumerConfigTest.java, v 0.1 2022年01月28日 2:33 下午 zhaowang
@@ -59,7 +63,7 @@ public void testDefaultValue() {
         assertEquals(30000, config.getHeartbeatPeriod());
         assertEquals(10000, config.getReconnectPeriod());
         assertEquals("DISCARD", config.getRejectedExecutionPolicy());
-        assertEquals(null, config.getRouter());
+        assertNotNull(config.getRouter());
         assertEquals(null, config.getRouterRef());
         assertEquals(null, config.getOnReturn());
         assertEquals(null, config.getOnConnect());
@@ -119,6 +123,20 @@ public void testGetInterfaceId() {
         assertEquals("serviceName", config.getInterfaceId());
     }
 
+    @Test
+    public void testRouter() {
+        List<String> router = config.getRouter();
+        List<String> addRouter = new ArrayList<>();
+        addRouter.add("testRouter");
+        config.addRouter(addRouter);
+        assertSame(router, config.getRouter());
+        Assert.assertTrue(router.contains("testRouter"));
+
+        config.setRouter(addRouter);
+        assertNotSame(router, config.getRouter());
+        assertSame(addRouter, config.getRouter());
+    }
+
     public interface InnerInterface {
     }
 

diff --git a/core/api/src/test/resources/sofa-rpc/rpc-config.json b/core/api/src/test/resources/sofa-rpc/rpc-config.json
@@ -1,5 +1,6 @@
 {
   "rpc.config.order": 999,
   "logger.impl": "com.alipay.sofa.rpc.log.SystemLogger",
-  "default.filters" :["testChainFilter0", "-testChainFilter8"]
+  "default.filters" : ["testChainFilter0", "-testChainFilter8"],
+  "consumer.routers" : ["testChainRouter0", "-testChainRouter8", "notExistChainRouter"]
 }
diff --git a/core/common/src/main/resources/com/alipay/sofa/rpc/common/rpc-config-default.json b/core/common/src/main/resources/com/alipay/sofa/rpc/common/rpc-config-default.json
@@ -213,6 +213,7 @@ PS:大家也看到了，本JSON文档是支持注释的，而标准JSON是不支
   "consumer.connect.create.when.absent": true,
   // 默认回调线程池满时的拒绝策略，可用值：DISCARD, CALLER_RUNS, CALLER_HANDLE_EXCEPTION
   "consumer.rejected.execution.policy": "DISCARD",
+  "consumer.routers": [],
   /*-------------Consumer相关配置结束-------------*/