1 files changed, 34 insertions, 0 deletions

diff --git a/subprojects/store-query-viatra/src/main/java/org/eclipse/viatra/query/runtime/rete/network/RefineryConnectionFactory.java b/subprojects/store-query-viatra/src/main/java/org/eclipse/viatra/query/runtime/rete/network/RefineryConnectionFactory.java
new file mode 100644
index 00000000..241c5b58
--- /dev/null
+++ b/subprojects/store-query-viatra/src/main/java/org/eclipse/viatra/query/runtime/rete/network/RefineryConnectionFactory.java
@@ -0,0 +1,34 @@
+/*
+ * SPDX-FileCopyrightText: 2023 The Refinery Authors <https://refinery.tools/>
+ *
+ * SPDX-License-Identifier: EPL-2.0
+ */
+package org.eclipse.viatra.query.runtime.rete.network;
+
+import org.eclipse.viatra.query.runtime.rete.traceability.RecipeTraceInfo;
+import tools.refinery.store.query.viatra.internal.rete.network.RefineryConnectionFactoryExtensions;
+
+/**
+ * This class overrides some RETE connection types from {@link ConnectionFactory}.
+ * <p>
+ * Since {@link ConnectionFactory} is package-private, this class has to be in the
+ * {@code org.eclipse.viatra.query.runtime.rete.network} package as well.
+ * However, due to JAR signature verification errors, <b>this class cannot be loaded directly</b>
+ * and has to be loaded at runtime as a byte array instead.
+ */
+@SuppressWarnings("unused")
+public class RefineryConnectionFactory extends ConnectionFactory {
+        private final RefineryConnectionFactoryExtensions extensions;
+
+        public RefineryConnectionFactory(ReteContainer reteContainer) {
+                super(reteContainer);
+                extensions = new RefineryConnectionFactoryExtensions(reteContainer);
+        }
+
+        @Override
+        public void connectToParents(RecipeTraceInfo recipeTrace, Node freshNode) {
+                if (!extensions.connectToParents(recipeTrace, freshNode)) {
+                        super.connectToParents(recipeTrace, freshNode);
+                }
+        }
+}