Examples with SendWorkerImpl cz.metacentrum.perun.engine.scheduling.impl.SendWorkerImpl used on opensource projects

Example 1 with SendWorkerImpl

use of cz.metacentrum.perun.engine.scheduling.impl.SendWorkerImpl in project perun by CESNET.

the class SendWorkerImplTest method testSendWorkerFailure.

@Test
public void testSendWorkerFailure() throws Exception {
    SendWorker worker = new SendWorkerImpl(sendTaskFalse, null);
    try {
        worker.call();
        fail("TaskExecutionException should be thrown");
    } catch (TaskExecutionException e) {
        assertEquals(sendTaskFalse.getTask(), e.getTask());
        assertEquals(sendTaskFalse.getDestination(), e.getDestination());
        assertEquals(1, e.getReturnCode());
    } catch (Exception e) {
        fail("Unknown exception caught " + e);
    }
}

Example 2 with SendWorkerImpl

use of cz.metacentrum.perun.engine.scheduling.impl.SendWorkerImpl in project perun by CESNET.

the class SendWorkerImplTest method testSendWorkerSuccess.

@Test
public void testSendWorkerSuccess() throws Exception {
    SendWorker worker = new SendWorkerImpl(sendTask1, null);
    SendTask resultSendTask = worker.call();
    assertEquals(SENT, resultSendTask.getStatus());
    assertEquals((long) 0, (long) resultSendTask.getReturnCode());
    Date now = new Date(System.currentTimeMillis());
    assertTrue(resultSendTask.getEndTime().before(now) || resultSendTask.getEndTime().equals(now));
}

Example 3 with SendWorkerImpl

use of cz.metacentrum.perun.engine.scheduling.impl.SendWorkerImpl in project perun by CESNET.

the class SendPlanner method run.

@Override
public void run() {
    BlockingQueue<Task> generatedTasks = schedulingPool.getGeneratedTasksQueue();
    while (!shouldStop()) {
        try {
            Task task = generatedTasks.take();
            // has destinations -> SENDING
            if (task.getDestinations().isEmpty()) {
                task.setStatus(Task.TaskStatus.ERROR);
                try {
                    jmsQueueManager.reportTaskStatus(task.getId(), task.getStatus(), System.currentTimeMillis());
                } catch (JMSException e) {
                    jmsLogError(task);
                }
                try {
                    schedulingPool.removeTask(task);
                } catch (TaskStoreException e) {
                    log.error("[{}] Generated Task without destinations could not be removed from SchedulingPool: {}", task.getId(), e);
                }
                // skip to next generated Task
                continue;
            }
            // Task has destinations
            task.setStatus(Task.TaskStatus.SENDING);
            // TODO - would be probably better to have this as one time call after first SendWorker is submitted
            // TODO   but then processing stuck tasks must reflect, that SENDING task might have sendStartTime=NULL
            task.setSendStartTime(LocalDateTime.now());
            schedulingPool.addSendTaskCount(task, task.getDestinations().size());
            try {
                jmsQueueManager.reportTaskStatus(task.getId(), task.getStatus(), task.getSendStartTime().atZone(ZoneId.systemDefault()).toInstant().toEpochMilli());
            } catch (JMSException e) {
                jmsLogError(task);
            }
            // create SendTask and SendWorker for each Destination
            for (Destination destination : task.getDestinations()) {
                // submit for execution
                SendTask sendTask = new SendTask(task, destination);
                SendWorker worker = new SendWorkerImpl(sendTask, directory);
                sendCompletionService.blockingSubmit(worker);
            }
        } catch (InterruptedException e) {
            String errorStr = "Thread planning SendTasks was interrupted.";
            log.error(errorStr);
            throw new RuntimeException(errorStr, e);
        } catch (Throwable ex) {
            log.error("Unexpected exception in SendPlanner thread. Stuck Tasks will be cleaned by PropagationMaintainer#endStuckTasks() later.", ex);
        }
    }
}