/*
* 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;
import org.apache.openejb.BeanContext;
import org.apache.openejb.BeanType;
import org.apache.openejb.InterfaceType;
import org.apache.openejb.InternalErrorException;
import org.apache.openejb.core.ivm.EjbObjectProxyHandler;
import org.apache.openejb.core.ivm.IntraVmProxy;
import org.apache.openejb.core.managed.ManagedObjectHandler;
import org.apache.openejb.core.singleton.SingletonEjbObjectHandler;
import org.apache.openejb.core.stateful.StatefulEjbObjectHandler;
import org.apache.openejb.core.stateless.StatelessEjbObjectHandler;
import org.apache.openejb.jee.SessionType;
import org.apache.openejb.spi.SecurityService;
import org.apache.openejb.util.proxy.LocalBeanProxyFactory;
import org.apache.openejb.util.proxy.ProxyManager;
import javax.ejb.EJBLocalObject;
import javax.ejb.EJBObject;
import javax.ejb.SessionContext;
import javax.transaction.UserTransaction;
import javax.xml.rpc.handler.MessageContext;
import java.lang.reflect.Method;
import java.util.ArrayList;
import java.util.List;
/**
* @version $Rev: 1299742 $ $Date: 2012-03-12 10:11:23 -0700 (Mon, 12 Mar 2012) $
*/
public abstract class BaseSessionContext extends BaseContext implements SessionContext {
protected BaseSessionContext(SecurityService securityService) {
super(securityService);
}
public BaseSessionContext(SecurityService securityService, UserTransaction userTransaction) {
super(securityService, userTransaction);
}
public boolean wasCancelCalled() {
ThreadContext threadContext = ThreadContext.getThreadContext();
BeanContext di = threadContext.getBeanContext();
Method runningMethod = threadContext.get(Method.class);
if (di.isAsynchronous(runningMethod)) {
if(runningMethod.getReturnType() == void.class) {
throw new IllegalStateException("Current running method " + runningMethod.getName() + " is an asynchronous method, but its return type is void :" + di.getDestinationId());
}
return ThreadContext.isAsynchronousCancelled();
}
throw new IllegalStateException("Current running method " + runningMethod.getName() + " is not an asynchronous method :" + di.getDestinationId());
}
public EJBLocalObject getEJBLocalObject() throws IllegalStateException {
check(Call.getEJBLocalObject);
ThreadContext threadContext = ThreadContext.getThreadContext();
BeanContext di = threadContext.getBeanContext();
if (di.getLocalHomeInterface() == null) throw new IllegalStateException("Bean does not have an EJBLocalObject interface: "+di.getDeploymentID());
return (EJBLocalObject) EjbObjectProxyHandler.createProxy(di, threadContext.getPrimaryKey(), InterfaceType.EJB_LOCAL, di.getLocalInterface());
}
public EJBObject getEJBObject() throws IllegalStateException {
check(Call.getEJBObject);
ThreadContext threadContext = ThreadContext.getThreadContext();
BeanContext di = threadContext.getBeanContext();
if (di.getHomeInterface() == null) throw new IllegalStateException("Bean does not have an EJBObject interface: "+di.getDeploymentID());
return (EJBObject) EjbObjectProxyHandler.createProxy(di, threadContext.getPrimaryKey(), InterfaceType.EJB_OBJECT, di.getRemoteInterface());
}
public MessageContext getMessageContext() throws IllegalStateException {
check(Call.getMessageContext);
ThreadContext threadContext = ThreadContext.getThreadContext();
MessageContext messageContext = threadContext.get(MessageContext.class);
if (messageContext == null) throw new IllegalStateException("Only calls on the service-endpoint have a MessageContext.");
return messageContext;
}
public Object getBusinessObject(Class interfce) {
check(Call.getBusinessObject);
if (interfce == null) throw new IllegalStateException("Interface argument cannot me null.");
final ThreadContext threadContext = ThreadContext.getThreadContext();
final BeanContext di = threadContext.getBeanContext();
final InterfaceType interfaceType = di.getInterfaceType(interfce);
final BeanType type = di.getComponentType();
if (interfaceType == null){
throw new IllegalStateException("Component has no such interface: " + interfce.getName());
}
if (!interfaceType.isBusiness()) {
throw new IllegalStateException("Interface is not a business interface for this bean: " + interfce.getName());
}
try {
EjbObjectProxyHandler handler;
switch(di.getComponentType()){
case STATEFUL: {
handler = new StatefulEjbObjectHandler(di, threadContext.getPrimaryKey(), interfaceType, new ArrayList<Class>(), interfce);
break;
}
case STATELESS: {
handler = new StatelessEjbObjectHandler(di, threadContext.getPrimaryKey(), interfaceType, new ArrayList<Class>(), interfce);
break;
}
case SINGLETON: {
handler = new SingletonEjbObjectHandler(di, threadContext.getPrimaryKey(), interfaceType, new ArrayList<Class>(), interfce);
break;
}
case MANAGED: {
handler = new ManagedObjectHandler(di, threadContext.getPrimaryKey(), interfaceType, new ArrayList<Class>(), interfce);
break;
}
default: throw new IllegalStateException("Bean is not a session bean: "+di.getComponentType());
}
if (InterfaceType.LOCALBEAN.equals(interfaceType)) {
return LocalBeanProxyFactory.newProxyInstance(di.getClassLoader(), di.getBeanClass(), handler);
} else {
List<Class> interfaces = new ArrayList<Class>();
interfaces.addAll(di.getInterfaces(interfaceType));
interfaces.add(IntraVmProxy.class);
if (SessionType.STATEFUL.equals(type) || SessionType.MANAGED.equals(type)) {
interfaces.add(BeanContext.Removable.class);
}
return ProxyManager.newProxyInstance(interfaces.toArray(new Class[interfaces.size()]), handler);
}
} catch (IllegalAccessException iae) {
throw new InternalErrorException("Could not create IVM proxy for " + interfce.getName() + " interface", iae);
}
}
public Class getInvokedBusinessInterface() {
check(Call.getInvokedBusinessInterface);
ThreadContext threadContext = ThreadContext.getThreadContext();
Class invokedInterface = threadContext.getInvokedInterface();
InterfaceType type = threadContext.getBeanContext().getInterfaceType(invokedInterface);
if (!type.isBusiness()) throw new IllegalStateException("The EJB spec requires us to cripple the use of this method for anything but business interface proxy. But FYI, your invoked interface is: "+invokedInterface.getName());
if (invokedInterface == null){
throw new IllegalStateException("Business interface not set into ThreadContext.");
}
return invokedInterface;
}
}