1 files changed, 120 insertions, 0 deletions

diff --git a/subprojects/viatra-runtime-localsearch/src/main/java/tools/refinery/viatra/runtime/localsearch/matcher/ISearchContext.java b/subprojects/viatra-runtime-localsearch/src/main/java/tools/refinery/viatra/runtime/localsearch/matcher/ISearchContext.java
new file mode 100644
index 00000000..71aa4aac
--- /dev/null
+++ b/subprojects/viatra-runtime-localsearch/src/main/java/tools/refinery/viatra/runtime/localsearch/matcher/ISearchContext.java
@@ -0,0 +1,120 @@
+/*******************************************************************************
+ * Copyright (c) 2010-2014, Zoltan Ujhelyi, Istvan Rath and Daniel Varro
+ * This program and the accompanying materials are made available under the
+ * terms of the Eclipse Public License v. 2.0 which is available at
+ * http://www.eclipse.org/legal/epl-v20.html.
+ *
+ * SPDX-License-Identifier: EPL-2.0
+ *******************************************************************************/
+package tools.refinery.viatra.runtime.localsearch.matcher;
+
+import org.apache.log4j.Logger;
+import tools.refinery.viatra.runtime.localsearch.matcher.integration.IAdornmentProvider;
+import tools.refinery.viatra.runtime.matchers.ViatraQueryRuntimeException;
+import tools.refinery.viatra.runtime.matchers.backend.IQueryResultProvider;
+import tools.refinery.viatra.runtime.matchers.backend.ResultProviderRequestor;
+import tools.refinery.viatra.runtime.matchers.context.IQueryBackendContext;
+import tools.refinery.viatra.runtime.matchers.context.IQueryRuntimeContext;
+import tools.refinery.viatra.runtime.matchers.util.ICache;
+import tools.refinery.viatra.runtime.matchers.util.IProvider;
+
+import java.util.Collections;
+
+/**
+ * The {@link ISearchContext} interface allows search operations to reuse platform services such as the indexer.
+ *
+ * @author Zoltan Ujhelyi
+ * @noreference This interface is not intended to be referenced by clients.
+ * @noimplement This interface is not intended to be implemented by clients.
+ * @noextend This interface is not intended to be extended by clients.
+ *
+ */
+public interface ISearchContext {
+
+    /**
+     * Provides access to the generic query runtime context of the current engine
+     * @since 1.7
+     */
+    IQueryRuntimeContext getRuntimeContext();
+
+    /**
+     * Returns a matcher for a selected query specification.
+     *
+     * @throws ViatraQueryRuntimeException
+     * @since 1.5
+     */
+    IQueryResultProvider getMatcher(CallWithAdornment dependency);
+
+    /**
+     * Allows search operations to cache values through the entire lifecycle of the local search backend. The values are
+     * calculated if not cached before using the given provider, or returned from the cache accordingly.
+     *
+     * @since 1.7
+     */
+    <T> T accessBackendLevelCache(Object key, Class<? extends T> clazz, IProvider<T> valueProvider);
+
+    /**
+     * Returns the engine-specific logger
+     *
+     * @since 2.0
+     */
+    Logger getLogger();
+
+    /**
+     * @noreference This class is not intended to be referenced by clients.
+     * @noimplement This interface is not intended to be implemented by clients.
+     * @noextend This interface is not intended to be extended by clients.
+     */
+    public class SearchContext implements ISearchContext {
+
+        private final IQueryRuntimeContext runtimeContext;
+
+        private final ICache backendLevelCache;
+        private final Logger logger;
+        private final ResultProviderRequestor resultProviderRequestor;
+
+        /**
+         * Initializes a search context using an arbitrary backend context
+         */
+        public SearchContext(IQueryBackendContext backendContext, ICache backendLevelCache,
+                ResultProviderRequestor resultProviderRequestor) {
+            this.resultProviderRequestor = resultProviderRequestor;
+            this.runtimeContext = backendContext.getRuntimeContext();
+            this.logger = backendContext.getLogger();
+
+            this.backendLevelCache = backendLevelCache;
+        }
+
+        /**
+         * @throws ViatraQueryRuntimeException
+         * @since 2.1
+         */
+        @Override
+        public IQueryResultProvider getMatcher(CallWithAdornment dependency) {
+            // Inject adornment for referenced pattern
+            IAdornmentProvider adornmentProvider = query -> {
+                if (query.equals(dependency.getReferredQuery())){
+                    return Collections.singleton(dependency.getAdornment());
+                }
+                return Collections.emptySet();
+            };
+            return resultProviderRequestor.requestResultProvider(dependency.getCall(),
+                    IAdornmentProvider.toHint(adornmentProvider));
+        }
+
+        @Override
+        public <T> T accessBackendLevelCache(Object key, Class<? extends T> clazz, IProvider<T> valueProvider) {
+            return backendLevelCache.getValue(key, clazz, valueProvider);
+        }
+
+        public IQueryRuntimeContext getRuntimeContext() {
+            return runtimeContext;
+        }
+
+        @Override
+        public Logger getLogger() {
+            return logger;
+        }
+
+    }
+}