Java interrompe o serviço executor quando uma de suas tarefas atribuídas falha por qualquer motivo

12

Preciso de algum tipo de serviço que execute algumas tarefas simultaneamente e em um intervalo de 1 segundo por 1 minuto.

Se uma das tarefas falhar, quero interromper o serviço e todas as tarefas executadas com algum tipo de indicador de que algo deu errado; caso contrário, após um minuto tudo correu bem, o serviço será interrompido com um indicador de que tudo correu bem.

Por exemplo, eu tenho 2 funções:

Runnable task1 = ()->{
      int num = Math.rand(1,100);
      if (num < 5){
          throw new Exception("something went wrong with this task,terminate");
      }
}

Runnable task2 = ()->{
      int num = Math.rand(1,100)
      return num < 50;
}



ScheduledExecutorService scheduledExecutorService = Executors.newScheduledThreadPool(2);
task1schedule = scheduledExecutorService.scheduleAtFixedRate(task1, 1, 60, TimeUnit.SECONDS);
task2schedule = scheduledExecutorService.scheduleAtFixedRate(task2, 1, 60, TimeUnit.SECONDS);

if (!task1schedule || !task2schedule) scheduledExecutorService.shutdown();

Alguma idéia de como devo lidar com isso e tornar as coisas o mais genéricas possível?

totothegreat
fonte
11
Poucas coisas além da pergunta real, Math.randnão é uma API interna. Uma implementação de Runnabledeve ter uma void rundefinição. O tipo de task1/2scheduleestaria ScheduledFuture<?>no contexto fornecido. Passando para a pergunta real, como ela é usada awaitTermination? Você poderia fazer isso como scheduledExecutorService.awaitTermination(1,TimeUnit.MINUTES);. Como alternativa, que tal verificar se alguma das tarefas foi cancelada antes de sua conclusão normal if (task1schedule.isCancelled() || task2schedule.isCancelled()) scheduledExecutorService.shutdown();:?
Naman
2
Não faz sentido agendar tarefas a serem repetidas a cada minuto, mas, digamos, você deseja interromper as tarefas "se depois de um minuto tudo correr bem". Como você está parando o executor nos dois casos, o agendamento de uma tarefa que desliga o executor após um minuto é trivial. E o futuro já indica se algo deu errado ou não. Você não disse que outro tipo de indicador deseja.
Holger

Respostas:

8

A ideia é que as tarefas sejam enviadas para um objeto comum TaskCompleteEvent. Se eles enviarem um erro, o planejador será parado e todas as tarefas serão interrompidas.

Você pode verificar os resultados de cada iteração de tarefas nos mapas "erros" e "sucesso".

public class SchedulerTest {

    @Test
    public void scheduler() throws InterruptedException {
        ScheduledExecutorService scheduledExecutorService = Executors.newScheduledThreadPool(2);
        TaskCompleteEvent taskCompleteEvent = new TaskCompleteEvent(scheduledExecutorService);
        Runnable task1 = () -> {
            int num = new Random().nextInt(100);
            if (num < 5) {
                taskCompleteEvent.message("task1-"+UUID.randomUUID().toString(), "Num "+num+" was obatined. Breaking all the executions.", true);
            }
        };
        Runnable task2 = () -> {
            int num = new Random().nextInt(100);
            taskCompleteEvent.message("task2-"+UUID.randomUUID().toString(), num < 50, false);
        };
        scheduledExecutorService.scheduleAtFixedRate(task1, 0, 1, TimeUnit.SECONDS);
        scheduledExecutorService.scheduleAtFixedRate(task2, 0, 1, TimeUnit.SECONDS);
        scheduledExecutorService.awaitTermination(60, TimeUnit.SECONDS);
        System.out.println("Success: "+taskCompleteEvent.getSuccess());
        System.out.println("Errors: "+taskCompleteEvent.getErrors());
        System.out.println("Went well?: "+taskCompleteEvent.getErrors().isEmpty());
    }

    public static class TaskCompleteEvent {

        private final ScheduledExecutorService scheduledExecutorService;
        private final Map<String, Object> errors = new LinkedHashMap<>();
        private final Map<String, Object> success = new LinkedHashMap<>();

        public TaskCompleteEvent(ScheduledExecutorService scheduledExecutorService) {
            this.scheduledExecutorService = scheduledExecutorService;
        }

        public synchronized void message(String id, Object response, boolean error) {
            if (error) {
                errors.put(id, response);
                scheduledExecutorService.shutdown();
            } else {
                success.put(id, response);
            }
        }

