Use architecture in more places in node repo

1 files changed, 6 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 113b32fc571..8fcdebc5d97 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
@@ -199,6 +199,12 @@ public class NodeResources {
         return new NodeResources(vcpu, memoryGb, diskGb, bandwidthGbps, diskSpeed, storageType, architecture);
     }
 
+    public NodeResources with(Architecture architecture) {
+        ensureSpecified();
+        if (architecture == this.architecture) return this;
+        return new NodeResources(vcpu, memoryGb, diskGb, bandwidthGbps, diskSpeed, storageType, architecture);
+    }
+
     /** Returns this with disk speed and storage type set to any */
     public NodeResources justNumbers() {
         if (isUnspecified()) return unspecified();