JSessionContext.java

package org.objectweb.jonas_ejb.container;

import java.io.Serializable;
import java.rmi.RemoteException;
import java.util.List;

import javax.ejb.EJBLocalObject;
import javax.ejb.EJBObject;
import javax.ejb.RemoveException;
import javax.ejb.SessionBean;
import javax.ejb.SessionContext;
import javax.xml.rpc.handler.MessageContext;

import org.objectweb.util.monolog.api.BasicLevel;

public abstract class JSessionContext extends JContext implements SessionContext, Serializable {

    protected JSessionSwitch bs = null;

    protected boolean ismarkedremoved;

    public JSessionContext(JSessionFactory bf, SessionBean eb) {
        super(bf, eb);
        TraceEjb.interp.log(BasicLevel.DEBUG, "");
        // session beans may have home or localHome
        this.home = bf.getHome();
        this.localhome = bf.getLocalHome();
    }

    // ------------------------------------------------------------------
    // SessionContext implementation
    // ------------------------------------------------------------------

    public EJBObject getEJBObject() throws IllegalStateException {
        TraceEjb.interp.log(BasicLevel.DEBUG, "");
        if (ismarkedremoved || bs == null) {
            if (ismarkedremoved)
                TraceEjb.logger.log(BasicLevel.ERROR, "marked removed");
            else
                TraceEjb.logger.log(BasicLevel.ERROR, "no SessionSwitch");
            throw new IllegalStateException("getEJBObject: EJB is removed");
        }
        EJBObject ejbobject = bs.getRemote();
        if (ejbobject == null) {
            throw new IllegalStateException("No Remote Interface for this bean");
        }
        return ejbobject;
    }

    public EJBLocalObject getEJBLocalObject() throws IllegalStateException {
        TraceEjb.interp.log(BasicLevel.DEBUG, "");
        if (ismarkedremoved || bs == null) {
            if (ismarkedremoved)
                TraceEjb.logger.log(BasicLevel.ERROR, "marked removed");
            else
                TraceEjb.logger.log(BasicLevel.ERROR, "no SessionSwitch");
            throw new IllegalStateException("getEJBLocalObject: EJB is removed");
        }
        EJBLocalObject ejblocalobject = bs.getLocal();
        if (ejblocalobject == null) throw new IllegalStateException("No Local Object for this bean");
        return ejblocalobject;
    }

    public abstract MessageContext getMessageContext() throws java.lang.IllegalStateException;

    public boolean getRollbackOnly() throws IllegalStateException {
        TraceEjb.interp.log(BasicLevel.DEBUG, "");

        if (getState() < 2) {
            throw new IllegalStateException("the instance is not allowed to call this method");
        }
        return super.getRollbackOnly();
    }
    
    // -------------------------------------------------------------------
    // Other public methods
    // -------------------------------------------------------------------

    public void initSessionContext(JSessionSwitch bs) {
        TraceEjb.interp.log(BasicLevel.DEBUG, "");
        this.bs = bs;
        ismarkedremoved = false;
    }

    // RemoteException is throwned to allow simpler genic templates
    public SessionBean getInstance() throws RemoteException {
        if (instance == null) {
            TraceEjb.logger.log(BasicLevel.ERROR, "null!");
            throw new RemoteException("No instance available");
        }
        return (SessionBean) instance;
    }

    public boolean isMarkedRemoved() {
        return ismarkedremoved;
    }

    // ------------------------------------------------------------------
    // Other methods
    // ------------------------------------------------------------------

    abstract public void setRemoved() throws RemoteException, RemoveException;
    abstract public void setConnectionList(List conlist);
}

---