Search in sources :

Example 16 with MockInvoker

use of org.apache.dubbo.rpc.cluster.router.MockInvoker in project dubbo by alibaba.

the class ScriptRouterTest method testRoute_throwException.

@Test
public void testRoute_throwException() {
    List<Invoker<String>> invokers = new ArrayList<Invoker<String>>();
    MockInvoker<String> invoker1 = new MockInvoker<String>(URL.valueOf("dubbo://10.134.108.1:20880/com.dubbo.HelloService"));
    MockInvoker<String> invoker2 = new MockInvoker<String>(URL.valueOf("dubbo://10.134.108.2:20880/com.dubbo.HelloService"));
    MockInvoker<String> invoker3 = new MockInvoker<String>(URL.valueOf("dubbo://10.134.108.3:20880/com.dubbo.HelloService"));
    invokers.add(invoker1);
    invokers.add(invoker2);
    invokers.add(invoker3);
    String script = "/";
    Router router = new ScriptRouterFactory().getRouter(getRouteUrl(script));
    List<Invoker<String>> routeResult = router.route(invokers, invokers.get(0).getUrl(), new RpcInvocation());
    Assertions.assertEquals(3, routeResult.size());
}
Also used : RpcInvocation(org.apache.dubbo.rpc.RpcInvocation) Invoker(org.apache.dubbo.rpc.Invoker) MockInvoker(org.apache.dubbo.rpc.cluster.router.MockInvoker) MockInvoker(org.apache.dubbo.rpc.cluster.router.MockInvoker) ArrayList(java.util.ArrayList) Router(org.apache.dubbo.rpc.cluster.Router) Test(org.junit.jupiter.api.Test)

Example 17 with MockInvoker

use of org.apache.dubbo.rpc.cluster.router.MockInvoker in project dubbo by alibaba.

the class ScriptRouterTest method testRouteHostFilter.

@Test
public void testRouteHostFilter() {
    List<Invoker<String>> invokers = new ArrayList<Invoker<String>>();
    MockInvoker<String> invoker1 = new MockInvoker<String>(URL.valueOf("dubbo://10.134.108.1:20880/com.dubbo.HelloService"));
    MockInvoker<String> invoker2 = new MockInvoker<String>(URL.valueOf("dubbo://10.134.108.2:20880/com.dubbo.HelloService"));
    MockInvoker<String> invoker3 = new MockInvoker<String>(URL.valueOf("dubbo://10.134.108.3:20880/com.dubbo.HelloService"));
    invokers.add(invoker1);
    invokers.add(invoker2);
    invokers.add(invoker3);
    String script = "function route(invokers, invocation, context){ " + "	var result = new java.util.ArrayList(invokers.size()); " + "	var targetHost = new java.util.ArrayList(); " + "	targetHost.add(\"10.134.108.2\"); " + "	for (var i = 0; i < invokers.length; i++) { " + "		if(targetHost.contains(invokers[i].getUrl().getHost())){ " + "			result.add(invokers[i]); " + "		} " + "	} " + "	return result; " + "} " + "route(invokers, invocation, context) ";
    Router router = new ScriptRouterFactory().getRouter(getRouteUrl(script));
    List<Invoker<String>> routeResult = router.route(invokers, invokers.get(0).getUrl(), new RpcInvocation());
    Assertions.assertEquals(1, routeResult.size());
    Assertions.assertEquals(invoker2, routeResult.get(0));
}
Also used : RpcInvocation(org.apache.dubbo.rpc.RpcInvocation) Invoker(org.apache.dubbo.rpc.Invoker) MockInvoker(org.apache.dubbo.rpc.cluster.router.MockInvoker) MockInvoker(org.apache.dubbo.rpc.cluster.router.MockInvoker) ArrayList(java.util.ArrayList) Router(org.apache.dubbo.rpc.cluster.Router) Test(org.junit.jupiter.api.Test)

Example 18 with MockInvoker

use of org.apache.dubbo.rpc.cluster.router.MockInvoker in project dubbo by alibaba.

the class ScriptRouterTest method testRoutePickInvokers.

@Test
public void testRoutePickInvokers() {
    String rule = "var result = new java.util.ArrayList(invokers.size());" + "for (i=0;i<invokers.size(); i++){ " + "if (invokers.get(i).isAvailable()) {" + "result.add(invokers.get(i)) ;" + "}" + "} ; " + "return result;";
    String script = "function route(invokers,invocation,context){" + rule + "} route(invokers,invocation,context)";
    Router router = new ScriptRouterFactory().getRouter(getRouteUrl(script));
    List<Invoker<String>> invokers = new ArrayList<Invoker<String>>();
    Invoker<String> invoker1 = new MockInvoker<String>(false);
    Invoker<String> invoker2 = new MockInvoker<String>(true);
    Invoker<String> invoker3 = new MockInvoker<String>(true);
    invokers.add(invoker1);
    invokers.add(invoker2);
    invokers.add(invoker3);
    List<Invoker<String>> filteredInvokers = router.route(invokers, invokers.get(0).getUrl(), new RpcInvocation());
    Assertions.assertEquals(2, filteredInvokers.size());
    Assertions.assertEquals(invoker2, filteredInvokers.get(0));
    Assertions.assertEquals(invoker3, filteredInvokers.get(1));
}
Also used : RpcInvocation(org.apache.dubbo.rpc.RpcInvocation) Invoker(org.apache.dubbo.rpc.Invoker) MockInvoker(org.apache.dubbo.rpc.cluster.router.MockInvoker) MockInvoker(org.apache.dubbo.rpc.cluster.router.MockInvoker) ArrayList(java.util.ArrayList) Router(org.apache.dubbo.rpc.cluster.Router) Test(org.junit.jupiter.api.Test)

Aggregations

ArrayList (java.util.ArrayList)18 Invoker (org.apache.dubbo.rpc.Invoker)18 RpcInvocation (org.apache.dubbo.rpc.RpcInvocation)18 Router (org.apache.dubbo.rpc.cluster.Router)18 MockInvoker (org.apache.dubbo.rpc.cluster.router.MockInvoker)18 Test (org.junit.jupiter.api.Test)18 Invocation (org.apache.dubbo.rpc.Invocation)2 URL (org.apache.dubbo.common.URL)1 ConditionRouterFactory (org.apache.dubbo.rpc.cluster.router.condition.ConditionRouterFactory)1