1 files changed, 100 insertions, 0 deletions
diff --git a/config-model/src/main/java/com/yahoo/vespa/model/search/StreamingSearchCluster.java b/config-model/src/main/java/com/yahoo/vespa/model/search/StreamingSearchCluster.java
new file mode 100644
index 00000000000..e188a086614
--- /dev/null
+++ b/config-model/src/main/java/com/yahoo/vespa/model/search/StreamingSearchCluster.java
@@ -0,0 +1,100 @@
+// Copyright Vespa.ai. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+package com.yahoo.vespa.model.search;
+
+import com.yahoo.config.model.deploy.DeployState;
+import com.yahoo.config.model.producer.AnyConfigProducer;
+import com.yahoo.config.model.producer.TreeConfigProducer;
+import com.yahoo.schema.Schema;
+import com.yahoo.schema.derived.AttributeFields;
+import com.yahoo.schema.derived.DerivedConfiguration;
+import com.yahoo.schema.derived.SchemaInfo;
+import com.yahoo.vespa.config.search.AttributesConfig;
+import com.yahoo.vespa.config.search.RankProfilesConfig;
+import com.yahoo.vespa.config.search.SummaryConfig;
+import com.yahoo.vespa.config.search.core.OnnxModelsConfig;
+import com.yahoo.vespa.config.search.core.ProtonConfig;
+import com.yahoo.vespa.config.search.core.RankingConstantsConfig;
+import com.yahoo.vespa.config.search.core.RankingExpressionsConfig;
+import com.yahoo.vespa.config.search.summary.JuniperrcConfig;
+import com.yahoo.vespa.config.search.vsm.VsmfieldsConfig;
+import com.yahoo.vespa.config.search.vsm.VsmsummaryConfig;
+
+/**
+ * A search cluster of type streaming.
+ * 
+ * @author baldersheim
+ * @author vegardh
+ */
+public class StreamingSearchCluster extends SearchCluster implements
+        AttributesConfig.Producer,
+        RankProfilesConfig.Producer,
+        RankingConstantsConfig.Producer,
+        RankingExpressionsConfig.Producer,
+        OnnxModelsConfig.Producer,
+        JuniperrcConfig.Producer,
+        SummaryConfig.Producer,
+        VsmsummaryConfig.Producer,
+        VsmfieldsConfig.Producer
+{
+    private final String storageRouteSpec;
+    private final AttributesProducer attributesConfig;
+    private final String docTypeName;
+
+    public StreamingSearchCluster(TreeConfigProducer<AnyConfigProducer> parent, String clusterName, int index,
+                                  String docTypeName, String storageRouteSpec) {
+        super(parent, clusterName, index);
+        attributesConfig = new AttributesProducer(parent, docTypeName);
+        this.docTypeName = docTypeName;
+        this.storageRouteSpec = storageRouteSpec;
+    }
+
+    @Override
+    protected IndexingMode getIndexingMode() { return IndexingMode.STREAMING; }
+    public final String getStorageRouteSpec() { return storageRouteSpec; }
+
+    public String getDocTypeName() { return docTypeName; }
+
+    public DerivedConfiguration derived() { return db().getDerivedConfiguration(); }
+
+    @Override
+    public void deriveFromSchemas(DeployState deployState) {
+        if (schemas().isEmpty()) return;
+        if (schemas().size() > 1) throw new IllegalArgumentException("Only a single schema is supported, got " + schemas().size());
+
+        Schema schema = schemas().values().stream().findAny().get().fullSchema();
+        if ( ! schema.getName().equals(docTypeName))
+            throw new IllegalArgumentException("Document type name '" + docTypeName +
+                                               "' must be the same as the schema name '" + schema.getName() + "'");
+        add(new DocumentDatabase(this, docTypeName, new DerivedConfiguration(deployState, schema, SchemaInfo.IndexMode.STREAMING)));
+    }
+
+    protected void fillDocumentDBConfig(DocumentDatabase sdoc, ProtonConfig.Documentdb.Builder ddbB) {
+        super.fillDocumentDBConfig(sdoc, ddbB);
+        ddbB.configid(attributesConfig.getConfigId()); // Temporary until fully cleaned up
+    }
+
+    private DocumentDatabase db() { return getDocumentDbs().get(0); }
+
+    // These are temporary until backend uses correct config id.
+    @Override public void getConfig(SummaryConfig.Builder builder) { db().getConfig(builder); }
+    @Override public void getConfig(OnnxModelsConfig.Builder builder) { db().getConfig(builder); }
+    @Override public void getConfig(RankingConstantsConfig.Builder builder) { db().getConfig(builder); }
+    @Override public void getConfig(RankProfilesConfig.Builder builder) { db().getConfig(builder); }
+    @Override public void getConfig(RankingExpressionsConfig.Builder builder) { db().getConfig(builder); }
+    @Override public void getConfig(JuniperrcConfig.Builder builder) { db().getConfig(builder); }
+    @Override public void getConfig(VsmfieldsConfig.Builder builder) { db().getConfig(builder); }
+    @Override public void getConfig(VsmsummaryConfig.Builder builder) { db().getConfig(builder);}
+
+    private class AttributesProducer extends AnyConfigProducer implements AttributesConfig.Producer {
+
+        AttributesProducer(TreeConfigProducer<AnyConfigProducer> parent, String docType) {
+            super(parent, docType);
+        }
+
+        @Override
+        public void getConfig(AttributesConfig.Builder builder) {
+            derived().getConfig(builder, AttributeFields.FieldSet.FAST_ACCESS);
+        }
+    }
+
+}