- move collectRules to CountBasedOrderingStrategy

apache · Jul 7, 2020 · 94f1493 · 94f1493
1 parent 38e80e7
commit 94f1493
Show file tree

Hide file tree

Showing 4 changed files with 22 additions and 16 deletions.
diff --git a/drools-core/src/main/java/org/drools/core/addon/AlphaNodeOrderingStrategy.java b/drools-core/src/main/java/org/drools/core/addon/AlphaNodeOrderingStrategy.java
@@ -15,13 +15,14 @@
 
 package org.drools.core.addon;
 
+import java.util.Collection;
 import java.util.List;
-import java.util.Set;
+import java.util.Map;
 
+import org.drools.core.definitions.InternalKnowledgePackage;
 import org.drools.core.spi.AlphaNodeFieldConstraint;
 import org.drools.core.spi.ObjectType;
 import org.kie.api.conf.AlphaNodeOrderingOption;
-import org.kie.api.definition.rule.Rule;
 
 /**
  * 
@@ -31,7 +32,7 @@
  */
 public interface AlphaNodeOrderingStrategy {
 
-    void analyzeAlphaConstraints(Set<Rule> ruleSet);
+    void analyzeAlphaConstraints(Map<String, InternalKnowledgePackage> pkgs, Collection<InternalKnowledgePackage> newPkgs);
 
     void reorderAlphaConstraints(List<AlphaNodeFieldConstraint> alphaConstraints, ObjectType objectType);
 

diff --git a/drools-core/src/main/java/org/drools/core/addon/CountBasedOrderingStrategy.java b/drools-core/src/main/java/org/drools/core/addon/CountBasedOrderingStrategy.java
@@ -16,12 +16,15 @@
 package org.drools.core.addon;
 
 import java.util.ArrayList;
+import java.util.Collection;
 import java.util.HashMap;
+import java.util.HashSet;
 import java.util.List;
 import java.util.Map;
 import java.util.Objects;
 import java.util.Set;
 
+import org.drools.core.definitions.InternalKnowledgePackage;
 import org.drools.core.definitions.rule.impl.RuleImpl;
 import org.drools.core.rule.Accumulate;
 import org.drools.core.rule.Collect;
@@ -48,7 +51,9 @@ public class CountBasedOrderingStrategy implements AlphaNodeOrderingStrategy {
     private Map<ObjectType, Map<AlphaNodeFieldConstraint, Integer>> analyzedAlphaConstraints = new HashMap<>();
 
     @Override
-    public void analyzeAlphaConstraints(Set<Rule> ruleSet) {
+    public void analyzeAlphaConstraints(Map<String, InternalKnowledgePackage> pkgs, Collection<InternalKnowledgePackage> newPkgs) {
+
+        Set<Rule> ruleSet = collectRules(pkgs, newPkgs);
 
         List<Pattern> patternList = new ArrayList<>();
         ruleSet.stream()
@@ -68,6 +73,13 @@ public void analyzeAlphaConstraints(Set<Rule> ruleSet) {
         logger.trace("analyzedAlphaConstraints : {}", analyzedAlphaConstraints);
     }
 
+    private Set<Rule> collectRules(Map<String, InternalKnowledgePackage> pkgs, Collection<InternalKnowledgePackage> newPkgs) {
+        Set<Rule> ruleSet = new HashSet<>();
+        pkgs.forEach((pkgName, pkg) -> ruleSet.addAll(pkg.getRules()));
+        newPkgs.forEach(pkg -> ruleSet.addAll(pkg.getRules())); // okay to overwrite
+        return ruleSet;
+    }
+
     private void collectPatterns(GroupElement ge, List<Pattern> patternList) {
         List<RuleConditionElement> children = ge.getChildren();
         for (RuleConditionElement child : children) {

diff --git a/drools-core/src/main/java/org/drools/core/addon/NoopOrderingStrategy.java b/drools-core/src/main/java/org/drools/core/addon/NoopOrderingStrategy.java
@@ -15,17 +15,18 @@
 
 package org.drools.core.addon;
 
+import java.util.Collection;
 import java.util.List;
-import java.util.Set;
+import java.util.Map;
 
+import org.drools.core.definitions.InternalKnowledgePackage;
 import org.drools.core.spi.AlphaNodeFieldConstraint;
 import org.drools.core.spi.ObjectType;
-import org.kie.api.definition.rule.Rule;
 
 public class NoopOrderingStrategy implements AlphaNodeOrderingStrategy {
 
     @Override
-    public void analyzeAlphaConstraints(Set<Rule> ruleSet) {
+    public void analyzeAlphaConstraints(Map<String, InternalKnowledgePackage> pkgs, Collection<InternalKnowledgePackage> newPkgs) {
         // no op
     }
 

diff --git a/drools-core/src/main/java/org/drools/core/impl/KnowledgeBaseImpl.java b/drools-core/src/main/java/org/drools/core/impl/KnowledgeBaseImpl.java
@@ -55,7 +55,6 @@
 import org.drools.core.common.InternalWorkingMemory;
 import org.drools.core.common.RuleBasePartitionId;
 import org.drools.core.definitions.InternalKnowledgePackage;
-import org.drools.core.definitions.impl.KnowledgePackageImpl;
 import org.drools.core.definitions.rule.impl.RuleImpl;
 import org.drools.core.event.KieBaseEventSupport;
 import org.drools.core.factmodel.ClassDefinition;
@@ -850,7 +849,7 @@ private void internalAddPackages(Collection<InternalKnowledgePackage> clonedPkgs
             wm.flushPropagations();
         }
 
-        alphaNodeOrderingStrategy.analyzeAlphaConstraints(collectRules(pkgs, clonedPkgs));
+        alphaNodeOrderingStrategy.analyzeAlphaConstraints(pkgs, clonedPkgs);
 
         // we need to merge all byte[] first, so that the root classloader can resolve classes
         for (InternalKnowledgePackage newPkg : clonedPkgs) {
@@ -961,13 +960,6 @@ private void internalAddPackages(Collection<InternalKnowledgePackage> clonedPkgs
         }
     }
 
-    private Set<Rule> collectRules(Map<String, InternalKnowledgePackage> pkgs, Collection<InternalKnowledgePackage> newPkgs) {
-        Set<Rule> ruleSet = new HashSet<>();
-        pkgs.forEach((pkgName, pkg) -> ruleSet.addAll(pkg.getRules()));
-        newPkgs.forEach(pkg -> ruleSet.addAll(pkg.getRules())); // okay to overwrite
-        return ruleSet;
-    }
-
     public void processAllTypesDeclaration( Collection<InternalKnowledgePackage> pkgs ) {
         List<TypeDeclaration> allTypeDeclarations = new ArrayList<>();
         // Add all Type Declarations, this has to be done first incase packages cross reference each other during build process.