diff options
Diffstat (limited to 'config-provisioning/src/main/java/com/yahoo/config/provision/NodeResources.java')
-rw-r--r-- | config-provisioning/src/main/java/com/yahoo/config/provision/NodeResources.java | 11 |
1 files changed, 11 insertions, 0 deletions
diff --git a/config-provisioning/src/main/java/com/yahoo/config/provision/NodeResources.java b/config-provisioning/src/main/java/com/yahoo/config/provision/NodeResources.java index 0548bc7520f..4454517d5ec 100644 --- a/config-provisioning/src/main/java/com/yahoo/config/provision/NodeResources.java +++ b/config-provisioning/src/main/java/com/yahoo/config/provision/NodeResources.java @@ -280,6 +280,17 @@ public class NodeResources { public boolean isUnspecified() { return this.equals(unspecified); } + // Returns squared euclidean distance of the relevant numerical values of two node resources + public double distanceTo(NodeResources other) { + if ( ! this.diskSpeed().compatibleWith(other.diskSpeed())) return Double.MAX_VALUE; + if ( ! this.storageType().compatibleWith(other.storageType())) return Double.MAX_VALUE; + + double distance = Math.pow(this.vcpu() - other.vcpu(), 2) + Math.pow(this.memoryGb() - other.memoryGb(), 2); + if (this.storageType() == StorageType.local || other.storageType() == StorageType.local) + distance += Math.pow(this.diskGb() - other.diskGb(), 2); + return distance; + } + /** Returns this.isUnspecified() ? Optional.empty() : Optional.of(this) */ public Optional<NodeResources> asOptional() { return this.isUnspecified() ? Optional.empty() : Optional.of(this); |