1 files changed, 76 insertions, 0 deletions
diff --git a/config-model/src/main/java/com/yahoo/schema/processing/AddDataTypeAndTransformToSummaryOfImportedFields.java b/config-model/src/main/java/com/yahoo/schema/processing/AddDataTypeAndTransformToSummaryOfImportedFields.java
new file mode 100644
index 00000000000..762279e3871
--- /dev/null
+++ b/config-model/src/main/java/com/yahoo/schema/processing/AddDataTypeAndTransformToSummaryOfImportedFields.java
@@ -0,0 +1,76 @@
+// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+package com.yahoo.schema.processing;
+
+import com.yahoo.config.application.api.DeployLogger;
+import com.yahoo.document.DataType;
+import com.yahoo.document.PositionDataType;
+import com.yahoo.schema.RankProfileRegistry;
+import com.yahoo.schema.Schema;
+import com.yahoo.schema.document.ImmutableImportedComplexSDField;
+import com.yahoo.schema.document.ImmutableSDField;
+import com.yahoo.vespa.documentmodel.SummaryField;
+import com.yahoo.vespa.documentmodel.SummaryTransform;
+import com.yahoo.vespa.model.container.search.QueryProfiles;
+
+import java.util.stream.Stream;
+
+/**
+ * Adds the data type and attribute summary transform ({@link SummaryTransform#ATTRIBUTE} to all {@link SummaryField} having an imported
+ * field as source.
+ *
+ * @author bjorncs
+ */
+public class AddDataTypeAndTransformToSummaryOfImportedFields extends Processor {
+
+    public AddDataTypeAndTransformToSummaryOfImportedFields(Schema schema,
+                                                            DeployLogger deployLogger,
+                                                            RankProfileRegistry rankProfileRegistry,
+                                                            QueryProfiles queryProfiles) {
+        super(schema, deployLogger, rankProfileRegistry, queryProfiles);
+    }
+
+    @Override
+    public void process(boolean validate, boolean documentsOnly) {
+        schema.allImportedFields()
+              .forEach(field -> setTransform(field));
+    }
+
+    private Stream<SummaryField> getSummaryFieldsForImportedField(ImmutableSDField importedField) {
+        return schema.getSummaryFields(importedField).stream();
+    }
+
+    private void setTransform(ImmutableSDField field) {
+        if (field instanceof ImmutableImportedComplexSDField) {
+            getSummaryFieldsForImportedField(field).forEach(summaryField -> setAttributeCombinerTransform(field, summaryField));
+        } else {
+            getSummaryFieldsForImportedField(field).forEach(summaryField -> setAttributeTransform(field, summaryField));
+        }
+    }
+
+    private static void setAttributeTransform(ImmutableSDField field, SummaryField summaryField) {
+        if (summaryField.hasUnresolvedType()) {
+            if (field.getDataType().equals(DataType.LONG) && summaryField.getTransform().equals(SummaryTransform.GEOPOS)) {
+                summaryField.setResolvedDataType(PositionDataType.INSTANCE);
+            } else {
+                summaryField.setResolvedDataType(field.getDataType());
+            }
+        }
+        if (summaryField.getTransform() == SummaryTransform.NONE) {
+            summaryField.setTransform(SummaryTransform.ATTRIBUTE);
+        }
+    }
+
+    private static void setAttributeCombinerTransform(ImmutableSDField field, SummaryField summaryField) {
+        if (summaryField.hasUnresolvedType()) {
+            summaryField.setResolvedDataType(field.getDataType());
+        }
+        if (summaryField.getTransform() == SummaryTransform.MATCHED_ATTRIBUTE_ELEMENTS_FILTER) {
+            // This field already has the correct transform.
+            return;
+        } else if (summaryField.getTransform() == SummaryTransform.MATCHED_ELEMENTS_FILTER) {
+            summaryField.setTransform(SummaryTransform.MATCHED_ATTRIBUTE_ELEMENTS_FILTER);
+        } else {
+            summaryField.setTransform(SummaryTransform.ATTRIBUTECOMBINER);
+        }
+    }
+}