org.directwebremoting.impl.DefaultTaskDispatcher Maven / Gradle / Ivy

Go to download
package org.directwebremoting.impl;

import java.util.ArrayList;
import java.util.Collection;
import java.util.Date;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;
import org.directwebremoting.ScriptSession;
import org.directwebremoting.ScriptSessionFilter;
import org.directwebremoting.extend.ScriptSessionManager;
import org.directwebremoting.extend.TaskDispatcher;

/**
 * Default single node implementation of TaskDispatcher
 * @author Joe Walker [joe at getahead dot ltd dot uk]
 */
public class DefaultTaskDispatcher implements TaskDispatcher
{

    public void dispatchTask(ScriptSessionFilter filter, Runnable task)
    {
        Collection all = scriptSessionManager.getAllScriptSessions();
        Collection use = new ArrayList();

        for (ScriptSession session : all)
        {
            if (filter.match(session))
            {
                use.add(session);
            }
        }

        if (use.size() > 0)
        {
            if (log.isDebugEnabled()) {
        	    log.debug("Execution time: " + new Date().toString() + " - Executing task (" + task.getClass().getSimpleName() + ") against " + use.size() + " sessions.");
            }
            target.set(use);
            task.run();
            target.remove();
        }
    }

    public Collection getTargetSessions()
    {
        return target.get();
    }

    /**
     * Connection to the DwrServlet that we are attached to
     * @param scriptSessionManager ...
     */
    public void setScriptSessionManager(ScriptSessionManager scriptSessionManager)
    {
        this.scriptSessionManager = scriptSessionManager;
    }

    /**
     * @see #setScriptSessionManager(ScriptSessionManager)
     */
    private ScriptSessionManager scriptSessionManager;

    /**
     * ThreadLocal in which the list of sessions are stored.
     */
    private final ThreadLocal> target = new ThreadLocal>();

    /**
     * The log stream
     */
    private static final Log log = LogFactory.getLog(DefaultTaskDispatcher.class);
}