Added new simple count(*) query and search equivalent (#498)

This is to compare ESQL to plan terms agg.
elastic · Oct 27, 2023 · 27c34d2 · 27c34d2
1 parent 17aa310
commit 27c34d2
Show file tree

Hide file tree

Showing 2 changed files with 46 additions and 0 deletions.
diff --git a/nyc_taxis/challenges/default.json b/nyc_taxis/challenges/default.json
@@ -987,6 +987,24 @@
           "warmup-iterations": 10,
           "iterations": 50
         },
+        {
+          "operation": "count_group_by_keyword_aggregation",
+          "clients": 1,
+          "warmup-iterations": 10,
+          "iterations": 50
+        },
+        {
+          "operation": "count_group_by_keyword_esql_shard_partitioning",
+          "clients": 1,
+          "warmup-iterations": 10,
+          "iterations": 50
+        },
+        {
+          "operation": "count_group_by_keyword_esql_doc_partitioning",
+          "clients": 1,
+          "warmup-iterations": 10,
+          "iterations": 50
+        },
         {
           "operation": "multi_terms-keyword",
           "clients": 1,

diff --git a/nyc_taxis/operations/default.json b/nyc_taxis/operations/default.json
@@ -1087,6 +1087,34 @@
       "query" : "FROM nyc_taxis | stats avg(total_amount) by rate_code_id | sort rate_code_id",
       "body": { "pragma": { "data_partitioning": "doc" } }
     },
+    {
+      "name": "count_group_by_keyword_aggregation",
+      "operation-type": "search",
+      "index": "nyc_taxis",
+      "request-timeout": 120,
+      "body": {
+        "size": 0,
+        "aggs": {
+          "by_rate_code_id": {
+            "terms": {
+              "field": "rate_code_id"
+            }
+          }
+        }
+      }
+    },
+    {
+      "name": "count_group_by_keyword_esql_shard_partitioning",
+      "operation-type": "esql",
+      "query" : "FROM nyc_taxis | stats count=count(*) by rate_code_id | sort count desc",
+      "body": { "pragma": { "data_partitioning": "shard" } }
+    },
+    {
+      "name": "count_group_by_keyword_esql_doc_partitioning",
+      "operation-type": "esql",
+      "query" : "FROM nyc_taxis | stats count=count(*) by rate_code_id | sort count desc",
+      "body": { "pragma": { "data_partitioning": "doc" } }
+    },
     {
       "name": "avg_passenger_count_filtered_aggregation",
       "operation-type": "search",