Examples with ComponentMainThreadExecutor org.apache.flink.runtime.concurrent.ComponentMainThreadExecutor used on opensource projects

Example 11 with ComponentMainThreadExecutor

use of org.apache.flink.runtime.concurrent.ComponentMainThreadExecutor in project flink by apache.

the class OperatorCoordinatorHolderTest method testCheckpointFailsIfSendingEventFailedBeforeTrigger.

@Test
public void testCheckpointFailsIfSendingEventFailedBeforeTrigger() throws Exception {
    final ReorderableManualExecutorService executor = new ReorderableManualExecutorService();
    final ComponentMainThreadExecutor mainThreadExecutor = new ComponentMainThreadExecutorServiceAdapter((ScheduledExecutorService) executor, Thread.currentThread());
    CompletableFuture<Acknowledge> eventSendingResult = new CompletableFuture<>();
    final EventReceivingTasks tasks = EventReceivingTasks.createForRunningTasksWithRpcResult(eventSendingResult);
    final OperatorCoordinatorHolder holder = createCoordinatorHolder(tasks, TestingOperatorCoordinator::new, mainThreadExecutor);
    // Send one event without finishing it.
    getCoordinator(holder).getSubtaskGateway(0).sendEvent(new TestOperatorEvent(0));
    executor.triggerAll();
    // Finish the event sending. This will insert one runnable that handles
    // failed events to the executor. And we delay this runnable to
    // simulates checkpoints triggered before the failure get processed.
    executor.setDelayNewRunnables(true);
    eventSendingResult.completeExceptionally(new RuntimeException("Artificial"));
    executor.setDelayNewRunnables(false);
    // Trigger one checkpoint, the checkpoint should not be confirmed
    // before the failure get triggered.
    CompletableFuture<byte[]> checkpointResult = new CompletableFuture<>();
    holder.checkpointCoordinator(1, checkpointResult);
    executor.triggerAll();
    getCoordinator(holder).getLastTriggeredCheckpoint().complete(new byte[0]);
    executor.triggerAll();
    assertFalse(checkpointResult.isDone());
    // Then the failure finally get processed by fail the corresponding tasks.
    executor.executeAllDelayedRunnables();
    executor.triggerAll();
    // The checkpoint would be finally confirmed.
    assertTrue(checkpointResult.isCompletedExceptionally());
}