Publish

1 files changed, 49 insertions, 0 deletions

diff --git a/container-search/src/main/java/com/yahoo/search/query/textserialize/item/ItemContext.java b/container-search/src/main/java/com/yahoo/search/query/textserialize/item/ItemContext.java
new file mode 100644
index 00000000000..fd21b4e02e1
--- /dev/null
+++ b/container-search/src/main/java/com/yahoo/search/query/textserialize/item/ItemContext.java
@@ -0,0 +1,49 @@
+// Copyright 2016 Yahoo Inc. Licensed under the terms of the Apache 2.0 license. See LICENSE in the project root.
+package com.yahoo.search.query.textserialize.item;
+
+import com.yahoo.prelude.query.Item;
+import com.yahoo.prelude.query.TaggableItem;
+
+import java.util.HashMap;
+import java.util.IdentityHashMap;
+import java.util.Map;
+
+/**
+ * @author tonytv
+ */
+public class ItemContext {
+    private class Connectivity {
+        final String id;
+        final double strength;
+
+        public Connectivity(String id, double strength) {
+            this.id = id;
+            this.strength = strength;
+        }
+    }
+
+    private final Map<String, Item> itemById = new HashMap<>();
+    private final Map<TaggableItem, Connectivity> connectivityByItem = new IdentityHashMap<>();
+
+
+    public void setItemId(String id, Item item) {
+        itemById.put(id, item);
+    }
+
+    public void setConnectivity(TaggableItem item, String id, Double strength) {
+        connectivityByItem.put(item, new Connectivity(id, strength));
+    }
+
+    public void connectItems() {
+        for (Map.Entry<TaggableItem, Connectivity> entry : connectivityByItem.entrySet()) {
+            entry.getKey().setConnectivity(getItem(entry.getValue().id), entry.getValue().strength);
+        }
+    }
+
+    private Item getItem(String id) {
+        Item item = itemById.get(id);
+        if (item == null)
+            throw new IllegalArgumentException("No item with id '" + id + "'.");
+        return item;
+    }
+}