Examples with JobHandle org.drools.core.time.JobHandle used on opensource projects

Example 1 with JobHandle

use of org.drools.core.time.JobHandle in project drools by kiegroup.

the class PseudoClockSchedulerTest method removeExistingJobWhenMultipleQueued.

@Test
public void removeExistingJobWhenMultipleQueued() {
    final Date triggerTime_1 = new Date(1000);
    final Date triggerTime_2 = new Date(2000);
    when(mockTrigger_1.hasNextFireTime()).thenReturn(triggerTime_1);
    when(mockTrigger_2.hasNextFireTime()).thenReturn(triggerTime_2);
    JobHandle jobHandle_1 = scheduler.scheduleJob(mockJob_1, this.mockContext_1, mockTrigger_1);
    JobHandle jobHandle_2 = scheduler.scheduleJob(mockJob_2, this.mockContext_2, mockTrigger_2);
    assertThat(scheduler.getTimeToNextJob(), is(triggerTime_1.getTime()));
    scheduler.removeJob(jobHandle_1);
    assertThat(scheduler.getTimeToNextJob(), is(triggerTime_2.getTime()));
    scheduler.removeJob(jobHandle_2);
    assertThat(scheduler.getTimeToNextJob(), is(-1L));
    verify(mockTrigger_1, atLeastOnce()).hasNextFireTime();
    verify(mockTrigger_2, atLeastOnce()).hasNextFireTime();
}

Example 2 with JobHandle

use of org.drools.core.time.JobHandle in project drools by kiegroup.

the class PseudoClockSchedulerTest method removeExistingJob.

@Test
public void removeExistingJob() {
    final Date triggerTime = new Date(1000);
    when(mockTrigger_1.hasNextFireTime()).thenReturn(triggerTime);
    JobHandle jobHandle = scheduler.scheduleJob(mockJob_1, this.mockContext_1, mockTrigger_1);
    assertThat(scheduler.getTimeToNextJob(), is(triggerTime.getTime()));
    scheduler.removeJob(jobHandle);
    assertThat(scheduler.getTimeToNextJob(), is(-1L));
    verify(mockTrigger_1, atLeastOnce()).hasNextFireTime();
}

Example 3 with JobHandle

use of org.drools.core.time.JobHandle in project jbpm by kiegroup.

the class TaskDeadlinesServiceImpl method unschedule.

public void unschedule(long taskId, Deadline deadline, DeadlineType type) {
    Task task = persistenceContext.findTask(taskId);
    String deploymentId = task.getTaskData().getDeploymentId();
    Deadlines deadlines = ((InternalTask) task).getDeadlines();
    TimerService timerService = TimerServiceRegistry.getInstance().get(deploymentId + TimerServiceRegistry.TIMER_SERVICE_SUFFIX);
    if (timerService != null && timerService instanceof GlobalTimerService) {
        TaskDeadlineJob deadlineJob = new TaskDeadlineJob(taskId, deadline.getId(), type, deploymentId, task.getTaskData().getProcessInstanceId());
        logger.debug("unscheduling timer job for deadline {} {} and task {}  using timer service {}", deadlineJob.getId(), deadline.getId(), taskId, timerService);
        JobHandle jobHandle = jobHandles.remove(deadlineJob.getId());
        if (jobHandle == null) {
            jobHandle = ((GlobalTimerService) timerService).buildJobHandleForContext(new TaskDeadlineJobContext(deadlineJob.getId(), task.getTaskData().getProcessInstanceId(), deploymentId));
        }
        timerService.removeJob(jobHandle);
        // mark the deadlines so they won't be rescheduled again
        deadline.setEscalated(true);
    }
}

Example 4 with JobHandle

use of org.drools.core.time.JobHandle in project jbpm by kiegroup.

the class TaskDeadlinesServiceImpl method schedule.

public void schedule(long taskId, long deadlineId, long delay, DeadlineType type) {
    Task task = persistenceContext.findTask(taskId);
    String deploymentId = task.getTaskData().getDeploymentId();
    TimerService timerService = TimerServiceRegistry.getInstance().get(deploymentId + TimerServiceRegistry.TIMER_SERVICE_SUFFIX);
    if (timerService != null && timerService instanceof GlobalTimerService) {
        TaskDeadlineJob deadlineJob = new TaskDeadlineJob(taskId, deadlineId, type, deploymentId, task.getTaskData().getProcessInstanceId());
        Trigger trigger = new IntervalTrigger(timerService.getCurrentTime(), null, null, -1, delay, 0, null, null);
        JobHandle handle = timerService.scheduleJob(deadlineJob, new TaskDeadlineJobContext(deadlineJob.getId(), task.getTaskData().getProcessInstanceId(), deploymentId), trigger);
        logger.debug("scheduling timer job for deadline {} and task {}  using timer service {}", deadlineJob.getId(), taskId, timerService);
        jobHandles.put(deadlineJob.getId(), handle);
    } else {
        ScheduledFuture<ScheduledTaskDeadline> scheduled = scheduler.schedule(new ScheduledTaskDeadline(taskId, deadlineId, type, deploymentId, task.getTaskData().getProcessInstanceId()), delay, TimeUnit.MILLISECONDS);
        List<ScheduledFuture<ScheduledTaskDeadline>> knownFutures = null;
        if (type == DeadlineType.START) {
            knownFutures = startScheduledTaskDeadlines.get(taskId);
        } else if (type == DeadlineType.END) {
            knownFutures = endScheduledTaskDeadlines.get(taskId);
        }
        if (knownFutures == null) {
            knownFutures = new CopyOnWriteArrayList<ScheduledFuture<ScheduledTaskDeadline>>();
        }
        knownFutures.add(scheduled);
        if (type == DeadlineType.START) {
            startScheduledTaskDeadlines.put(taskId, knownFutures);
        } else if (type == DeadlineType.END) {
            endScheduledTaskDeadlines.put(taskId, knownFutures);
        }
    }
}

Example 5 with JobHandle

use of org.drools.core.time.JobHandle in project drools by kiegroup.

the class EventFactHandle method unscheduleAllJobs.

public void unscheduleAllJobs(InternalWorkingMemory workingMemory) {
    if (!jobs.isEmpty()) {
        synchronized (jobs) {
            TimerService clock = workingMemory.getTimerService();
            while (!jobs.isEmpty()) {
                JobHandle job = jobs.removeFirst();
                clock.removeJob(job);
            }
        }
    }
}