Skip to content

Commit e93ad36

Browse files
committed
YARN-11536. Fix CheckStyle.
1 parent eb5d99d commit e93ad36

File tree

4 files changed

+44
-24
lines changed

4 files changed

+44
-24
lines changed

hadoop-yarn-project/hadoop-yarn/hadoop-yarn-client/src/main/java/org/apache/hadoop/yarn/client/cli/RouterCLI.java

Lines changed: 21 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -349,6 +349,7 @@ private int handlePolicy(String[] args)
349349
opts.addOption(saveOpt);
350350
opts.addOption(batchSaveOpt);
351351
opts.addOption(formatOpt);
352+
opts.addOption(fileOpt);
352353

353354
// Parse command line arguments.
354355
CommandLine cliParser;
@@ -418,8 +419,20 @@ private int handleSavePolicy(String policy) {
418419
}
419420

420421
private int handBatchSavePolicies(String format, String policyFile) {
422+
423+
if(StringUtils.isBlank(format)) {
424+
LOG.error("Batch Save Federation Policies. Format is Empty.");
425+
return EXIT_ERROR;
426+
}
427+
428+
if(StringUtils.isBlank(policyFile)) {
429+
LOG.error("Batch Save Federation Policies. policyFile is Empty.");
430+
return EXIT_ERROR;
431+
}
432+
421433
LOG.info("Batch Save Federation Policies. Format = {}, PolicyFile = {}.",
422434
format, policyFile);
435+
423436
switch (format) {
424437
case FORMAT_XML:
425438
return parseXml2Policies(policyFile);
@@ -561,14 +574,18 @@ private String parsePolicyWeightsNode(Element queueElement, String weightType) {
561574
List<String> amRmPolicyWeights = new ArrayList<>();
562575
if (amrmPolicyWeightsNode.getNodeType() == Node.ELEMENT_NODE) {
563576
Element amrmPolicyWeightsElement = (Element) amrmPolicyWeightsNode;
564-
NodeList subClusterIdInfoList = amrmPolicyWeightsElement.getElementsByTagName(XML_TAG_SUBCLUSTERIDINFO);
577+
NodeList subClusterIdInfoList =
578+
amrmPolicyWeightsElement.getElementsByTagName(XML_TAG_SUBCLUSTERIDINFO);
565579
for (int i = 0; i < subClusterIdInfoList.getLength(); i++) {
566580
Node subClusterIdInfoNode = subClusterIdInfoList.item(i);
567581
if (subClusterIdInfoNode.getNodeType() == Node.ELEMENT_NODE) {
568582
Element subClusterIdInfoElement = (Element) subClusterIdInfoNode;
569-
String subClusterId = subClusterIdInfoElement.getElementsByTagName("id").item(0).getTextContent();
570-
String weight = subClusterIdInfoElement.getElementsByTagName("weight").item(0).getTextContent();
571-
LOG.debug("WeightType[{}] - SubCluster ID: {}, Weight: {}.", weightType, subClusterId, weight);
583+
String subClusterId =
584+
subClusterIdInfoElement.getElementsByTagName("id").item(0).getTextContent();
585+
String weight =
586+
subClusterIdInfoElement.getElementsByTagName("weight").item(0).getTextContent();
587+
LOG.debug("WeightType[{}] - SubCluster ID: {}, Weight: {}.",
588+
weightType, subClusterId, weight);
572589
amRmPolicyWeights.add(subClusterId + ":" + weight);
573590
}
574591
}

hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-router/src/main/java/org/apache/hadoop/yarn/server/router/rmadmin/FederationRMAdminInterceptor.java

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1057,6 +1057,7 @@ private void saveFederationQueuePolicy(FederationQueueWeight federationQueueWeig
10571057
*
10581058
* @param policyWeight policyWeight.
10591059
* @return Map of SubClusterWeight.
1060+
* @throws YarnException exceptions from yarn servers.
10601061
*/
10611062
protected Map<SubClusterIdInfo, Float> getSubClusterWeightMap(String policyWeight)
10621063
throws YarnException {

hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-router/src/test/java/org/apache/hadoop/yarn/server/router/TestRouterMetrics.java

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2267,11 +2267,13 @@ public void testGetBatchSaveFederationQueuePoliciesRetrieved() {
22672267
Assert.assertEquals(totalGoodBefore + 1,
22682268
metrics.getNumSucceededBatchSaveFederationQueuePoliciesRetrieved());
22692269
Assert.assertEquals(150,
2270-
metrics.getLatencySucceededBatchSaveFederationQueuePoliciesRetrieved(), ASSERT_DOUBLE_DELTA);
2270+
metrics.getLatencySucceededBatchSaveFederationQueuePoliciesRetrieved(),
2271+
ASSERT_DOUBLE_DELTA);
22712272
goodSubCluster.getBatchSaveFederationQueuePoliciesRetrieved(300);
22722273
Assert.assertEquals(totalGoodBefore + 2,
22732274
metrics.getNumSucceededBatchSaveFederationQueuePoliciesRetrieved());
22742275
Assert.assertEquals(225,
2275-
metrics.getLatencySucceededBatchSaveFederationQueuePoliciesRetrieved(), ASSERT_DOUBLE_DELTA);
2276+
metrics.getLatencySucceededBatchSaveFederationQueuePoliciesRetrieved(),
2277+
ASSERT_DOUBLE_DELTA);
22762278
}
22772279
}

hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-router/src/test/java/org/apache/hadoop/yarn/server/router/rmadmin/TestFederationRMAdminInterceptor.java

Lines changed: 18 additions & 18 deletions
Original file line numberDiff line numberDiff line change
@@ -673,14 +673,14 @@ public void testSaveFederationQueuePolicyRequest() throws IOException, YarnExcep
673673
public void testBatchSaveFederationQueuePoliciesRequest() throws IOException, YarnException {
674674

675675
// subClusters
676-
List<String> subClusters = new ArrayList<>();
677-
subClusters.add("SC-1");
678-
subClusters.add("SC-2");
676+
List<String> subClusterLists = new ArrayList<>();
677+
subClusterLists.add("SC-1");
678+
subClusterLists.add("SC-2");
679679

680680
// generate queue A, queue B, queue C
681-
FederationQueueWeight rootA = generateFederationQueueWeight("root.a", subClusters);
682-
FederationQueueWeight rootB = generateFederationQueueWeight("root.b", subClusters);
683-
FederationQueueWeight rootC = generateFederationQueueWeight("root.b", subClusters);
681+
FederationQueueWeight rootA = generateFederationQueueWeight("root.a", subClusterLists);
682+
FederationQueueWeight rootB = generateFederationQueueWeight("root.b", subClusterLists);
683+
FederationQueueWeight rootC = generateFederationQueueWeight("root.b", subClusterLists);
684684

685685
List<FederationQueueWeight> federationQueueWeights = new ArrayList<>();
686686
federationQueueWeights.add(rootA);
@@ -689,14 +689,14 @@ public void testBatchSaveFederationQueuePoliciesRequest() throws IOException, Ya
689689

690690
// Step1. Save Queue Policies in Batches
691691
BatchSaveFederationQueuePoliciesRequest request =
692-
BatchSaveFederationQueuePoliciesRequest.newInstance(federationQueueWeights);
692+
BatchSaveFederationQueuePoliciesRequest.newInstance(federationQueueWeights);
693693

694694
BatchSaveFederationQueuePoliciesResponse policiesResponse =
695695
interceptor.batchSaveFederationQueuePolicies(request);
696696

697-
Assert.assertNotNull(policiesResponse);
698-
Assert.assertNotNull(policiesResponse.getMessage());
699-
Assert.assertEquals("batch save policies success.", policiesResponse.getMessage());
697+
assertNotNull(policiesResponse);
698+
assertNotNull(policiesResponse.getMessage());
699+
assertEquals("batch save policies success.", policiesResponse.getMessage());
700700

701701
// Step2. We query Policy information from FederationStateStore.
702702
FederationStateStoreFacade federationFacade = interceptor.getFederationFacade();
@@ -748,13 +748,13 @@ public void testBatchSaveFederationQueuePoliciesRequest() throws IOException, Ya
748748
* We will generate the weight information of the queue.
749749
*
750750
* @param queue queue name
751-
* @param subClusters subClusters
751+
* @param pSubClusters subClusters
752752
* @return subCluster FederationQueueWeight
753753
*/
754754
private FederationQueueWeight generateFederationQueueWeight(
755-
String queue, List<String> subClusters) {
756-
String routerWeight = generatePolicyWeight(subClusters);
757-
String amrmWeight = generatePolicyWeight(subClusters);
755+
String queue, List<String> pSubClusters) {
756+
String routerWeight = generatePolicyWeight(pSubClusters);
757+
String amrmWeight = generatePolicyWeight(pSubClusters);
758758
String policyTypeName = WeightedLocalityPolicyManager.class.getCanonicalName();
759759
String headRoomAlpha = "1.0";
760760
return FederationQueueWeight.newInstance(routerWeight, amrmWeight, headRoomAlpha,
@@ -764,14 +764,14 @@ private FederationQueueWeight generateFederationQueueWeight(
764764
/**
765765
* Generating Policy Weight Data.
766766
*
767-
* @param subClusters set of sub-clusters.
767+
* @param pSubClusters set of sub-clusters.
768768
* @return policy Weight String, like SC-1:0.7,SC-2:0.
769769
*/
770-
private String generatePolicyWeight(List<String> subClusters) {
770+
private String generatePolicyWeight(List<String> pSubClusters) {
771771
List<String> weights = generateWeights(subClusters.size());
772772
List<String> subClusterWeight = new ArrayList<>();
773-
for (int i = 0; i < subClusters.size(); i++) {
774-
String subCluster = subClusters.get(i);
773+
for (int i = 0; i < pSubClusters.size(); i++) {
774+
String subCluster = pSubClusters.get(i);
775775
String weight = weights.get(i);
776776
subClusterWeight.add(subCluster + ":" + weight);
777777
}

0 commit comments

Comments
 (0)