Use Flavor object instead of String.

1 files changed, 4 insertions, 3 deletions

diff --git a/config-model/src/main/java/com/yahoo/vespa/model/HostResource.java b/config-model/src/main/java/com/yahoo/vespa/model/HostResource.java
index f3e3abeff91..a0ad80c3070 100644
--- a/config-model/src/main/java/com/yahoo/vespa/model/HostResource.java
+++ b/config-model/src/main/java/com/yahoo/vespa/model/HostResource.java
@@ -3,6 +3,7 @@ package com.yahoo.vespa.model;
 
 import com.yahoo.config.model.api.HostInfo;
 import com.yahoo.config.provision.ClusterMembership;
+import com.yahoo.config.provision.Flavor;
 
 import javax.annotation.Nullable;
 import java.util.ArrayList;
@@ -40,7 +41,7 @@ public class HostResource implements Comparable<HostResource> {
     private Set<ClusterMembership> clusterMemberships = new LinkedHashSet<>();
 
     // Empty for self-hosted Vespa.
-    private Optional<String> flavor = Optional.empty();
+    private Optional<Flavor> flavor = Optional.empty();
 
     /**
      * Create a new {@link HostResource} bound to a specific {@link com.yahoo.vespa.model.Host}.
@@ -225,10 +226,10 @@ public class HostResource implements Comparable<HostResource> {
                 .collect(Collectors.toSet()));
     }
 
-    public void setFlavor(Optional<String> flavor) { this.flavor = flavor; }
+    public void setFlavor(Optional<Flavor> flavor) { this.flavor = flavor; }
 
     /** Returns the flavor of this resource. Empty for self-hosted Vespa. */
-    public Optional<String> getFlavor() { return flavor; }
+    public Optional<Flavor> getFlavor() { return flavor; }
 
     public void addClusterMembership(@Nullable ClusterMembership clusterMembership) {
         if (clusterMembership != null)