#31348905

Author: panxiaojun233

dubbo-cluster/src/main/java/org/apache/dubbo/rpc/cluster/RouterChain.java

@@ -83,6 +83,10 @@ public static <T> RouterChain<T> buildChain(URL url) {
         return new RouterChain<>(url);
     }
 
+    public RouterChain(List<Invoker<T>> invokers) {
+        this.invokers = invokers;
+    }
+
     private RouterChain(URL url) {
         List<RouterFactory> extensionFactories = ExtensionLoader.getExtensionLoader(RouterFactory.class)
                 .getActivateExtension(url, "router");
@@ -146,6 +150,14 @@ public void addRouters(List<Router> routers) {
         this.routers = newRouters;
     }
 
+    public void addStateRouters(List<StateRouter> stateRouters) {
+        List<StateRouter> newStateRouters = new ArrayList<>();
+        newStateRouters.addAll(builtinStateRouters);
+        newStateRouters.addAll(stateRouters);
+        CollectionUtils.sort(newStateRouters);
+        this.stateRouters = newStateRouters;
+    }
+
     private void sort() {
         Collections.sort(routers);
     }

dubbo-cluster/src/main/java/org/apache/dubbo/rpc/cluster/router/StateRouterTest.java

@@ -0,0 +1,46 @@
+package org.apache.dubbo.rpc.cluster.router;
+
+import java.util.ArrayList;
+import java.util.List;
+
+import org.apache.dubbo.common.URL;
+import org.apache.dubbo.rpc.Invocation;
+import org.apache.dubbo.rpc.Invoker;
+import org.apache.dubbo.rpc.cluster.RouterChain;
+import org.apache.dubbo.rpc.cluster.router.state.StateRouter;
+import org.apache.dubbo.rpc.cluster.router.tag.TagDynamicStateRouter;
+
+public class StateRouterTest {
+    public static void main(String[] args) {
+        List<Invoker> list = mockInvokers();
+
+        RouterChain chain = new RouterChain(list);
+        List<StateRouter> routers = new ArrayList<>();
+        routers.add(new TagDynamicStateRouter(null));
+        chain.addStateRouters(routers);
+
+        URL url = mockURL();
+        Invocation invocation = mockInvocation();
+        long start = System.currentTimeMillis();
+        chain.route(url, invocation);
+        System.out.println("cost:" + (System.currentTimeMillis() - start));
+    }
+
+    private static List<Invoker> mockInvokers() {
+        List<Invoker> list = new ArrayList<>();
+
+
+
+        return list;
+    }
+
+    private static URL mockURL() {
+
+        return null;
+    }
+
+    private static Invocation mockInvocation() {
+
+        return null;
+    }
+}