com.enioka.jqm.tools.RunningJobInstanceManager Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of jqm-engine Show documentation

A library containing the JQM engine. Cannot be used alone.

There is a newer version: 2.2.9

package com.enioka.jqm.tools;

import java.util.concurrent.ConcurrentHashMap;

import com.enioka.jqm.model.Instruction;
import com.enioka.jqm.model.JobInstance;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

/**
 * Responsible for creating and storing references to the the {@link RunningJobInstance}.
 */
class RunningJobInstanceManager
{
    private Logger jqmlogger = LoggerFactory.getLogger(RunningJobInstanceManager.class);

    private ConcurrentHashMap instancesByTracker = new ConcurrentHashMap();
    private ConcurrentHashMap instancesById = new ConcurrentHashMap();

    void startNewJobInstance(JobInstance ji, QueuePoller qp)
    {
        RjiRegistration reg = new RjiRegistration();

        reg.ji = ji;
        reg.qp = qp;
        reg.rji = new RunningJobInstance(ji, qp);

        instancesByTracker.put(reg.rji, reg);
        instancesById.put(reg.ji.getId(), reg);

        (new Thread(reg.rji)).start();
    }

    void signalEndOfRun(RunningJobInstance rji)
    {
        if (!instancesByTracker.containsKey(rji))
        {
            jqmlogger.warn("Tried to signal the end of a job instance which was not registered inside the manager");
            return;
        }

        RjiRegistration reg = instancesByTracker.get(rji);

        // Remove from manager
        instancesByTracker.remove(rji);
        instancesById.remove(reg.ji.getId());

        // Signal queue poller.
        if (reg.qp != null)
        {
            reg.qp.releaseResources(reg.ji);
        }
    }

    void handleInstruction(int jobInstanceId, Instruction instruction)
    {
        if (!instancesById.containsKey(jobInstanceId))
        {
            jqmlogger.warn("Tried to send an instruction to an instance which was not registered inside the manager");
            return;
        }

        RjiRegistration reg = instancesById.get(jobInstanceId);
        reg.rji.handleInstruction(instruction);
    }

    private class RjiRegistration
    {
        RunningJobInstance rji;
        QueuePoller qp;
        JobInstance ji;
    }

    /**
     * Send a kill signal to all running instances and return as soon as the signal is sent.
     */
    void killAll()
    {
        for (RjiRegistration reg : this.instancesById.values().toArray(new RjiRegistration[] {}))
        {
            reg.rji.handleInstruction(Instruction.KILL);
        }
    }
}