/// <exception cref="System.IO.IOException"/>
public override void StoreNewClusterNodeLabels(ICollection <string> labels)
{
EnsureAppendEditlogFile();
editlogOs.WriteInt((int)(FileSystemNodeLabelsStore.SerializedLogType.AddLabels));
((AddToClusterNodeLabelsRequestPBImpl)AddToClusterNodeLabelsRequest.NewInstance(labels
)).GetProto().WriteDelimitedTo(editlogOs);
EnsureCloseEditlogFile();
}
/// <exception cref="Org.Apache.Hadoop.Yarn.Exceptions.YarnException"/>
/// <exception cref="System.IO.IOException"/>
public virtual AddToClusterNodeLabelsResponse AddToClusterNodeLabels(AddToClusterNodeLabelsRequest
request)
{
YarnServerResourceManagerServiceProtos.AddToClusterNodeLabelsRequestProto requestProto
= ((AddToClusterNodeLabelsRequestPBImpl)request).GetProto();
try
{
return(new AddToClusterNodeLabelsResponsePBImpl(proxy.AddToClusterNodeLabels(null
, requestProto)));
}
catch (ServiceException e)
{
RPCUtil.UnwrapAndThrowException(e);
return(null);
}
}
/// <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);
}
/// <exception cref="Org.Apache.Hadoop.Yarn.Exceptions.YarnException"/>
/// <exception cref="System.IO.IOException"/>
public virtual AddToClusterNodeLabelsResponse AddToClusterNodeLabels(AddToClusterNodeLabelsRequest
request)
{
string argName = "addToClusterNodeLabels";
string msg = "add labels.";
UserGroupInformation user = CheckAcls(argName);
CheckRMStatus(user.GetShortUserName(), argName, msg);
AddToClusterNodeLabelsResponse response = recordFactory.NewRecordInstance <AddToClusterNodeLabelsResponse
>();
try
{
rmContext.GetNodeLabelManager().AddToCluserNodeLabels(request.GetNodeLabels());
RMAuditLogger.LogSuccess(user.GetShortUserName(), argName, "AdminService");
return(response);
}
catch (IOException ioe)
{
throw LogAndWrapException(ioe, user.GetShortUserName(), argName, msg);
}
}
