/* | |
* 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.proxy; | |
import java.lang.reflect.InvocationTargetException; | |
import java.rmi.Remote; | |
import java.rmi.RemoteException; | |
import java.rmi.server.RemoteServer; | |
import java.rmi.server.ServerNotActiveException; | |
import org.springframework.remoting.rmi.RmiInvocationHandler; | |
import org.springframework.remoting.support.RemoteInvocation; | |
import com.alibaba.dubbo.common.URL; | |
import com.alibaba.dubbo.common.utils.ReflectUtils; | |
import com.alibaba.dubbo.common.utils.StringUtils; | |
import com.alibaba.dubbo.rpc.Invocation; | |
import com.alibaba.dubbo.rpc.Invoker; | |
import com.alibaba.dubbo.rpc.Result; | |
import com.alibaba.dubbo.rpc.RpcContext; | |
import com.alibaba.dubbo.rpc.RpcException; | |
import com.alibaba.dubbo.rpc.RpcInvocation; | |
import com.alibaba.dubbo.rpc.RpcResult; | |
import com.alibaba.dubbo.rpc.protocol.rmi.RmiProxyFactory; | |
/** | |
* SpringRmiProxyFactory | |
* | |
* @author william.liangf | |
*/ | |
public class SpringRmiProxyFactory implements RmiProxyFactory { | |
public boolean isSupported(Remote remote, Class<?> serviceType, URL url) { | |
return ReflectUtils.isInstance(remote, "org.springframework.remoting.rmi.RmiInvocationHandler"); | |
} | |
private static void assertRmiInvocationHandler() { | |
try { | |
Class.forName("org.springframework.remoting.rmi.RmiInvocationHandler"); | |
} catch (ClassNotFoundException e1) { | |
throw new RpcException( | |
"set codec spring for protocol rmi," | |
+ " but NO spring class org.springframework.remoting.rmi.RmiInvocationHandler at provider side!"); | |
} | |
} | |
public <T> Remote getProxy(final Invoker<T> invoker) { | |
assertRmiInvocationHandler(); | |
return new org.springframework.remoting.rmi.RmiInvocationHandler() { | |
public Object invoke(RemoteInvocation invocation) throws RemoteException, | |
NoSuchMethodException, IllegalAccessException, InvocationTargetException { | |
String client = null; | |
try { | |
client = RemoteServer.getClientHost(); | |
} catch (ServerNotActiveException e) { | |
// Ignore it. | |
} | |
Invocation inv = new RpcInvocation(invocation.getMethodName(), | |
invocation.getParameterTypes(), invocation.getArguments()); | |
try { | |
RpcContext.getContext().setRemoteAddress(client, 0); | |
return invoker.invoke(inv).recreate(); | |
} catch (RpcException e) { | |
throw new RemoteException(StringUtils.toString(e)); | |
} catch (Throwable t) { | |
throw new InvocationTargetException(t); | |
} | |
} | |
public String getTargetInterfaceName() throws RemoteException { | |
return invoker.getInterface().getName(); | |
} | |
}; | |
} | |
public <T> Invoker<T> getInvoker(final Remote remote, final Class<T> serviceType, final URL url) { | |
assertRmiInvocationHandler(); | |
return new Invoker<T>() { | |
public Class<T> getInterface() { | |
return serviceType; | |
} | |
public URL getUrl() { | |
return url; | |
} | |
public boolean isAvailable() { | |
return true; | |
} | |
public Result invoke(Invocation invocation) throws RpcException { | |
RpcResult result = new RpcResult(); | |
try { | |
RemoteInvocation i = new RemoteInvocation(); | |
i.setMethodName(invocation.getMethodName()); | |
i.setParameterTypes(invocation.getParameterTypes()); | |
i.setArguments(invocation.getArguments()); | |
result.setValue(((RmiInvocationHandler) remote).invoke(i)); | |
} catch (InvocationTargetException e) { | |
result.setException(e.getTargetException()); | |
} catch (Exception e) { | |
throw new RpcException(StringUtils.toString(e), e); | |
} | |
return result; | |
} | |
public void destroy() { | |
} | |
}; | |
} | |
} |