Examples with JobStatusMessage org.apache.flink.runtime.client.JobStatusMessage used on opensource projects

Example 6 with JobStatusMessage

use of org.apache.flink.runtime.client.JobStatusMessage in project flink by apache.

the class ProcessFailureCancelingITCase method cancelRunningJob.

private void cancelRunningJob(ActorRef jobManager) throws Exception {
    final FiniteDuration askTimeout = new FiniteDuration(10, TimeUnit.SECONDS);
    // try at most for 30 seconds
    final long deadline = System.currentTimeMillis() + 30000;
    JobID jobId = null;
    do {
        Future<Object> response = Patterns.ask(jobManager, JobManagerMessages.getRequestRunningJobsStatus(), new Timeout(askTimeout));
        Object result;
        try {
            result = Await.result(response, askTimeout);
        } catch (Exception e) {
            throw new Exception("Could not retrieve running jobs from the JobManager.", e);
        }
        if (result instanceof JobManagerMessages.RunningJobsStatus) {
            List<JobStatusMessage> jobs = ((JobManagerMessages.RunningJobsStatus) result).getStatusMessages();
            if (jobs.size() == 1) {
                jobId = jobs.get(0).getJobId();
                break;
            }
        }
    } while (System.currentTimeMillis() < deadline);
    if (jobId == null) {
        // we never found it running, must have failed already
        return;
    }
    // tell the JobManager to cancel the job
    jobManager.tell(new JobManagerMessages.CancelJob(jobId), ActorRef.noSender());
}