/* * Copyright 1999-2011 Alibaba Group. * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ package com.alibaba.dubbo.rpc.protocol.rmi; import static junit.framework.Assert.assertEquals; import org.junit.Ignore; import org.junit.Test; import com.alibaba.dubbo.common.URL; import com.alibaba.dubbo.common.extension.ExtensionLoader; import com.alibaba.dubbo.rpc.Exporter; import com.alibaba.dubbo.rpc.Protocol; import com.alibaba.dubbo.rpc.ProxyFactory; import com.alibaba.dubbo.rpc.RpcException; import com.alibaba.dubbo.rpc.service.EchoService; public class RmiProtocolTest { private Protocol protocol = ExtensionLoader.getExtensionLoader(Protocol.class).getAdaptiveExtension(); private ProxyFactory proxy = ExtensionLoader.getExtensionLoader(ProxyFactory.class).getAdaptiveExtension(); public static interface NonStdRmiInterface { void bark(); } /* @Test public void test_getRemoteClass() throws Exception { Class<NonStdRmiInterface> clazz = RmiProtocol.getRemoteClass(NonStdRmiInterface.class); assertEquals(clazz, RmiProtocol.getRemoteClass(NonStdRmiInterface.class)); } */ @Test public void testRmiProtocolTimeout() throws Exception { System.setProperty("sun.rmi.transport.tcp.responseTimeout", "1000"); DemoService service = new DemoServiceImpl(); Exporter<?> rpcExporter = protocol.export(proxy.getInvoker(service, DemoService.class, URL.valueOf("rmi://127.0.0.1:9001/TestService"))); service = proxy.getProxy(protocol.refer(DemoService.class, URL.valueOf("rmi://127.0.0.1:9001/TestService"))); try { try { service.throwTimeout(); } catch (RpcException e) { assertEquals(true, e.isTimeout()); assertEquals(true, e.getMessage().contains("Read timed out")); } } finally { rpcExporter.unexport(); } } @Test public void testRmiProtocol() throws Exception { { DemoService service = new DemoServiceImpl(); Exporter<?> rpcExporter = protocol.export(proxy.getInvoker(service, DemoService.class, URL.valueOf("rmi://127.0.0.1:9001/TestService"))); service = proxy.getProxy(protocol.refer(DemoService.class, URL.valueOf("rmi://127.0.0.1:9001/TestService"))); assertEquals(service.getSize(null), -1); assertEquals(service.getSize(new String[]{"", "", ""}), 3); Object result = service.invoke("rmi://127.0.0.1:9001/TestService", "invoke"); assertEquals("rmi://127.0.0.1:9001/TestService:invoke", result); rpcExporter.unexport(); } { RemoteService remoteService = new RemoteServiceImpl(); Exporter<?> rpcExporter = protocol.export(proxy.getInvoker(remoteService, RemoteService.class, URL.valueOf("rmi://127.0.0.1:9002/remoteService"))); remoteService = proxy.getProxy(protocol.refer(RemoteService.class, URL.valueOf("rmi://127.0.0.1:9002/remoteService"))); remoteService.getThreadName(); for(int i=0;i<100;i++) { String say = remoteService.sayHello("abcd"); assertEquals("hello abcd@" + RemoteServiceImpl.class.getName(), say); } rpcExporter.unexport(); } } // FIXME RMI协议目前的实现不支持转型成 EchoService @Ignore @Test public void testRmiProtocol_echoService() throws Exception { DemoService service = new DemoServiceImpl(); Exporter<?> rpcExporter = protocol.export(proxy.getInvoker(service, DemoService.class, URL.valueOf("rmi://127.0.0.1:9002/TestService"))); // cast to EchoService EchoService echo = proxy.getProxy(protocol.refer(EchoService.class, URL.valueOf("rmi://127.0.0.1:9002/TestService"))); assertEquals(echo.$echo("test"), "test"); assertEquals(echo.$echo("abcdefg"), "abcdefg"); assertEquals(echo.$echo(1234), 1234); rpcExporter.unexport(); RemoteService remoteService = new RemoteServiceImpl(); rpcExporter = protocol.export(proxy.getInvoker(remoteService, RemoteService.class, URL.valueOf("rmi://127.0.0.1:9002/remoteService"))); // cast to EchoService echo = proxy.getProxy(protocol.refer(EchoService.class, URL.valueOf("rmi://127.0.0.1:9002/remoteService"))); assertEquals(echo.$echo("test"), "test"); assertEquals(echo.$echo("abcdefg"), "abcdefg"); assertEquals(echo.$echo(1234), 1234); rpcExporter.unexport(); } /*@Test public void testRpcInvokerGroup() throws Exception { DemoService service = new DemoServiceImpl(); RpcUtils.export("demo://127.0.0.1:9030/com.alibaba.dubbo.rpc.TestService",DemoService.class,service); RpcUtils.export("dubbo://127.0.0.1:9031/TestService",DemoService.class,service); RpcUtils.export("rmi://127.0.0.1:9032/com.alibaba.dubbo.rpc.TestService",DemoService.class,service); RpcUtils.export("rmi://127.0.0.1:9033/com.alibaba.dubbo.rpc.TestService",DemoService.class,service); service = RpcUtils.createProxy(DemoService.class, new String[]{ "demo://127.0.0.1:9030/com.alibaba.dubbo.rpc.TestService?weight=20", "dubbo://127.0.0.1:9031/TestService?weight=20", "rmi://127.0.0.1:9032/com.alibaba.dubbo.rpc.TestService", }); assertEquals(service.getSize(null), -1); assertEquals(service.getSize(new String[]{"","",""}), 3); // cast to EchoService EchoService echo = RpcUtils.createProxy(EchoService.class, new String[]{ "demo://127.0.0.1:9030/com.alibaba.dubbo.rpc.TestService?weight=20", "dubbo://127.0.0.1:9031/TestService?weight=20", "rmi://127.0.0.1:9032/com.alibaba.dubbo.rpc.TestService", }); assertEquals(echo.$echo("test"), "test"); assertEquals(echo.$echo("abcdefg"), "abcdefg"); assertEquals(echo.$echo(1234), 1234); }*/ /*public void testForkInvoke() throws Exception { DemoService service = new DemoServiceImpl(); protocol.export(proxy.createInvoker("dubbo://127.0.0.1:9040/TestService", DemoService.class, service); protocol.export(proxy.createInvoker("dubbo://127.0.0.1:9041/TestService", DemoService.class, service); protocol.export(proxy.createInvoker("rmi://127.0.0.1:9042/com.alibaba.dubbo.rpc.TestService", DemoService.class, service); protocol.export(proxy.createInvoker("rmi://127.0.0.1:9043/com.alibaba.dubbo.rpc.TestService", DemoService.class, service); RpcInvokerGroup group = Proxies.createInvoker(DemoService.class, new String[]{ "dubbo://127.0.0.1:9040/TestService", "dubbo://127.0.0.1:9041/TestService", "rmi://127.0.0.1:9042/com.alibaba.dubbo.rpc.TestService", "rmi://127.0.0.1:9043/com.alibaba.dubbo.rpc.TestService", }); group.getMethodSettings("echo").setFork(true); group.getMethodSettings("echo").setForkInvokeCallback(new ForkInvokeCallback(){ public Object merge(RpcInvocation invocation, RpcResult[] results) throws Throwable { System.out.println("merge result begin:"); for( RpcResult result : results ) { if( result.hasException() ) System.out.println("exception:"+result.getException().getMessage()); else System.out.println("result:"+result.getResult()); } System.out.println("merge result end:"); return "aaaa"; } }); service = proxy.createProxy(protocol.refer(DemoService.class, group); service.echo("test"); // cast to EchoService EchoService echo = proxy.createProxy(protocol.refer(EchoService.class, group); assertEquals(echo.$echo("test"), "test"); assertEquals(echo.$echo("abcdefg"), "abcdefg"); assertEquals(echo.$echo(1234), 1234); }*/ }