1 files changed, 6 insertions, 0 deletions
diff --git a/controller-server/src/main/java/com/yahoo/vespa/hosted/controller/application/EndpointList.java b/controller-server/src/main/java/com/yahoo/vespa/hosted/controller/application/EndpointList.java
index 4da34dad737..e847667bf45 100644
--- a/controller-server/src/main/java/com/yahoo/vespa/hosted/controller/application/EndpointList.java
+++ b/controller-server/src/main/java/com/yahoo/vespa/hosted/controller/application/EndpointList.java
@@ -2,6 +2,7 @@
 package com.yahoo.vespa.hosted.controller.application;
 
 import com.yahoo.collections.AbstractFilteringList;
+import com.yahoo.config.provision.ClusterSpec;
 import com.yahoo.config.provision.zone.ZoneId;
 
 import java.util.Collection;
@@ -33,6 +34,11 @@ public class EndpointList extends AbstractFilteringList<Endpoint, EndpointList>
         return matching(endpoint -> endpoint.name().equals(id.id()));
     }
 
+    /** Returns the subset of endpoints pointing to given cluster */
+    public EndpointList cluster(ClusterSpec.Id cluster) {
+        return matching(endpoint -> endpoint.cluster().equals(cluster));
+    }
+
     /** Returns the subset of endpoints which target all of the given zones */
     public EndpointList targets(List<ZoneId> zones) {
         return matching(endpoint -> endpoint.zones().containsAll(zones));