MAPREDUCE-7151. RMContainerAllocator#handleJobPriorityChange expects application_priority always. Contributed by Bilwa S T.
This commit is contained in:
parent
c1874046e2
commit
74a5e683fe
@ -1020,12 +1020,14 @@ private void handleUpdatedNodes(AllocateResponse response) {
|
||||
}
|
||||
}
|
||||
|
||||
private void handleJobPriorityChange(AllocateResponse response) {
|
||||
Priority priorityFromResponse = Priority.newInstance(response
|
||||
.getApplicationPriority().getPriority());
|
||||
|
||||
// Update the job priority to Job directly.
|
||||
getJob().setJobPriority(priorityFromResponse);
|
||||
void handleJobPriorityChange(AllocateResponse response) {
|
||||
Priority applicationPriority = response.getApplicationPriority();
|
||||
if (null != applicationPriority) {
|
||||
Priority priorityFromResponse = Priority
|
||||
.newInstance(applicationPriority.getPriority());
|
||||
// Update the job priority to Job directly.
|
||||
getJob().setJobPriority(priorityFromResponse);
|
||||
}
|
||||
}
|
||||
|
||||
@Private
|
||||
|
@ -2160,6 +2160,20 @@ protected AllocateResponse makeRemoteRequest() throws IOException,
|
||||
}
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testIfApplicationPriorityIsNotSet() {
|
||||
Job mockJob = mock(Job.class);
|
||||
RMCommunicator communicator = mock(RMCommunicator.class);
|
||||
ClientService service = mock(ClientService.class);
|
||||
AppContext context = mock(AppContext.class);
|
||||
AMPreemptionPolicy policy = mock(AMPreemptionPolicy.class);
|
||||
when(communicator.getJob()).thenReturn(mockJob);
|
||||
RMContainerAllocator allocator = new RMContainerAllocator(service, context,
|
||||
policy);
|
||||
AllocateResponse response = Records.newRecord(AllocateResponse.class);
|
||||
allocator.handleJobPriorityChange(response);
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testReduceScheduling() throws Exception {
|
||||
int totalMaps = 10;
|
||||
|
Loading…
Reference in New Issue
Block a user