YARN-1723. AMRMClientAsync missing blacklist addition and removal

functionality. Contributed by Bartosz Ługowski.
This commit is contained in:
Siddharth Seth 2015-02-04 12:52:02 -08:00
parent 40a415799b
commit e04e8fa810
3 changed files with 27 additions and 0 deletions

View File

@ -239,6 +239,9 @@ Release 2.7.0 - UNRELEASED
YARN-1393. SLS: Add how-to-use instructions. (Wei Yan via kasha)
YARN-1723. AMRMClientAsync missing blacklist addition and removal
functionality. (Bartosz Ługowski via sseth)
OPTIMIZATIONS
BUG FIXES

View File

@ -194,6 +194,17 @@ public abstract void unregisterApplicationMaster(
*/
public abstract int getClusterNodeCount();
/**
* Update application's blacklist with addition or removal resources.
*
* @param blacklistAdditions list of resources which should be added to the
* application blacklist
* @param blacklistRemovals list of resources which should be removed from the
* application blacklist
*/
public abstract void updateBlacklist(List<String> blacklistAdditions,
List<String> blacklistRemovals);
/**
* Wait for <code>check</code> to return true for each 1000 ms.
* See also {@link #waitFor(com.google.common.base.Supplier, int)}

View File

@ -206,6 +206,19 @@ public int getClusterNodeCount() {
return client.getClusterNodeCount();
}
/**
* Update application's blacklist with addition or removal resources.
*
* @param blacklistAdditions list of resources which should be added to the
* application blacklist
* @param blacklistRemovals list of resources which should be removed from the
* application blacklist
*/
public void updateBlacklist(List<String> blacklistAdditions,
List<String> blacklistRemovals) {
client.updateBlacklist(blacklistAdditions, blacklistRemovals);
}
private class HeartbeatThread extends Thread {
public HeartbeatThread() {
super("AMRM Heartbeater thread");