JStatelessContext.java

package org.objectweb.jonas_ejb.container;

import java.io.Serializable;
import java.rmi.RemoteException;
import java.security.Principal;
import java.util.Collection;
import java.util.Date;
import java.util.List;

import javax.ejb.EJBException;
import javax.ejb.EJBHome;
import javax.ejb.EJBLocalHome;
import javax.ejb.RemoveException;
import javax.ejb.SessionBean;
import javax.ejb.Timer;
import javax.ejb.TimerService;
import javax.xml.rpc.handler.MessageContext;

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

public class JStatelessContext extends JSessionContext implements TimerService {

    public JStatelessContext(JSessionFactory bf, SessionBean sb) {
        super(bf, sb);
        TraceEjb.interp.log(BasicLevel.DEBUG, "");
    }

    public TimerService getTimerService() throws IllegalStateException {
        TraceEjb.interp.log(BasicLevel.DEBUG, "");
        if (getState() == 0) {
            throw new IllegalStateException("the instance is not allowed to call this method");
        }
        if (getState() == 1) {
            // Called from ejbCreate : getTimerService should work, but
            // not the timer operations! (See spec EJB 2.1 page 100)
            return this;
        }
        return bf.getTimerService();
    }

    public void setRemoved() throws RemoteException, RemoveException {
        TraceEjb.interp.log(BasicLevel.DEBUG, "");
        // Set a flag to finish remove at postInvoke.
        ismarkedremoved = true;
    }

    public MessageContext getMessageContext() throws IllegalStateException {
        if (bs == null) {
            throw new IllegalStateException("No SessionSwitch for that bean");
        }
        MessageContext mc = ((JStatelessSwitch) bs).getMsgContext();
        if (mc == null) {
            throw new IllegalStateException("No ServiceEndpoint for that bean");
        }
        return mc;
    }
    
    public void setConnectionList(List conlist) {
        throw new IllegalStateException("Stateless beans should not reuse connections");
    }

    public Principal getCallerPrincipal() throws IllegalStateException {
        if (getState() < 2) {
            throw new IllegalStateException("the instance is not allowed to call this method");
        }
        return super.getCallerPrincipal();
    }

    public boolean isCallerInRole(String roleName) throws IllegalStateException {
        if (getState() < 2) {
            throw new IllegalStateException("the instance is not allowed to call this method");
        }
        return super.isCallerInRole(roleName);
    }

    public Timer createTimer(long arg0, Serializable arg1) throws IllegalArgumentException, IllegalStateException, EJBException {
        // Not allowed when called from ejbCreate
        if (getState() < 2) {
            throw new IllegalStateException("the instance is not allowed to call this method");
        }
        return getTimerService().createTimer(arg0, arg1);
    }

    public Timer createTimer(long arg0, long arg1, Serializable arg2) throws IllegalArgumentException,
            IllegalStateException, EJBException {
        // Not allowed when called from ejbCreate
        if (getState() < 2) {
            throw new IllegalStateException("the instance is not allowed to call this method");
        }
        return getTimerService().createTimer(arg0, arg1, arg2);
    }

    public Timer createTimer(Date arg0, Serializable arg1) throws IllegalArgumentException, IllegalStateException, EJBException {
        // Not allowed when called from ejbCreate
        if (getState() < 2) {
            throw new IllegalStateException("the instance is not allowed to call this method");
        }
        return getTimerService().createTimer(arg0, arg1);
    }

    public Timer createTimer(Date arg0, long arg1, Serializable arg2) throws IllegalArgumentException, IllegalStateException, EJBException {
        // Not allowed when called from ejbCreate
        if (getState() < 2) {
            throw new IllegalStateException("the instance is not allowed to call this method");
        }
        return getTimerService().createTimer(arg0, arg1, arg2);
  }

    public Collection getTimers() throws IllegalStateException, EJBException {
        // Not allowed when called from ejbCreate
        if (getState() < 2) {
            throw new IllegalStateException("the instance is not allowed to call this method");
        }
        return getTimerService().getTimers();
   }

 
}

---