/*
* 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.openejb.core.interceptor;
import static org.apache.openejb.util.Join.join;
import java.lang.reflect.Method;
import java.lang.reflect.Proxy;
import java.util.ArrayList;
import java.util.List;
import java.util.Map;
import java.util.Set;
import java.util.Stack;
import javax.interceptor.InvocationContext;
import org.apache.openejb.core.Operation;
import org.apache.openejb.core.ThreadContext;
import org.apache.openejb.util.Classes;
import org.apache.openejb.util.LogCategory;
import org.apache.openejb.util.Logger;
import org.apache.openejb.util.proxy.DynamicProxyImplFactory;
/**
* @version $Rev: 1311042 $ $Date: 2012-04-08 09:54:59 -0700 (Sun, 08 Apr 2012) $
*/
public class InterceptorStack {
private static final Logger logger = Logger.getInstance(LogCategory.OPENEJB, "org.apache.openejb.util.resources");
private final Object beanInstance;
private final List<Interceptor> interceptors;
private final Method targetMethod;
private final Operation operation;
public InterceptorStack(Object beanInstance, Method targetMethod, Operation operation, List<InterceptorData> interceptorDatas, Map<String, Object> interceptorInstances) {
if (interceptorDatas == null) throw new NullPointerException("interceptorDatas is null");
if (interceptorInstances == null) throw new NullPointerException("interceptorInstances is null");
this.beanInstance = beanInstance;
this.targetMethod = targetMethod;
this.operation = operation;
interceptors = new ArrayList<Interceptor>(interceptorDatas.size());
// try {
// interceptors.add(new Interceptor(new Debug(), Debug.class.getMethod("invoke", InvocationContext.class)));
// } catch (Throwable e) {
// }
for (InterceptorData interceptorData : interceptorDatas) {
Class interceptorClass = interceptorData.getInterceptorClass();
Object interceptorInstance = interceptorInstances.get(interceptorClass.getName());
if (interceptorInstance == null) {
throw new IllegalArgumentException("No interceptor of type " + interceptorClass.getName());
}
Set<Method> methods = interceptorData.getMethods(operation);
for (Method method : methods) {
final Interceptor interceptor;
Object handler = DynamicProxyImplFactory.realHandler(interceptorInstance);
if (handler != null && method.getDeclaringClass().equals(handler.getClass())) { // dynamic impl
interceptor = new Interceptor(handler, method);
} else {
interceptor = new Interceptor(interceptorInstance, method);
}
interceptors.add(interceptor);
}
}
}
private static final ThreadLocal<Stack> stack = new ThreadLocal<Stack>();
private class Debug {
private Stack stack() {
Stack s = stack.get();
if (s == null){
s = new Stack();
stack.set(s);
}
return s;
}
public Object invoke(InvocationContext context) throws Exception {
try {
StringBuilder sb = new StringBuilder();
ThreadContext threadContext = ThreadContext.getThreadContext();
String txPolicy = threadContext.getTransactionPolicy().getClass().getSimpleName();
String ejbName = threadContext.getBeanContext().getEjbName();
String methodName = targetMethod.getName() + "(" + join(", ", Classes.getSimpleNames(targetMethod.getParameterTypes())) + ")";
sb.append(join("", stack()));
sb.append(ejbName).append(".");
sb.append(methodName).append(" <").append(txPolicy).append("> {");
synchronized (System.out){
System.out.println(sb.toString());
}
} catch (Throwable e) {
}
try {
stack().push(" ");
return context.proceed();
} finally {
stack().pop();
StringBuilder sb = new StringBuilder();
sb.append(join("", stack()));
sb.append("}");
synchronized (System.out){
System.out.println(sb.toString());
}
}
}
}
public InvocationContext createInvocationContext(Object... parameters) {
InvocationContext invocationContext = new ReflectionInvocationContext(operation, interceptors, beanInstance, targetMethod, parameters);
return invocationContext;
}
public Object invoke(Object... parameters) throws Exception {
try {
InvocationContext invocationContext = createInvocationContext(parameters);
if (ThreadContext.getThreadContext() != null) {
ThreadContext.getThreadContext().set(InvocationContext.class, invocationContext);
}
Object value = invocationContext.proceed();
return value;
} finally {
if (ThreadContext.getThreadContext() != null) {
ThreadContext.getThreadContext().remove(InvocationContext.class);
}
}
}
public Object invoke(javax.xml.ws.handler.MessageContext messageContext, Object... parameters) throws Exception {
try {
InvocationContext invocationContext = new JaxWsInvocationContext(operation, interceptors, beanInstance, targetMethod, messageContext, parameters);
ThreadContext.getThreadContext().set(InvocationContext.class, invocationContext);
Object value = invocationContext.proceed();
return value;
} finally {
ThreadContext.getThreadContext().remove(InvocationContext.class);
}
}
public Object invoke(javax.xml.rpc.handler.MessageContext messageContext, Object... parameters) throws Exception {
try {
InvocationContext invocationContext = new JaxRpcInvocationContext(operation, interceptors, beanInstance, targetMethod, messageContext, parameters);
ThreadContext.getThreadContext().set(InvocationContext.class, invocationContext);
Object value = invocationContext.proceed();
return value;
} finally {
ThreadContext.getThreadContext().remove(InvocationContext.class);
}
}
}