        public synchronized Map<String, Object> getErrors() {
            return errors;
        }

        public synchronized Map<String, Object> getSuccess() {
            return success;
        }

    }

}
Ravenskater
fonte
2

Você só precisa adicionar uma tarefa adicional cujo trabalho é monitorar todas as outras tarefas em execução - e quando alguma das tarefas monitoradas falha, elas precisam definir um semáforo (sinalizador) que o assassino pode inspecionar.

    ScheduledExecutorService executor = (ScheduledExecutorService) Executors.newScheduledThreadPool(2);

    // INSTANTIATE THE REMOTE-FILE-MONITOR:
    RemoteFileMonitor monitor = new RemoteFileMonitor(remotesource, localtarget);

    // THIS TimerTask PERIODICALLY TRIGGERS THE RemoteFileMonitor: 
    TimerTask remote = new TimerTask() {

        // RUN FORREST... RUN !
        public void run() {

            try { 

                kae.trace("TimerTask::run() --> Calling RemoteFileMonitor.check()");
                monitor.check();

            } catch (Exception ex) {

                // NULL TRAP: ALLOWS US TO CONTINUE AND RETRY:

            }

        }

    };

    // THIS TimerTask PERIODICALLY TRIES TO KILL THE REMOTE-FILE-MONITOR:
    TimerTask assassin = new TimerTask() {

        // WHERE DO BAD FOLKS GO WHEN THEY DIE ? 
        private final LocalDateTime death = LocalDateTime.now().plus(ConfigurationOptions.getPollingCycleTime(), ChronoUnit.MINUTES);

        // RUN FORREST... RUN !
        public void run() {

            // IS THERE LIFE AFTER DEATH ???
            if (LocalDateTime.now().isAfter(death)) {

                // THEY GO TO A LAKE OF FIRE AND FRY:
                kae.error(ReturnCode.MONITOR_POLLING_CYCLE_EXCEEDED);                   

            }

        }

    };

    // SCHEDULE THE PERIODIC EXECUTION OF THE RemoteFileMonitor: (remote --> run() monitor --> check())
    executor.scheduleAtFixedRate(remote, delay, interval, TimeUnit.MINUTES);

    // SCHEDULE PERIODIC ASSASSINATION ATTEMPTS AGAINST THE RemoteFileMonitor: (assassin --> run() --> after death --> die())
    executor.scheduleAtFixedRate(assassin, delay, 60L, TimeUnit.SECONDS);

    // LOOP UNTIL THE MONITOR COMPLETES:
    do {

        try {

            // I THINK I NEED A NAP:
            Thread.sleep(interval * 10);                

        } catch (InterruptedException e) {

            // FAIL && THEN cleanexit();
            kae.error(ReturnCode.MONITORING_ERROR, "Monitoring of the XXXXXX-Ingestion site was interrupted");

        }

        // NOTE: THE MONITOR IS SET TO 'FINISHED' WHEN THE DONE-File IS DELIVERED AND RETRIEVED:
    } while (monitor.isNotFinished());

    // SHUTDOWN THE MONITOR TASK:
    executor.shutdown();
Greg Patnude
fonte
2
A classe não TimerTasktem nenhuma relação com ScheduledExecutorService; apenas acontece de implementar Runnable. Além disso, não faz sentido agendar uma tarefa periódica, apenas para verificar se um horário específico ( ConfigurationOptions.getPollingCycleTime()) foi atingido. Você tem um ScheduledExecutorService, para que possa agendar a tarefa corretamente para o tempo desejado.
Holger
A implementação no exemplo que usei foi matar uma tarefa de execução após um certo período de tempo, se a tarefa não tivesse sido concluída. O caso de uso foi: Se o servidor remoto não soltar um arquivo dentro de 2 horas - elimine a tarefa. é isso que o OP pediu.
Greg Patnude
Você leu e entendeu meu comentário? Não importa o que o código faça, ele usa uma classe desanimada sem motivo, apenas substitua TimerTaskpor Runnablee você corrigiu o problema, sem alterar o que o código faz. Além disso, basta usar executor.schedule(assassin, ConfigurationOptions.getPollingCycleTime(), ChronoUnit.MINUTES);e ele será executado uma vez no horário desejado; portanto, a if(LocalDateTime.now().isAfter(death))verificação é obsoleta. Novamente, isso não muda o que o código faz, além de fazê-lo substancialmente mais simples e mais eficiente.
Holger