Control attribute maxuncommittedmemory with feature flag

2 files changed, 22 insertions, 5 deletions

diff --git a/config-model/src/main/java/com/yahoo/searchdefinition/derived/AttributeFields.java b/config-model/src/main/java/com/yahoo/searchdefinition/derived/AttributeFields.java
index 67ed37800f9..2890f1cc019 100644
--- a/config-model/src/main/java/com/yahoo/searchdefinition/derived/AttributeFields.java
+++ b/config-model/src/main/java/com/yahoo/searchdefinition/derived/AttributeFields.java
@@ -190,7 +190,8 @@ public class AttributeFields extends Derived implements AttributesConfig.Produce
 
     @Override
     public void getConfig(AttributesConfig.Builder builder) {
-        getConfig(builder, FieldSet.ALL);
+        //TODO This is just to get some exporting tests to work, Should be undone and removed
+        getConfig(builder, FieldSet.ALL, 77777);
     }
 
     private boolean isAttributeInFieldSet(Attribute attribute, FieldSet fs) {
@@ -295,15 +296,19 @@ public class AttributeFields extends Derived implements AttributesConfig.Produce
         return AttributesConfig.Attribute.Match.UNCASED;
     }
 
-    public void getConfig(AttributesConfig.Builder builder, FieldSet fs) {
+    public void getConfig(AttributesConfig.Builder builder, FieldSet fs, long maxUnCommittedMemory) {
         for (Attribute attribute : attributes.values()) {
             if (isAttributeInFieldSet(attribute, fs)) {
-                builder.attribute(getConfig(attribute.getName(), attribute, false));
+                AttributesConfig.Attribute.Builder attrBuilder = getConfig(attribute.getName(), attribute, false);
+                attrBuilder.maxuncommittedmemory(maxUnCommittedMemory);
+                builder.attribute(attrBuilder);
             }
         }
         if (fs == FieldSet.ALL) {
             for (Map.Entry<String, Attribute> entry : importedAttributes.entrySet()) {
-                builder.attribute(getConfig(entry.getKey(), entry.getValue(), true));
+                AttributesConfig.Attribute.Builder attrBuilder = getConfig(entry.getKey(), entry.getValue(), true);
+                attrBuilder.maxuncommittedmemory(maxUnCommittedMemory);
+                builder.attribute(attrBuilder);
             }
         }
     }
diff --git a/config-model/src/main/java/com/yahoo/searchdefinition/derived/DerivedConfiguration.java b/config-model/src/main/java/com/yahoo/searchdefinition/derived/DerivedConfiguration.java
index 54e9a4ccb61..6165cdd2dc1 100644
--- a/config-model/src/main/java/com/yahoo/searchdefinition/derived/DerivedConfiguration.java
+++ b/config-model/src/main/java/com/yahoo/searchdefinition/derived/DerivedConfiguration.java
@@ -16,6 +16,7 @@ import com.yahoo.search.query.profile.QueryProfileRegistry;
 import com.yahoo.searchdefinition.RankProfileRegistry;
 import com.yahoo.searchdefinition.Schema;
 import com.yahoo.searchdefinition.derived.validation.Validation;
+import com.yahoo.vespa.config.search.AttributesConfig;
 import com.yahoo.vespa.model.container.search.QueryProfiles;
 
 import java.io.IOException;
@@ -28,7 +29,7 @@ import java.util.concurrent.ExecutorService;
  *
  * @author bratseth
  */
-public class DerivedConfiguration {
+public class DerivedConfiguration implements AttributesConfig.Producer {
 
     private final Schema schema;
     private Summaries summaries;
@@ -43,6 +44,7 @@ public class DerivedConfiguration {
     private IndexSchema indexSchema;
     private ImportedFields importedFields;
     private final QueryProfileRegistry queryProfiles;
+    private final long maxUncommittedMemory;
 
     /**
      * Creates a complete derived configuration from a search definition.
@@ -82,6 +84,7 @@ public class DerivedConfiguration {
         Validator.ensureNotNull("Search definition", schema);
         this.schema = schema;
         this.queryProfiles = queryProfiles;
+        this.maxUncommittedMemory = deployProperties.featureFlags().maxUnCommittedMemory();
         if ( ! schema.isDocumentsOnly()) {
             streamingFields = new VsmFields(schema);
             streamingSummary = new VsmSummary(schema);
@@ -159,6 +162,15 @@ public class DerivedConfiguration {
         return attributeFields;
     }
 
+    @Override
+    public void getConfig(AttributesConfig.Builder builder) {
+        getConfig(builder, AttributeFields.FieldSet.ALL);
+    }
+
+    public void getConfig(AttributesConfig.Builder builder, AttributeFields.FieldSet fs) {
+        attributeFields.getConfig(builder, fs, maxUncommittedMemory);
+    }
+
     public IndexingScript getIndexingScript() {
         return indexingScript;
     }