Examples with JobManagerRegistration org.apache.flink.runtime.resourcemanager.registration.JobManagerRegistration used on opensource projects

Example 6 with JobManagerRegistration

use of org.apache.flink.runtime.resourcemanager.registration.JobManagerRegistration in project flink by apache.

the class ResourceManager method registerJobMasterInternal.

// ------------------------------------------------------------------------
// Internal methods
// ------------------------------------------------------------------------
/**
 * Registers a new JobMaster.
 *
 * @param jobMasterGateway to communicate with the registering JobMaster
 * @param jobId of the job for which the JobMaster is responsible
 * @param jobManagerAddress address of the JobMaster
 * @param jobManagerResourceId ResourceID of the JobMaster
 * @return RegistrationResponse
 */
private RegistrationResponse registerJobMasterInternal(final JobMasterGateway jobMasterGateway, JobID jobId, String jobManagerAddress, ResourceID jobManagerResourceId) {
    if (jobManagerRegistrations.containsKey(jobId)) {
        JobManagerRegistration oldJobManagerRegistration = jobManagerRegistrations.get(jobId);
        if (Objects.equals(oldJobManagerRegistration.getJobMasterId(), jobMasterGateway.getFencingToken())) {
            // same registration
            log.debug("Job manager {}@{} was already registered.", jobMasterGateway.getFencingToken(), jobManagerAddress);
        } else {
            // tell old job manager that he is no longer the job leader
            closeJobManagerConnection(oldJobManagerRegistration.getJobID(), ResourceRequirementHandling.RETAIN, new Exception("New job leader for job " + jobId + " found."));
            JobManagerRegistration jobManagerRegistration = new JobManagerRegistration(jobId, jobManagerResourceId, jobMasterGateway);
            jobManagerRegistrations.put(jobId, jobManagerRegistration);
            jmResourceIdRegistrations.put(jobManagerResourceId, jobManagerRegistration);
        }
    } else {
        // new registration for the job
        JobManagerRegistration jobManagerRegistration = new JobManagerRegistration(jobId, jobManagerResourceId, jobMasterGateway);
        jobManagerRegistrations.put(jobId, jobManagerRegistration);
        jmResourceIdRegistrations.put(jobManagerResourceId, jobManagerRegistration);
    }
    log.info("Registered job manager {}@{} for job {}.", jobMasterGateway.getFencingToken(), jobManagerAddress, jobId);
    jobManagerHeartbeatManager.monitorTarget(jobManagerResourceId, new JobMasterHeartbeatSender(jobMasterGateway));
    return new JobMasterRegistrationSuccess(getFencingToken(), resourceId);
}