/// <exception cref="System.IO.IOException"/>
/// <exception cref="Org.Apache.Hadoop.Yarn.Exceptions.YarnException"/>
private int AddToClusterNodeLabels(string args)
{
ICollection <string> labels = BuildNodeLabelsSetFromStr(args);
if (directlyAccessNodeLabelStore)
{
GetNodeLabelManagerInstance(GetConf()).AddToCluserNodeLabels(labels);
}
else
{
ResourceManagerAdministrationProtocol adminProtocol = CreateAdminProtocol();
AddToClusterNodeLabelsRequest request = AddToClusterNodeLabelsRequest.NewInstance
(labels);
adminProtocol.AddToClusterNodeLabels(request);
}
return(0);
}
AddToClusterNodeLabels(RpcController controller, YarnServerResourceManagerServiceProtos.AddToClusterNodeLabelsRequestProto
proto)
{
AddToClusterNodeLabelsRequestPBImpl request = new AddToClusterNodeLabelsRequestPBImpl
(proto);
try
{
AddToClusterNodeLabelsResponse response = real.AddToClusterNodeLabels(request);
return(((AddToClusterNodeLabelsResponsePBImpl)response).GetProto());
}
catch (YarnException e)
{
throw new ServiceException(e);
}
catch (IOException e)
{
throw new ServiceException(e);
}
}
public virtual void Configure()
{
remoteAdminServiceAccessed = false;
admin = Org.Mockito.Mockito.Mock <ResourceManagerAdministrationProtocol>();
Org.Mockito.Mockito.When(admin.AddToClusterNodeLabels(Matchers.Any <AddToClusterNodeLabelsRequest
>())).ThenAnswer(new _Answer_79(this));
haadmin = Org.Mockito.Mockito.Mock <HAServiceProtocol>();
Org.Mockito.Mockito.When(haadmin.GetServiceStatus()).ThenReturn(new HAServiceStatus
(HAServiceProtocol.HAServiceState.Initializing));
HAServiceTarget haServiceTarget = Org.Mockito.Mockito.Mock <HAServiceTarget>();
Org.Mockito.Mockito.When(haServiceTarget.GetProxy(Matchers.Any <Configuration>(),
Matchers.AnyInt())).ThenReturn(haadmin);
rmAdminCLI = new _RMAdminCLI_96(this, haServiceTarget, new Configuration());
InitDummyNodeLabelsManager();
rmAdminCLI.localNodeLabelsManager = dummyNodeLabelsManager;
YarnConfiguration conf = new YarnConfiguration();
conf.SetBoolean(YarnConfiguration.RmHaEnabled, true);
rmAdminCLIWithHAEnabled = new _RMAdminCLI_113(this, haServiceTarget, conf);
}
