Skip to content

Commit 0714657

Browse files
adamantalzmolnar-inf
authored andcommitted
CDPD-39756. YARN-10950. Code cleanup in QueueCapacities (apache#3454)
(cherry picked from commit a9b2469) Change-Id: Ife0d60af3022a4653c2c297865bb703699c70e36 (cherry picked from commit 9011b874bfe827a40afcebdf6d2b313da7ee2b4b)
1 parent b692189 commit 0714657

File tree

1 file changed

+18
-33
lines changed
  • hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/scheduler/capacity

1 file changed

+18
-33
lines changed

hadoop-yarn-project/hadoop-yarn/hadoop-yarn-server/hadoop-yarn-server-resourcemanager/src/main/java/org/apache/hadoop/yarn/server/resourcemanager/scheduler/capacity/QueueCapacities.java

Lines changed: 18 additions & 33 deletions
Original file line numberDiff line numberDiff line change
@@ -33,17 +33,16 @@
3333
public class QueueCapacities {
3434
private static final String NL = CommonNodeLabelsManager.NO_LABEL;
3535
private static final float LABEL_DOESNT_EXIST_CAP = 0f;
36-
private Map<String, Capacities> capacitiesMap;
37-
private ReadLock readLock;
38-
private WriteLock writeLock;
36+
private final Map<String, Capacities> capacitiesMap;
37+
private final ReadLock readLock;
38+
private final WriteLock writeLock;
3939
private final boolean isRoot;
4040

4141
public QueueCapacities(boolean isRoot) {
4242
ReentrantReadWriteLock lock = new ReentrantReadWriteLock();
4343
readLock = lock.readLock();
4444
writeLock = lock.writeLock();
45-
46-
capacitiesMap = new HashMap<String, Capacities>();
45+
capacitiesMap = new HashMap<>();
4746
this.isRoot = isRoot;
4847
}
4948

@@ -52,15 +51,15 @@ private enum CapacityType {
5251
USED_CAP(0), ABS_USED_CAP(1), MAX_CAP(2), ABS_MAX_CAP(3), CAP(4), ABS_CAP(5),
5352
MAX_AM_PERC(6), RESERVED_CAP(7), ABS_RESERVED_CAP(8), WEIGHT(9), NORMALIZED_WEIGHT(10);
5453

55-
private int idx;
54+
private final int idx;
5655

57-
private CapacityType(int idx) {
56+
CapacityType(int idx) {
5857
this.idx = idx;
5958
}
6059
}
6160

6261
private static class Capacities {
63-
private float[] capacitiesArr;
62+
private final float[] capacitiesArr;
6463

6564
public Capacities() {
6665
capacitiesArr = new float[CapacityType.values().length];
@@ -71,19 +70,17 @@ public Capacities() {
7170

7271
@Override
7372
public String toString() {
74-
StringBuilder sb = new StringBuilder();
75-
sb.append("{used=" + capacitiesArr[0] + "%, ");
76-
sb.append("abs_used=" + capacitiesArr[1] + "%, ");
77-
sb.append("max_cap=" + capacitiesArr[2] + "%, ");
78-
sb.append("abs_max_cap=" + capacitiesArr[3] + "%, ");
79-
sb.append("cap=" + capacitiesArr[4] + "%, ");
80-
sb.append("abs_cap=" + capacitiesArr[5] + "%}");
81-
sb.append("max_am_perc=" + capacitiesArr[6] + "%}");
82-
sb.append("reserved_cap=" + capacitiesArr[7] + "%}");
83-
sb.append("abs_reserved_cap=" + capacitiesArr[8] + "%}");
84-
sb.append("weight=" + capacitiesArr[9] + "w, ");
85-
sb.append("normalized_weight=" + capacitiesArr[10] + "w}");
86-
return sb.toString();
73+
return "{used=" + capacitiesArr[0] + "%, " +
74+
"abs_used=" + capacitiesArr[1] + "%, " +
75+
"max_cap=" + capacitiesArr[2] + "%, " +
76+
"abs_max_cap=" + capacitiesArr[3] + "%, " +
77+
"cap=" + capacitiesArr[4] + "%, " +
78+
"abs_cap=" + capacitiesArr[5] + "%, " +
79+
"max_am_perc=" + capacitiesArr[6] + "%, " +
80+
"reserved_cap=" + capacitiesArr[7] + "%, " +
81+
"abs_reserved_cap=" + capacitiesArr[8] + "%, " +
82+
"weight=" + capacitiesArr[9] + "w, " +
83+
"normalized_weight=" + capacitiesArr[10] + "w}";
8784
}
8885
}
8986

@@ -118,7 +115,6 @@ private void _set(String label, CapacityType type, float value) {
118115
}
119116
}
120117

121-
/* Used Capacity Getter and Setter */
122118
public float getUsedCapacity() {
123119
return _get(NL, CapacityType.USED_CAP);
124120
}
@@ -135,7 +131,6 @@ public void setUsedCapacity(String label, float value) {
135131
_set(label, CapacityType.USED_CAP, value);
136132
}
137133

138-
/* Absolute Used Capacity Getter and Setter */
139134
public float getAbsoluteUsedCapacity() {
140135
return _get(NL, CapacityType.ABS_USED_CAP);
141136
}
@@ -152,7 +147,6 @@ public void setAbsoluteUsedCapacity(String label, float value) {
152147
_set(label, CapacityType.ABS_USED_CAP, value);
153148
}
154149

155-
/* Capacity Getter and Setter */
156150
public float getCapacity() {
157151
return _get(NL, CapacityType.CAP);
158152
}
@@ -173,7 +167,6 @@ public void setCapacity(String label, float value) {
173167
_set(label, CapacityType.CAP, value);
174168
}
175169

176-
/* Absolute Capacity Getter and Setter */
177170
public float getAbsoluteCapacity() {
178171
return _get(NL, CapacityType.ABS_CAP);
179172
}
@@ -193,7 +186,6 @@ public void setAbsoluteCapacity(String label, float value) {
193186
_set(label, CapacityType.ABS_CAP, value);
194187
}
195188

196-
/* Maximum Capacity Getter and Setter */
197189
public float getMaximumCapacity() {
198190
return _get(NL, CapacityType.MAX_CAP);
199191
}
@@ -210,7 +202,6 @@ public void setMaximumCapacity(String label, float value) {
210202
_set(label, CapacityType.MAX_CAP, value);
211203
}
212204

213-
/* Absolute Maximum Capacity Getter and Setter */
214205
public float getAbsoluteMaximumCapacity() {
215206
return _get(NL, CapacityType.ABS_MAX_CAP);
216207
}
@@ -227,8 +218,6 @@ public void setAbsoluteMaximumCapacity(String label, float value) {
227218
_set(label, CapacityType.ABS_MAX_CAP, value);
228219
}
229220

230-
/* Absolute Maximum AM resource percentage Getter and Setter */
231-
232221
public float getMaxAMResourcePercentage() {
233222
return _get(NL, CapacityType.MAX_AM_PERC);
234223
}
@@ -245,7 +234,6 @@ public void setMaxAMResourcePercentage(float value) {
245234
_set(NL, CapacityType.MAX_AM_PERC, value);
246235
}
247236

248-
/* Reserved Capacity Getter and Setter */
249237
public float getReservedCapacity() {
250238
return _get(NL, CapacityType.RESERVED_CAP);
251239
}
@@ -262,7 +250,6 @@ public void setReservedCapacity(String label, float value) {
262250
_set(label, CapacityType.RESERVED_CAP, value);
263251
}
264252

265-
/* Absolute Reserved Capacity Getter and Setter */
266253
public float getAbsoluteReservedCapacity() {
267254
return _get(NL, CapacityType.ABS_RESERVED_CAP);
268255
}
@@ -279,7 +266,6 @@ public void setAbsoluteReservedCapacity(String label, float value) {
279266
_set(label, CapacityType.ABS_RESERVED_CAP, value);
280267
}
281268

282-
/* Weight Getter and Setter */
283269
public float getWeight() {
284270
return _get(NL, CapacityType.WEIGHT);
285271
}
@@ -296,7 +282,6 @@ public void setWeight(String label, float value) {
296282
_set(label, CapacityType.WEIGHT, value);
297283
}
298284

299-
/* Weight Getter and Setter */
300285
public float getNormalizedWeight() {
301286
return _get(NL, CapacityType.NORMALIZED_WEIGHT);
302287
}

0 commit comments

Comments
 (0)