YARN-10139. ValidateAndGetSchedulerConfiguration API fails when cluster max allocation > default 8GB. Contributed by Prabhu Joseph.
(cherry picked from commit 6526f95bd2
)
This commit is contained in:
parent
b5321f1600
commit
de63115a2a
|
@ -31,6 +31,7 @@ import java.util.Collections;
|
|||
import java.util.EnumSet;
|
||||
import java.util.HashMap;
|
||||
import java.util.HashSet;
|
||||
import java.util.Iterator;
|
||||
import java.util.List;
|
||||
import java.util.Map;
|
||||
import java.util.Map.Entry;
|
||||
|
@ -2391,14 +2392,22 @@ public class RMWebServices extends WebServices implements RMWebServiceProtocol {
|
|||
((MutableConfScheduler) scheduler).getMutableConfProvider();
|
||||
Configuration schedulerConf = mutableConfigurationProvider
|
||||
.getConfiguration();
|
||||
Configuration newConfig = mutableConfigurationProvider
|
||||
Configuration newSchedulerConf = mutableConfigurationProvider
|
||||
.applyChanges(schedulerConf, mutationInfo);
|
||||
Configuration yarnConf = ((CapacityScheduler) scheduler).getConf();
|
||||
|
||||
Configuration newConfig = new Configuration(yarnConf);
|
||||
Iterator<Map.Entry<String, String>> iter = newSchedulerConf.iterator();
|
||||
Entry<String, String> e = null;
|
||||
while (iter.hasNext()) {
|
||||
e = iter.next();
|
||||
newConfig.set(e.getKey(), e.getValue());
|
||||
}
|
||||
CapacitySchedulerConfigValidator.validateCSConfiguration(yarnConf,
|
||||
newConfig, rm.getRMContext());
|
||||
|
||||
return Response.status(Status.OK)
|
||||
.entity(new ConfInfo(newConfig))
|
||||
.entity(new ConfInfo(newSchedulerConf))
|
||||
.build();
|
||||
} catch (Exception e) {
|
||||
String errorMsg = "CapacityScheduler configuration validation failed:"
|
||||
|
|
|
@ -731,6 +731,32 @@ public class TestRMWebServicesConfigurationMutation extends JerseyTestBase {
|
|||
newCSConf.getMaximumSystemApplications());
|
||||
}
|
||||
|
||||
@Test
|
||||
public void testValidateWithClusterMaxAllocation() throws Exception {
|
||||
WebResource r = resource();
|
||||
int clusterMax = YarnConfiguration.
|
||||
DEFAULT_RM_SCHEDULER_MAXIMUM_ALLOCATION_MB * 2;
|
||||
conf.setInt(YarnConfiguration.RM_SCHEDULER_MAXIMUM_ALLOCATION_MB,
|
||||
clusterMax);
|
||||
|
||||
SchedConfUpdateInfo updateInfo = new SchedConfUpdateInfo();
|
||||
Map<String, String> updateParam = new HashMap<>();
|
||||
updateParam.put(CapacitySchedulerConfiguration.MAXIMUM_APPLICATIONS_SUFFIX,
|
||||
"100");
|
||||
QueueConfigInfo aUpdateInfo = new QueueConfigInfo("root.a", updateParam);
|
||||
updateInfo.getUpdateQueueInfo().add(aUpdateInfo);
|
||||
|
||||
ClientResponse response =
|
||||
r.path("ws").path("v1").path("cluster")
|
||||
.path(RMWSConsts.SCHEDULER_CONF_VALIDATE)
|
||||
.queryParam("user.name", userName)
|
||||
.accept(MediaType.APPLICATION_JSON)
|
||||
.entity(YarnWebServiceUtils.toJson(updateInfo,
|
||||
SchedConfUpdateInfo.class), MediaType.APPLICATION_JSON)
|
||||
.post(ClientResponse.class);
|
||||
assertEquals(Status.OK.getStatusCode(), response.getStatus());
|
||||
}
|
||||
|
||||
@Override
|
||||
@After
|
||||
public void tearDown() throws Exception {
|
||||
|
|
Loading…
Reference in New Issue