JSessionLocal.java

package org.objectweb.jonas_ejb.container;

import javax.ejb.EJBException;
import javax.ejb.EJBLocalHome;
import javax.ejb.EJBLocalObject;
import javax.ejb.RemoveException;

import org.objectweb.jonas_ejb.lib.EJBInvocation;

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

public abstract class JSessionLocal extends JLocal {

    protected JSessionFactory bf;

    protected JSessionSwitch bs;

    public JSessionLocal(JSessionFactory bf) {
        super(bf);
        TraceEjb.interp.log(BasicLevel.DEBUG, "");
        this.bf = bf;
    }

    // --------------------------------------------------------------------------
    // EJBLocalObject implementation
    // remove() is implemented in the generated part.
    // --------------------------------------------------------------------------

    public abstract void remove() throws RemoveException;

    public EJBLocalHome getEJBLocalHome() {
        return bf.getLocalHome();
    }

    public Object getPrimaryKey() throws EJBException {
        throw new EJBException("Session bean has no primary key");
    }

    public boolean isIdentical(EJBLocalObject obj) {
        TraceEjb.interp.log(BasicLevel.DEBUG, "");
        boolean ret = false;
        JSessionFactory sf = bf;
        if (sf.isStateful()) {
            // For stateful sessions, just compare both objects.
            if (obj != null) {
                ret = obj.equals(this);
            }
        } else {
            // In case of Stateless session bean, we must compare the 2
            // factories
            if (obj != null) {
                try {
                    //a simple cast should work on a local object
                    JSessionLocal ejb2 = (JSessionLocal) obj;
                    JSessionSwitch bs2 = ejb2.getSessionSwitch();
                    JSessionFactory bf2 = bs2.getBeanFactory();
                    ret = bf2.equals(sf);
                } catch (Exception e) {
                    TraceEjb.logger.log(BasicLevel.WARN, "exception:" + e);
                    throw new EJBException("isIdentical failed", e);
                }
            }
        }
        return ret;
    }

    // ---------------------------------------------------------------
    // other public methods, for internal use.
    // ---------------------------------------------------------------

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

    public JSessionSwitch getSessionSwitch() {
        return bs;
    }

    public RequestCtx preInvoke(int txa) {
        TraceEjb.interp.log(BasicLevel.DEBUG, "");
        RequestCtx rctx = bf.preInvoke(txa);
        bs.setMustCommit(rctx.mustCommit); // for remove stateful session
        bs.enlistConnections(rctx.currTx); // Enlist connection list to tx
        return rctx;
    }

    public void checkSecurity(EJBInvocation ejbInv) {
        TraceEjb.interp.log(BasicLevel.DEBUG, "");
        bf.checkSecurity(ejbInv);
    }

    public void postInvoke(RequestCtx rctx) {
        TraceEjb.interp.log(BasicLevel.DEBUG, "");
        bs.delistConnections(rctx.currTx);
        // save current tx (for Stateful Bean Managed only)
        bs.saveBeanTx();
        try {
            bf.postInvoke(rctx);
        } finally {
            if (rctx.sysExc != null) {
                bs.discardICtx(rctx.currTx);
            } else {
                bs.releaseICtx(rctx.currTx);
            }
        }
    }

}

---