diff options
author | gjoranv <gv@yahoo-inc.com> | 2017-02-28 01:05:19 +0100 |
---|---|---|
committer | gjoranv <gv@yahoo-inc.com> | 2017-02-28 01:05:19 +0100 |
commit | 83bc3e7dd8f9d9bbd0d5e00402b6f28679fe9f70 (patch) | |
tree | 7fb15c8433fee73a88cf65edbd77ae56150909f8 /config-model/src | |
parent | c5920e1d51319b31430ff62b51606304e32a1d7e (diff) |
Return ClusterMembership instead of ClusterSpec.
Diffstat (limited to 'config-model/src')
-rw-r--r-- | config-model/src/main/java/com/yahoo/vespa/model/HostResource.java | 7 | ||||
-rw-r--r-- | config-model/src/test/java/com/yahoo/vespa/model/HostResourceTest.java | 2 |
2 files changed, 4 insertions, 5 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 a1cef1af863..d54dc912da6 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 @@ -246,16 +246,15 @@ public class HostResource implements Comparable<HostResource> { * Content clusters are preferred, then container clusters, and finally admin clusters. * If there is more than one cluster of the preferred type, the cluster that was added first will be chosen. */ - public Optional<ClusterSpec> primaryClusterMembership() { + public Optional<ClusterMembership> primaryClusterMembership() { return clusterMemberships().stream() - .map(ClusterMembership::cluster) .sorted(HostResource::compareClusters) .findFirst(); } - private static int compareClusters(ClusterSpec cluster1, ClusterSpec cluster2) { + private static int compareClusters(ClusterMembership cluster1, ClusterMembership cluster2) { // This depends on the declared order of enum constants. - return cluster2.type().compareTo(cluster1.type()); + return cluster2.cluster().type().compareTo(cluster1.cluster().type()); } @Override diff --git a/config-model/src/test/java/com/yahoo/vespa/model/HostResourceTest.java b/config-model/src/test/java/com/yahoo/vespa/model/HostResourceTest.java index ecfbf8cd96c..16c63bc5251 100644 --- a/config-model/src/test/java/com/yahoo/vespa/model/HostResourceTest.java +++ b/config-model/src/test/java/com/yahoo/vespa/model/HostResourceTest.java @@ -135,7 +135,7 @@ public class HostResourceTest { } private void assertClusterMembership(HostResource host, ClusterSpec.Type type, String id) { - ClusterSpec membership = host.primaryClusterMembership() + ClusterSpec membership = host.primaryClusterMembership().map(ClusterMembership::cluster) .orElseThrow(() -> new RuntimeException("No cluster membership!")); assertEquals(type, membership.type()); |