Add support for NAryUnion to evaluation statistics.

pulquero · Oct 27, 2023 · a3207d6 · a3207d6
1 parent 7c4339d
commit a3207d6
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 3 deletions.
diff --git a/strategy/src/main/java/com/msd/gin/halyard/optimizers/ExtendedEvaluationStatistics.java b/strategy/src/main/java/com/msd/gin/halyard/optimizers/ExtendedEvaluationStatistics.java
@@ -1,6 +1,7 @@
 package com.msd.gin.halyard.optimizers;
 
 import com.google.common.collect.Iterables;
+import com.msd.gin.halyard.algebra.NAryUnion;
 import com.msd.gin.halyard.algebra.StarJoin;
 
 import java.io.IOException;
@@ -308,11 +309,24 @@ public void meetOther(QueryModelNode node) {
     			meet((TupleFunctionCall)node);
     		} else if (node instanceof StarJoin) {
     			meet((StarJoin)node);
+    		} else if (node instanceof NAryUnion) {
+    			meet((NAryUnion)node);
     		} else {
     			super.meetOther(node);
     		}
     	}
 
+        public void meet(NAryUnion node) {
+        	double card = 0.0;
+        	for (TupleExpr expr : node.getArgs()) {
+	            expr.visit(this);
+	            updateMap(expr);
+	            card += this.cardinality;
+        	}
+        	cardinality = card;
+            updateMap(node);
+        }
+
         public void meet(StarJoin node) {
         	TupleExpr sp = node.getArg(0);
         	sp.visit(this);

diff --git a/strategy/src/main/java/com/msd/gin/halyard/optimizers/NAryUnionOptimizer.java b/strategy/src/main/java/com/msd/gin/halyard/optimizers/NAryUnionOptimizer.java
@@ -20,11 +20,11 @@
 public class NAryUnionOptimizer implements QueryOptimizer {
 	private final int minUnions;
 
-	public NAryUnionOptimizer(int minJoins) {
-		if (minJoins < 1) {
+	public NAryUnionOptimizer(int minUnions) {
+		if (minUnions < 1) {
 			throw new IllegalArgumentException("Minimum unions must be greater than or equal to one");
 		}
-		this.minUnions = minJoins;
+		this.minUnions = minUnions;
 	}
 
 	@Override