/// <exception cref="System.IO.IOException"/>
public virtual void TestNormalizeNodeLabelExpression()
{
// mock queue and scheduler
YarnScheduler scheduler = Org.Mockito.Mockito.Mock <YarnScheduler>();
ICollection <string> queueAccessibleNodeLabels = Sets.NewHashSet();
QueueInfo queueInfo = Org.Mockito.Mockito.Mock <QueueInfo>();
Org.Mockito.Mockito.When(queueInfo.GetQueueName()).ThenReturn("queue");
Org.Mockito.Mockito.When(queueInfo.GetAccessibleNodeLabels()).ThenReturn(queueAccessibleNodeLabels
);
Org.Mockito.Mockito.When(queueInfo.GetDefaultNodeLabelExpression()).ThenReturn(" x "
);
Org.Mockito.Mockito.When(scheduler.GetQueueInfo(Matchers.Any <string>(), Matchers.AnyBoolean
(), Matchers.AnyBoolean())).ThenReturn(queueInfo);
Org.Apache.Hadoop.Yarn.Api.Records.Resource maxResource = Resources.CreateResource
(YarnConfiguration.DefaultRmSchedulerMaximumAllocationMb, YarnConfiguration.DefaultRmSchedulerMaximumAllocationVcores
);
// queue has labels, success cases
try
{
// set queue accessible node labels to [x, y]
queueAccessibleNodeLabels.Clear();
Sharpen.Collections.AddAll(queueAccessibleNodeLabels, Arrays.AsList("x", "y"));
rmContext.GetNodeLabelManager().AddToCluserNodeLabels(ImmutableSet.Of("x", "y"));
Org.Apache.Hadoop.Yarn.Api.Records.Resource resource = Resources.CreateResource(0
, YarnConfiguration.DefaultRmSchedulerMinimumAllocationVcores);
ResourceRequest resReq = BuilderUtils.NewResourceRequest(Org.Mockito.Mockito.Mock
<Priority>(), ResourceRequest.Any, resource, 1);
SchedulerUtils.NormalizeAndvalidateRequest(resReq, maxResource, "queue", scheduler
, rmContext);
NUnit.Framework.Assert.IsTrue(resReq.GetNodeLabelExpression().Equals("x"));
resReq.SetNodeLabelExpression(" y ");
SchedulerUtils.NormalizeAndvalidateRequest(resReq, maxResource, "queue", scheduler
, rmContext);
NUnit.Framework.Assert.IsTrue(resReq.GetNodeLabelExpression().Equals("y"));
}
catch (InvalidResourceRequestException e)
{
Sharpen.Runtime.PrintStackTrace(e);
NUnit.Framework.Assert.Fail("Should be valid when request labels is a subset of queue labels"
);
}
finally
{
rmContext.GetNodeLabelManager().RemoveFromClusterNodeLabels(Arrays.AsList("x", "y"
));
}
}
/// <summary>
/// Utility method to validate a resource request, by insuring that the
/// requested memory/vcore is non-negative and not greater than max
/// </summary>
/// <exception cref="Org.Apache.Hadoop.Yarn.Exceptions.InvalidResourceRequestException
/// ">when there is invalid request</exception>
private static void ValidateResourceRequest(ResourceRequest resReq, Org.Apache.Hadoop.Yarn.Api.Records.Resource
maximumResource, QueueInfo queueInfo, RMContext rmContext)
{
if (resReq.GetCapability().GetMemory() < 0 || resReq.GetCapability().GetMemory()
> maximumResource.GetMemory())
{
throw new InvalidResourceRequestException("Invalid resource request" + ", requested memory < 0"
+ ", or requested memory > max configured" + ", requestedMemory=" + resReq.GetCapability
().GetMemory() + ", maxMemory=" + maximumResource.GetMemory());
}
if (resReq.GetCapability().GetVirtualCores() < 0 || resReq.GetCapability().GetVirtualCores
() > maximumResource.GetVirtualCores())
{
throw new InvalidResourceRequestException("Invalid resource request" + ", requested virtual cores < 0"
+ ", or requested virtual cores > max configured" + ", requestedVirtualCores="
+ resReq.GetCapability().GetVirtualCores() + ", maxVirtualCores=" + maximumResource
.GetVirtualCores());
}
string labelExp = resReq.GetNodeLabelExpression();
// we don't allow specify label expression other than resourceName=ANY now
if (!ResourceRequest.Any.Equals(resReq.GetResourceName()) && labelExp != null &&
!labelExp.Trim().IsEmpty())
{
throw new InvalidResourceRequestException("Invailid resource request, queue=" + queueInfo
.GetQueueName() + " specified node label expression in a " + "resource request has resource name = "
+ resReq.GetResourceName());
}
// we don't allow specify label expression with more than one node labels now
if (labelExp != null && labelExp.Contains("&&"))
{
throw new InvalidResourceRequestException("Invailid resource request, queue=" + queueInfo
.GetQueueName() + " specified more than one node label " + "in a node label expression, node label expression = "
+ labelExp);
}
if (labelExp != null && !labelExp.Trim().IsEmpty() && queueInfo != null)
{
if (!CheckQueueLabelExpression(queueInfo.GetAccessibleNodeLabels(), labelExp, rmContext
))
{
throw new InvalidResourceRequestException("Invalid resource request" + ", queue="
+ queueInfo.GetQueueName() + " doesn't have permission to access all labels " +
"in resource request. labelExpression of resource request=" + labelExp + ". Queue labels="
+ (queueInfo.GetAccessibleNodeLabels() == null ? string.Empty : StringUtils.Join
(queueInfo.GetAccessibleNodeLabels().GetEnumerator(), ',')));
}
}
}
private static void NormalizeNodeLabelExpressionInRequest(ResourceRequest resReq,
QueueInfo queueInfo)
{
string labelExp = resReq.GetNodeLabelExpression();
// if queue has default label expression, and RR doesn't have, use the
// default label expression of queue
if (labelExp == null && queueInfo != null && ResourceRequest.Any.Equals(resReq.GetResourceName
()))
{
labelExp = queueInfo.GetDefaultNodeLabelExpression();
}
// If labelExp still equals to null, set it to be NO_LABEL
if (labelExp == null)
{
labelExp = RMNodeLabelsManager.NoLabel;
}
resReq.SetNodeLabelExpression(labelExp);
}
/// <exception cref="Org.Apache.Hadoop.Yarn.Exceptions.InvalidResourceRequestException
/// "/>
private ResourceRequest ValidateAndCreateResourceRequest(ApplicationSubmissionContext
submissionContext, bool isRecovery)
{
// Validation of the ApplicationSubmissionContext needs to be completed
// here. Only those fields that are dependent on RM's configuration are
// checked here as they have to be validated whether they are part of new
// submission or just being recovered.
// Check whether AM resource requirements are within required limits
if (!submissionContext.GetUnmanagedAM())
{
ResourceRequest amReq = submissionContext.GetAMContainerResourceRequest();
if (amReq == null)
{
amReq = BuilderUtils.NewResourceRequest(RMAppAttemptImpl.AmContainerPriority, ResourceRequest
.Any, submissionContext.GetResource(), 1);
}
// set label expression for AM container
if (null == amReq.GetNodeLabelExpression())
{
amReq.SetNodeLabelExpression(submissionContext.GetNodeLabelExpression());
}
try
{
SchedulerUtils.NormalizeAndValidateRequest(amReq, scheduler.GetMaximumResourceCapability
(), submissionContext.GetQueue(), scheduler, isRecovery, rmContext);
}
catch (InvalidResourceRequestException e)
{
Log.Warn("RM app submission failed in validating AM resource request" + " for application "
+ submissionContext.GetApplicationId(), e);
throw;
}
SchedulerUtils.NormalizeRequest(amReq, scheduler.GetResourceCalculator(), scheduler
.GetClusterResource(), scheduler.GetMinimumResourceCapability(), scheduler.GetMaximumResourceCapability
(), scheduler.GetMinimumResourceCapability());
return(amReq);
}
return(null);
}
