Add architeecture to node repo rest api

1 files changed, 19 insertions, 0 deletions

diff --git a/node-repository/src/main/java/com/yahoo/vespa/hosted/provision/restapi/NodeResourcesSerializer.java b/node-repository/src/main/java/com/yahoo/vespa/hosted/provision/restapi/NodeResourcesSerializer.java
index 1d254b3e531..e094bc69735 100644
--- a/node-repository/src/main/java/com/yahoo/vespa/hosted/provision/restapi/NodeResourcesSerializer.java
+++ b/node-repository/src/main/java/com/yahoo/vespa/hosted/provision/restapi/NodeResourcesSerializer.java
@@ -16,6 +16,7 @@ public class NodeResourcesSerializer {
         object.setDouble("bandwidthGbps", resources.bandwidthGbps());
         object.setString("diskSpeed", toString(resources.diskSpeed()));
         object.setString("storageType", toString(resources.storageType()));
+        object.setString("architecture", toString(resources.architecture()));
     }
 
     public static NodeResources.DiskSpeed diskSpeedFrom(String diskSpeed) {
@@ -54,4 +55,22 @@ public class NodeResourcesSerializer {
         }
     }
 
+    private static String toString(NodeResources.Architecture architecture) {
+        switch (architecture) {
+            case arm64 : return "arm64";
+            case x86_64: return "x86_64";
+            case any   : return "any";
+            default: throw new IllegalArgumentException("Unknown architecture '" + architecture.name() + "'");
+        }
+    }
+
+    public static NodeResources.Architecture architectureFrom(String architecture) {
+        switch (architecture) {
+            case "arm64" : return NodeResources.Architecture.arm64;
+            case "x86_64": return NodeResources.Architecture.x86_64;
+            case "any"   : return NodeResources.Architecture.any;
+            default: throw new IllegalArgumentException("Unknown architecture '" + architecture + "'");
+        }
+    }
+
 }