Add config to cluster controller on whether it should block feed based on reported resource usage from content nodes.

Each resource category has a specified limit.
This is currently disabled by default.

2 files changed, 32 insertions, 4 deletions

diff --git a/config-model/src/main/java/com/yahoo/vespa/model/content/ClusterControllerConfig.java b/config-model/src/main/java/com/yahoo/vespa/model/content/ClusterControllerConfig.java
index 882e56b82a3..ad5b472387f 100644
--- a/config-model/src/main/java/com/yahoo/vespa/model/content/ClusterControllerConfig.java
+++ b/config-model/src/main/java/com/yahoo/vespa/model/content/ClusterControllerConfig.java
@@ -131,5 +131,19 @@ public class ClusterControllerConfig extends AbstractConfigProducer<ClusterContr
         if (minNodeRatioPerGroup != null) {
             builder.min_node_ratio_per_group(minNodeRatioPerGroup);
         }
+
+        setDefaultClusterFeedBlockLimits(builder);
+    }
+
+    private static void setDefaultClusterFeedBlockLimits(FleetcontrollerConfig.Builder builder) {
+        // TODO: Override these based on resource-limits in services.xml (if they are specified).
+        // TODO: Choose other defaults when this is default enabled.
+        // Note: The resource categories must match the ones used in host info reporting
+        // between content nodes and cluster controller:
+        // storage/src/vespa/storage/persistence/filestorage/service_layer_host_info_reporter.cpp
+        builder.cluster_feed_block_limit.put("memory", 0.79);
+        builder.cluster_feed_block_limit.put("disk", 0.79);
+        builder.cluster_feed_block_limit.put("attribute-enum-store", 0.89);
+        builder.cluster_feed_block_limit.put("attribute-multi-value", 0.89);
     }
 }
diff --git a/config-model/src/test/java/com/yahoo/vespa/model/content/FleetControllerClusterTest.java b/config-model/src/test/java/com/yahoo/vespa/model/content/FleetControllerClusterTest.java
index 41fb4a3d35b..137f5351299 100644
--- a/config-model/src/test/java/com/yahoo/vespa/model/content/FleetControllerClusterTest.java
+++ b/config-model/src/test/java/com/yahoo/vespa/model/content/FleetControllerClusterTest.java
@@ -81,13 +81,27 @@ public class FleetControllerClusterTest {
 
     @Test
     public void min_node_ratio_per_group_is_implicitly_zero_when_omitted() {
-        FleetcontrollerConfig.Builder builder = new FleetcontrollerConfig.Builder();
+        var config = getConfigForBasicCluster();
+        assertEquals(0.0, config.min_node_ratio_per_group(), 0.01);
+    }
+
+    @Test
+    public void default_cluster_feed_block_limits_are_set() {
+        var config = getConfigForBasicCluster();
+        var limits = config.cluster_feed_block_limit();
+        assertEquals(4, limits.size());
+        assertEquals(0.79, limits.get("memory"), 0.0001);
+        assertEquals(0.79, limits.get("disk"), 0.0001);
+        assertEquals(0.89, limits.get("attribute-enum-store"), 0.0001);
+        assertEquals(0.89, limits.get("attribute-multi-value"), 0.0001);
+    }
+
+    FleetcontrollerConfig getConfigForBasicCluster() {
+        var builder = new FleetcontrollerConfig.Builder();
         parse("<cluster id=\"storage\">\n" +
                 "  <documents/>\n" +
                 "</cluster>").
                 getConfig(builder);
-
-        FleetcontrollerConfig config = new FleetcontrollerConfig(builder);
-        assertEquals(0.0, config.min_node_ratio_per_group(), 0.01);
+        return new FleetcontrollerConfig(builder);
     }
 }