JTimerHandle.java

package org.objectweb.jonas_ejb.container;

import java.io.Serializable;

import javax.ejb.EJBException;
import javax.ejb.NoSuchObjectLocalException;
import javax.ejb.Timer;
import javax.ejb.TimerHandle;
import javax.naming.Context;
import javax.naming.InitialContext;
import javax.naming.NamingException;

import org.objectweb.jonas.naming.NamingManager;
import org.objectweb.jonas_ejb.container.JContext;
import org.objectweb.jonas_lib.naming.ContainerNaming;
import org.objectweb.jonas.container.EJBServiceImpl;
import org.objectweb.jonas.service.ServiceManager;
import org.objectweb.jonas_timer.TraceTimer;
import org.objectweb.util.monolog.api.BasicLevel;

public class JTimerHandle implements TimerHandle {

    private long period;
    private long starttime;
    private Serializable info;
    private String beanname;
    private String container;
    
    private Serializable pk;
    
    public JTimerHandle(long starttime, long period, Serializable info, String beanname, String container, Serializable pk) {
        TraceTimer.logger.log(BasicLevel.DEBUG, "New JTimerHandle initial = " + starttime + ", period = " + period);
        this.starttime = starttime;
        this.period = period;
        this.info = info;
        this.beanname = beanname;
        this.container = container;
        this.pk = pk;
    }

    public Timer getTimer() throws IllegalStateException, NoSuchObjectLocalException, EJBException {

        // Must reject Timer operations according to the EJB spec
        // See EJB 2.1 page 87 (Stateful session beans)
        try {
            ContainerNaming naming = NamingManager.getInstance();
            Context ctx = naming.getComponentContext();
            JContext sc = (JContext) ctx.lookup("MY_SF_CONTEXT");
            if (sc.getState() != 2) {
                throw new IllegalStateException("This operation is not allowed here");
            }
        } catch (NamingException ne) {
            // No context registered. Assume it's OK.
            // It's the case for Entity or MDB.
        }

        EJBServiceImpl ejbserv = null;
        try {
            ejbserv = (EJBServiceImpl) ServiceManager.getInstance().getEjbService();
        } catch (Exception e) {
            throw new IllegalStateException("Cannot use Timer outside jonas server");
        }
        JContainer cont = (JContainer) ejbserv.getContainer(container);
        if (cont == null) {
            TraceTimer.logger.log(BasicLevel.ERROR, "Cannot get container =" + container);
            throw new IllegalStateException("Cannot get container");
        }
        JFactory bf = (JFactory) cont.getBeanFactory(beanname);
        JTimerService timerservice = null;
        if (bf instanceof JEntityFactory) {
            // entity bean
            JEntityFactory ef = (JEntityFactory) bf;
            Serializable pks = ef.decodePK(pk);
            TraceEjb.interp.log(BasicLevel.DEBUG, "encoded PK=" + pk);
            TraceEjb.interp.log(BasicLevel.DEBUG, "decoded PK=" + pks);
            JEntitySwitch es = ef.existEJB(pks);
            if (es == null) {
                throw new NoSuchObjectLocalException("No entity for this pk");
            }
            timerservice = (JTimerService) es.getEntityTimerService();
        } else {
            // stateless session or message driven bean
            timerservice = (JTimerService) bf.getTimerService();
        }
        if (timerservice == null) {
            throw new IllegalStateException("Cannot retrieve TimerService");
        }
        // Get the Timer from the list. If not found, it may have been cancelled.
        Timer ret = timerservice.getTimer(starttime, period, info);
        if (ret == null) {
            throw new NoSuchObjectLocalException("Timer does not exist any longer");
        }
        return ret;
    }

}

---