/**
* 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.geronimo.cxf.ejb;
import java.lang.reflect.Method;
import java.util.List;
import java.util.Map;
import javax.interceptor.InvocationContext;
import javax.xml.ws.handler.MessageContext.Scope;
import org.apache.cxf.Bus;
import org.apache.cxf.interceptor.Fault;
import org.apache.cxf.jaxws.AbstractJAXWSMethodInvoker;
import org.apache.cxf.jaxws.context.WebServiceContextImpl;
import org.apache.cxf.message.Exchange;
import org.apache.cxf.message.FaultMode;
import org.apache.cxf.message.MessageContentsList;
import org.apache.cxf.service.invoker.Factory;
import org.apache.openejb.DeploymentInfo;
import org.apache.openejb.InterfaceType;
import org.apache.openejb.RpcContainer;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
public class EJBMethodInvoker extends AbstractJAXWSMethodInvoker {
private static final Logger LOG = LoggerFactory.getLogger(EJBMethodInvoker.class);
private static final String HANDLER_PROPERTIES =
"HandlerProperties";
private DeploymentInfo deploymentInfo;
private Bus bus;
private EJBEndpoint endpoint;
public EJBMethodInvoker(EJBEndpoint endpoint, Bus bus, DeploymentInfo deploymentInfo) {
super((Factory)null);
this.endpoint = endpoint;
this.bus = bus;
this.deploymentInfo = deploymentInfo;
}
@Override
public Object getServiceObject(Exchange context) {
return null;
}
@Override
public void releaseServiceObject(Exchange ex, Object obj) {
// do nothing
}
@Override
protected Object invoke(Exchange exchange,
Object serviceObject,
Method m,
List<Object> params) {
Object result = null;
InvocationContext invContext = exchange.get(InvocationContext.class);
if (invContext == null) {
LOG.debug("PreEJBInvoke");
result = preEjbInvoke(exchange, serviceObject, m, params);
} else {
LOG.debug("EJBInvoke"); // calls performInvocation()
result = super.invoke(exchange, serviceObject, m, params);
}
return result;
}
@Override
protected Object performInvocation(Exchange exchange,
Object serviceObject,
Method m,
Object[] paramArray) throws Exception {
InvocationContext invContext = exchange.get(InvocationContext.class);
invContext.setParameters(paramArray);
Object res = invContext.proceed();
EJBMessageContext ctx = (EJBMessageContext)invContext.getContextData();
Map<String, Object> handlerProperties = (Map<String, Object>)exchange.get(HANDLER_PROPERTIES);
addHandlerProperties(ctx, handlerProperties);
updateWebServiceContext(exchange, ctx);
return res;
}
private Object preEjbInvoke(Exchange exchange,
Object serviceObject,
Method method,
List<Object> params) {
EJBMessageContext ctx = new EJBMessageContext(exchange.getInMessage(), Scope.APPLICATION);
WebServiceContextImpl.setMessageContext(ctx);
Map<String, Object> handlerProperties = removeHandlerProperties(ctx);
exchange.put(HANDLER_PROPERTIES, handlerProperties);
try {
EJBInterceptor interceptor = new EJBInterceptor(params, method, this.endpoint, this.bus, exchange);
Object[] arguments = { ctx, interceptor, ctx };
RpcContainer container = (RpcContainer) this.deploymentInfo.getContainer();
Class callInterface = this.deploymentInfo.getServiceEndpointInterface();
method = getMostSpecificMethod(method, callInterface);
Object res = container.invoke(this.deploymentInfo.getDeploymentID(), InterfaceType.SERVICE_ENDPOINT, callInterface, method, arguments, null);
if (exchange.isOneWay()) {
return null;
}
return new MessageContentsList(res);
} catch (Fault f) {
exchange.getInMessage().put(FaultMode.class, FaultMode.UNCHECKED_APPLICATION_FAULT);
throw f;
} catch (Exception e) {
exchange.getInMessage().put(FaultMode.class, FaultMode.UNCHECKED_APPLICATION_FAULT);
throw createFault(e, method, params, false);
} finally {
WebServiceContextImpl.clear();
}
}
public Object directEjbInvoke(Exchange exchange,
Method m,
List<Object> params) throws Exception {
Object[] paramArray;
if (params != null) {
paramArray = params.toArray();
} else {
paramArray = new Object[]{};
}
return performInvocation(exchange, null, m, paramArray);
}
}