All Downloads are FREE. Search and download functionalities are using the official Maven repository.
Please wait. This can take some minutes ...
Many resources are needed to download a project. Please understand that we have to compensate our server costs. Thank you in advance.
Project price only 1 $
You can buy this project and download/modify it how often you want.
com.yahoo.vespa.model.content.ClusterResourceLimits Maven / Gradle / Ivy
// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
package com.yahoo.vespa.model.content;
import com.yahoo.vespa.model.builder.xml.dom.ModelElement;
import com.yahoo.vespa.model.content.cluster.DomResourceLimitsBuilder;
import java.util.Optional;
import java.util.function.Consumer;
/**
* Class tracking the feed block resource limits for a content cluster.
*
* This includes the limits used by the cluster controller and the content nodes (proton).
*
* @author geirst
*/
public class ClusterResourceLimits {
private final ResourceLimits clusterControllerLimits;
private final ResourceLimits contentNodeLimits;
private ClusterResourceLimits(Builder builder) {
clusterControllerLimits = builder.ctrlBuilder.build();
contentNodeLimits = builder.nodeBuilder.build();
}
public ResourceLimits getClusterControllerLimits() {
return clusterControllerLimits;
}
public ResourceLimits getContentNodeLimits() {
return contentNodeLimits;
}
public static class Builder {
private final boolean hostedVespa;
private final double resourceLimitDisk;
private final double resourceLimitMemory;
private ResourceLimits.Builder ctrlBuilder = new ResourceLimits.Builder();
private ResourceLimits.Builder nodeBuilder = new ResourceLimits.Builder();
public Builder(boolean hostedVespa,
double resourceLimitDisk,
double resourceLimitMemory) {
this.hostedVespa = hostedVespa;
this.resourceLimitDisk = resourceLimitDisk;
this.resourceLimitMemory = resourceLimitMemory;
verifyLimits(resourceLimitDisk, resourceLimitMemory);
}
public ClusterResourceLimits build(ModelElement clusterElem) {
ctrlBuilder = createBuilder(clusterElem.childByPath("tuning"));
nodeBuilder = createBuilder(clusterElem.childByPath("engine.proton"));
deriveLimits();
return new ClusterResourceLimits(this);
}
private ResourceLimits.Builder createBuilder(ModelElement element) {
return element == null
? new ResourceLimits.Builder()
: DomResourceLimitsBuilder.createBuilder(element, hostedVespa);
}
public void setClusterControllerBuilder(ResourceLimits.Builder builder) {
ctrlBuilder = builder;
}
public void setContentNodeBuilder(ResourceLimits.Builder builder) {
nodeBuilder = builder;
}
public ClusterResourceLimits build() {
deriveLimits();
return new ClusterResourceLimits(this);
}
private void deriveLimits() {
// This also ensures that content nodes limits are derived according to the formula in calcContentNodeLimit().
considerSettingDefaultClusterControllerLimit(ctrlBuilder.getDiskLimit(),
nodeBuilder.getDiskLimit(),
ctrlBuilder::setDiskLimit,
resourceLimitDisk);
considerSettingDefaultClusterControllerLimit(ctrlBuilder.getMemoryLimit(),
nodeBuilder.getMemoryLimit(),
ctrlBuilder::setMemoryLimit,
resourceLimitMemory);
deriveClusterControllerLimit(ctrlBuilder.getDiskLimit(), nodeBuilder.getDiskLimit(), ctrlBuilder::setDiskLimit);
deriveClusterControllerLimit(ctrlBuilder.getMemoryLimit(), nodeBuilder.getMemoryLimit(), ctrlBuilder::setMemoryLimit);
deriveContentNodeLimit(nodeBuilder.getDiskLimit(), ctrlBuilder.getDiskLimit(), 0.6, nodeBuilder::setDiskLimit);
deriveContentNodeLimit(nodeBuilder.getMemoryLimit(), ctrlBuilder.getMemoryLimit(), 0.5, nodeBuilder::setMemoryLimit);
}
private void considerSettingDefaultClusterControllerLimit(Optional clusterControllerLimit,
Optional contentNodeLimit,
Consumer setter,
double resourceLimit) {
// TODO: remove this when feed block in distributor is default enabled.
if (clusterControllerLimit.isEmpty() && contentNodeLimit.isEmpty()) {
setter.accept(resourceLimit);
}
}
private void deriveClusterControllerLimit(Optional clusterControllerLimit,
Optional contentNodeLimit,
Consumer setter) {
if (clusterControllerLimit.isEmpty()) {
contentNodeLimit.ifPresent(limit ->
// TODO: emit warning when feed block in distributor is default enabled.
setter.accept(Double.max(0.0, (limit - 0.01))));
}
}
private void deriveContentNodeLimit(Optional contentNodeLimit,
Optional clusterControllerLimit,
double scaleFactor,
Consumer setter) {
if (contentNodeLimit.isEmpty()) {
clusterControllerLimit.ifPresent(limit ->
setter.accept(calcContentNodeLimit(limit, scaleFactor)));
}
}
private double calcContentNodeLimit(double clusterControllerLimit, double scaleFactor) {
return clusterControllerLimit + ((1.0 - clusterControllerLimit) * scaleFactor);
}
private void verifyLimits(double resourceLimitDisk, double resourceLimitMemory) {
verifyLimitInRange(resourceLimitDisk, "disk");
verifyLimitInRange(resourceLimitMemory, "memory");
}
private void verifyLimitInRange(double limit, String type) {
String message = "Resource limit for " + type + " is set to illegal value " + limit +
", but must be in the range [0.0, 1.0]";
if (limit < 0.0)
throw new IllegalArgumentException(message);
if (limit > 1.0)
throw new IllegalArgumentException(message);
}
}
}