wip

apache · Jul 17, 2023 · fedfb3d · fedfb3d
1 parent 26947f0
commit fedfb3d
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/...ls-kiesession/src/main/java/org/drools/kiesession/consequence/DefaultKnowledgeHelper.java b/...ls-kiesession/src/main/java/org/drools/kiesession/consequence/DefaultKnowledgeHelper.java
@@ -70,6 +70,8 @@ public class DefaultKnowledgeHelper implements KnowledgeHelper, Externalizable {
     protected ReteEvaluator reteEvaluator;
     private StatefulKnowledgeSessionForRHS wrappedEvaluator;
 
+    private KnowledgeHelper tmsKnowledgeHelper;
+
     public DefaultKnowledgeHelper() { }
 
     public DefaultKnowledgeHelper(ReteEvaluator reteEvaluator) {
@@ -161,10 +163,13 @@ private KnowledgeHelper executeOnTMS() {
         if (!TruthMaintenanceSystemFactory.present()) {
             TruthMaintenanceSystemFactory.throwExceptionForMissingTms();
         }
+        if (tmsKnowledgeHelper != null) {
+            return tmsKnowledgeHelper;
+        }
         reteEvaluator.enableTMS();
-        KnowledgeHelper knowledgeHelper = reteEvaluator.createKnowledgeHelper();
-        knowledgeHelper.setActivation(internalMatch);
-        return knowledgeHelper;
+        tmsKnowledgeHelper = reteEvaluator.createKnowledgeHelper();
+        tmsKnowledgeHelper.setActivation(internalMatch);
+        return tmsKnowledgeHelper;
     }
 
     public void cancelMatch(Match act) {