/*
* Licensed to the Apache Software Foundation (ASF) under one
* or more contributor license agreements. See the NOTICE file
* distributed with this work for additional information
* regarding copyright ownership. The ASF licenses this file
* to you 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 org.apache.tuscany.sca.core.invocation;
import java.lang.reflect.Method;
import java.util.List;
import net.sf.cglib.proxy.Callback;
import net.sf.cglib.proxy.Enhancer;
import net.sf.cglib.proxy.Factory;
import net.sf.cglib.proxy.MethodInterceptor;
import net.sf.cglib.proxy.MethodProxy;
import org.apache.tuscany.sca.core.context.ServiceReferenceExt;
import org.apache.tuscany.sca.core.context.impl.ServiceReferenceImpl;
import org.apache.tuscany.sca.core.invocation.impl.JDKCallbackInvocationHandler;
import org.apache.tuscany.sca.core.invocation.impl.JDKInvocationHandler;
import org.apache.tuscany.sca.interfacedef.InterfaceContractMapper;
import org.apache.tuscany.sca.invocation.MessageFactory;
import org.apache.tuscany.sca.runtime.Invocable;
import org.apache.tuscany.sca.runtime.RuntimeEndpoint;
import org.oasisopen.sca.ServiceReference;
/**
* The implementation of a wire service that uses cglib dynamic proxies
*
* @version $Rev$ $Date$
* @tuscany.spi.extension.asclient
*/
public class CglibProxyFactory implements ProxyFactory {
private MessageFactory messageFactory;
public CglibProxyFactory(MessageFactory messageFactory, InterfaceContractMapper mapper) {
this.messageFactory = messageFactory;
}
public <T> T createProxy(final Class<T> interfaze, Invocable invocable) throws ProxyCreationException {
if (invocable instanceof RuntimeEndpoint) {
Enhancer enhancer = new Enhancer();
enhancer.setSuperclass(interfaze);
enhancer.setCallback(new CglibMethodInterceptor<T>(interfaze, invocable));
Object proxy = enhancer.create();
return interfaze.cast(proxy);
}
ServiceReference<T> serviceReference = new ServiceReferenceImpl(interfaze, invocable, null);
return createProxy(serviceReference);
}
/**
* create the proxy with cglib. use the same JDKInvocationHandler as
* JDKProxyService.
*/
public <T> T createProxy(ServiceReference<T> callableReference) throws ProxyCreationException {
Enhancer enhancer = new Enhancer();
Class<T> interfaze = callableReference.getBusinessInterface();
enhancer.setSuperclass(interfaze);
enhancer.setCallback(new CglibMethodInterceptor<T>(callableReference));
Object proxy = enhancer.create();
((ServiceReferenceImpl)callableReference).setProxy(proxy);
return interfaze.cast(proxy);
}
/**
* create the callback proxy with cglib. use the same
* JDKCallbackInvocationHandler as JDKProxyService.
*/
public <T> T createCallbackProxy(Class<T> interfaze, final List<? extends Invocable> wires) throws ProxyCreationException {
ServiceReferenceImpl<T> callbackReference = new ServiceReferenceImpl(interfaze, wires.get(0), null);
return callbackReference != null ? createCallbackProxy(callbackReference) : null;
}
/**
* create the callback proxy with cglib. use the same
* JDKCallbackInvocationHandler as JDKProxyService.
*/
public <T> T createCallbackProxy(ServiceReference<T> callbackReference) throws ProxyCreationException {
Enhancer enhancer = new Enhancer();
Class<T> interfaze = callbackReference.getBusinessInterface();
enhancer.setSuperclass(interfaze);
enhancer.setCallback(new CglibMethodInterceptor<T>(callbackReference));
Object object = enhancer.create();
T proxy = interfaze.cast(object);
((ServiceReferenceExt<T>)callbackReference).setProxy(proxy);
return proxy;
}
@SuppressWarnings("unchecked")
public <B, R extends ServiceReference<B>> R cast(B target) throws IllegalArgumentException {
if (isProxyClass(target.getClass())) {
Factory factory = (Factory)target;
Callback[] callbacks = factory.getCallbacks();
if (callbacks.length != 1 || !(callbacks[0] instanceof CglibMethodInterceptor)) {
throw new IllegalArgumentException("The object is not a known proxy.");
}
CglibMethodInterceptor interceptor = (CglibMethodInterceptor)callbacks[0];
return (R)interceptor.invocationHandler.getCallableReference();
} else {
throw new IllegalArgumentException("The object is not a known proxy.");
}
}
/**
* @see org.apache.tuscany.sca.core.invocation.ProxyFactory#isProxyClass(java.lang.Class)
*/
public boolean isProxyClass(Class<?> clazz) {
return Factory.class.isAssignableFrom(clazz);
}
private class CglibMethodInterceptor<T> implements MethodInterceptor {
private JDKInvocationHandler invocationHandler;
public CglibMethodInterceptor(ServiceReference<T> callableReference) {
invocationHandler = new JDKInvocationHandler(messageFactory, callableReference);
}
public CglibMethodInterceptor(Class<?> interfaze, Invocable invocable) {
invocationHandler = new JDKInvocationHandler(messageFactory, interfaze, invocable);
}
public CglibMethodInterceptor(ServiceReferenceImpl<T> callbackReference) {
invocationHandler = new JDKCallbackInvocationHandler(messageFactory, callbackReference);
}
/**
* @see net.sf.cglib.proxy.MethodInterceptor#intercept(java.lang.Object, java.lang.reflect.Method, java.lang.Object[], net.sf.cglib.proxy.MethodProxy)
*/
public Object intercept(Object obj, Method method, Object[] args, MethodProxy proxy) throws Throwable {
Object result = invocationHandler.invoke(proxy, method, args);
return result;
}
}
}