/* | |
* 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); | |
}*/ | |
} |