Publish

1 files changed, 37 insertions, 0 deletions

diff --git a/container-search/src/main/java/com/yahoo/search/query/profile/types/QueryProfileTypeRegistry.java b/container-search/src/main/java/com/yahoo/search/query/profile/types/QueryProfileTypeRegistry.java
new file mode 100644
index 00000000000..3f64caa7ab1
--- /dev/null
+++ b/container-search/src/main/java/com/yahoo/search/query/profile/types/QueryProfileTypeRegistry.java
@@ -0,0 +1,37 @@
+// 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.profile.types;
+
+import com.yahoo.component.provider.ComponentRegistry;
+import com.yahoo.search.Query;
+import com.yahoo.search.query.profile.QueryProfileRegistry;
+
+/**
+ * A registry of query profile types
+ *
+ * @author bratseth
+ */
+public class QueryProfileTypeRegistry extends ComponentRegistry<QueryProfileType> {
+
+    public QueryProfileTypeRegistry() {
+        Query.addNativeQueryProfileTypesTo(this);
+    }
+
+    /** Register this type by its id */
+    public void register(QueryProfileType type) {
+        super.register(type.getId(), type);
+    }
+
+    @Override
+    public void freeze() {
+        if (isFrozen()) return;
+        for (QueryProfileType queryProfileType : allComponents())
+            queryProfileType.freeze();
+    }
+
+    public static QueryProfileTypeRegistry emptyFrozen() {
+        QueryProfileTypeRegistry registry = new QueryProfileTypeRegistry();
+        registry.freeze();
+        return registry;
+    }
+
+}