Fix projection profiler (#4908)

kuzudb · Feb 15, 2025 · 9e4cdc4 · 9e4cdc4
1 parent cf860af
commit 9e4cdc4
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 1 deletion.
diff --git a/src/processor/operator/projection.cpp b/src/processor/operator/projection.cpp
@@ -39,7 +39,12 @@ bool Projection::getNextTuplesInternal(ExecutionContext* context) {
     }
     // The if statement is added to avoid the cost of calculating numTuples when metric is disabled.
     if (metrics->numOutputTuple.enabled) [[unlikely]] {
-        metrics->numOutputTuple.increase(resultSet->getNumTuples(info.outputChunkPosSet));
+        if (info.outputChunkPosSet.empty()) {
+            // In COUNT(*) case we are projecting away everything and only track multiplicity
+            metrics->numOutputTuple.increase(resultSet->multiplicity);
+        } else {
+            metrics->numOutputTuple.increase(resultSet->getNumTuples(info.outputChunkPosSet));
+        }
     }
     return true;
 }

diff --git a/test/test_files/explain/explain.test b/test/test_files/explain/explain.test
@@ -65,3 +65,5 @@
 -LOG ProfileQuery
 -STATEMENT Profile MATCH (p:npytable) RETURN p.id
 ---- ok
+-STATEMENT PROFILE MATCH (a:person)-[]->(b:person) RETURN COUNT(*)
+---- ok