1 files changed, 129 insertions, 0 deletions
diff --git a/config-model/src/main/java/com/yahoo/searchdefinition/derived/SchemaInfo.java b/config-model/src/main/java/com/yahoo/searchdefinition/derived/SchemaInfo.java
new file mode 100644
index 00000000000..3698345eb11
--- /dev/null
+++ b/config-model/src/main/java/com/yahoo/searchdefinition/derived/SchemaInfo.java
@@ -0,0 +1,129 @@
+// Copyright Yahoo. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+package com.yahoo.searchdefinition.derived;
+
+import com.yahoo.search.config.SchemaInfoConfig;
+import com.yahoo.searchdefinition.RankProfile;
+import com.yahoo.searchdefinition.RankProfileRegistry;
+import com.yahoo.searchdefinition.Schema;
+import com.yahoo.searchlib.rankingexpression.Reference;
+import com.yahoo.tensor.TensorType;
+import com.yahoo.vespa.config.search.SummarymapConfig;
+import com.yahoo.vespa.documentmodel.SummaryTransform;
+import com.yahoo.vespa.model.search.SearchCluster;
+
+import java.util.Collection;
+import java.util.Collections;
+import java.util.LinkedHashMap;
+import java.util.Map;
+
+/**
+ * Information about a schema.
+ *
+ * @author bratseth
+ */
+public final class SchemaInfo implements SchemaInfoConfig.Producer {
+
+    private final Schema schema;
+
+    // Info about profiles needed in memory after build.
+    // The rank profile registry itself is not kept around due to its size.
+    private final Map<String, RankProfileInfo> rankProfiles;
+
+    private final Summaries summaries;
+    private final SummaryMap summaryMap;
+
+    public SchemaInfo(Schema schema, RankProfileRegistry rankProfileRegistry,
+                      Summaries summaries, SummaryMap summaryMap) {
+        this.schema = schema;
+        this.rankProfiles = Collections.unmodifiableMap(toRankProfiles(rankProfileRegistry.rankProfilesOf(schema)));
+        this.summaries = summaries;
+        this.summaryMap = summaryMap;
+    }
+
+    public String name() { return schema.getName(); }
+
+    public Schema fullSchema() { return schema; }
+
+    public Map<String, RankProfileInfo> rankProfiles() { return rankProfiles; }
+
+    private Map<String, RankProfileInfo> toRankProfiles(Collection<RankProfile> rankProfiles) {
+        Map<String, RankProfileInfo> rankProfileInfos = new LinkedHashMap<>();
+        rankProfiles.forEach(profile -> rankProfileInfos.put(profile.name(), new RankProfileInfo(profile)));
+        return rankProfileInfos;
+    }
+
+    @Override
+    public void getConfig(SchemaInfoConfig.Builder builder) {
+        var schemaBuilder = new SchemaInfoConfig.Schema.Builder();
+        schemaBuilder.name(schema.getName());
+        addSummaryConfig(schemaBuilder);
+        addRankProfilesConfig(schemaBuilder);
+        builder.schema(schemaBuilder);
+    }
+
+    private void addSummaryConfig(SchemaInfoConfig.Schema.Builder schemaBuilder) {
+        for (var summary : summaries.asList()) {
+            var summaryBuilder = new SchemaInfoConfig.Schema.Summaryclass.Builder();
+            summaryBuilder.name(summary.getName());
+            for (var field : summary.fields().values()) {
+                var fieldsBuilder = new SchemaInfoConfig.Schema.Summaryclass.Fields.Builder();
+                fieldsBuilder.name(field.getName())
+                             .type(field.getType().getName())
+                             .dynamic(isDynamic(field.getName()));
+                summaryBuilder.fields(fieldsBuilder);
+            }
+            schemaBuilder.summaryclass(summaryBuilder);
+        }
+    }
+
+    /** Returns whether the given field is a dynamic summary field. */
+    private boolean isDynamic(String fieldName) {
+        if (summaryMap == null) return false; // not know for streaming, but also not used
+
+        var fieldTransform = summaryMap.resultTransforms().get(fieldName);
+        if (fieldTransform == null) return false;
+        // TODO: Move this into SummaryTransform and call it something else than "dynamic"
+        return fieldTransform.getTransform().isDynamic() ||
+               fieldTransform.getTransform() == SummaryTransform.MATCHED_ELEMENTS_FILTER ||
+               fieldTransform.getTransform() == SummaryTransform.MATCHED_ATTRIBUTE_ELEMENTS_FILTER;
+    }
+
+    private void addRankProfilesConfig(SchemaInfoConfig.Schema.Builder schemaBuilder) {
+        for (RankProfileInfo rankProfile : rankProfiles().values()) {
+            var rankProfileConfig = new SchemaInfoConfig.Schema.Rankprofile.Builder();
+            rankProfileConfig.name(rankProfile.name());
+            rankProfileConfig.hasSummaryFeatures(rankProfile.hasSummaryFeatures());
+            rankProfileConfig.hasRankFeatures(rankProfile.hasRankFeatures());
+            for (var input : rankProfile.inputs().entrySet()) {
+                var inputConfig = new SchemaInfoConfig.Schema.Rankprofile.Input.Builder();
+                inputConfig.name(input.getKey().toString());
+                inputConfig.type(input.getValue().toString());
+                rankProfileConfig.input(inputConfig);
+            }
+            schemaBuilder.rankprofile(rankProfileConfig);
+        }
+    }
+
+    /** A store of a *small* (in memory) amount of rank profile info. */
+    public static final class RankProfileInfo {
+
+        private final String name;
+        private final boolean hasSummaryFeatures;
+        private final boolean hasRankFeatures;
+        private final Map<Reference, TensorType> inputs;
+
+        public RankProfileInfo(RankProfile profile) {
+            this.name = profile.name();
+            this.hasSummaryFeatures =  ! profile.getSummaryFeatures().isEmpty();
+            this.hasRankFeatures =  ! profile.getRankFeatures().isEmpty();
+            this.inputs = profile.inputs();
+        }
+
+        public String name() { return name; }
+        public boolean hasSummaryFeatures() { return hasSummaryFeatures; }
+        public boolean hasRankFeatures() { return hasRankFeatures; }
+        public Map<Reference, TensorType> inputs() { return inputs; }
+
+    }
+
+}