Add context argument to garrow_execute_plan_new

Author: stenlarsson

c_glib/arrow-glib/compute.cpp

@@ -1889,6 +1889,7 @@ garrow_execute_plan_class_init(GArrowExecutePlanClass *klass)
 
 /**
  * garrow_execute_plan_new:
+ * @context: (nullable): A #GArrowExecuteContext or %NULL.
  * @error: (nullable): Return location for a #GError or %NULL.
  *
  * Returns: (nullable): A newly created #GArrowExecutePlan on success,
@@ -1897,9 +1898,15 @@ garrow_execute_plan_class_init(GArrowExecutePlanClass *klass)
  * Since: 6.0.0
  */
 GArrowExecutePlan *
-garrow_execute_plan_new(GError **error)
+garrow_execute_plan_new(GArrowExecuteContext *context, GError **error)
 {
-  auto arrow_plan_result = arrow::acero::ExecPlan::Make();
+  arrow::Result<std::shared_ptr<arrow::acero::ExecPlan>> arrow_plan_result;
+  if (context) {
+    auto arrow_context = garrow_execute_context_get_raw(context);
+    arrow_plan_result = arrow::acero::ExecPlan::Make(*arrow_context);
+  } else {
+    arrow_plan_result = arrow::acero::ExecPlan::Make();
+  }
   if (garrow::check(error, arrow_plan_result, "[execute-plan][new]")) {
     return GARROW_EXECUTE_PLAN(
       g_object_new(GARROW_TYPE_EXECUTE_PLAN, "plan", &(*arrow_plan_result), NULL));

c_glib/arrow-glib/compute.h

@@ -323,7 +323,7 @@ struct _GArrowExecutePlanClass
 
 GARROW_AVAILABLE_IN_6_0
 GArrowExecutePlan *
-garrow_execute_plan_new(GError **error);
+garrow_execute_plan_new(GArrowExecuteContext *context, GError **error);
 GARROW_AVAILABLE_IN_6_0
 GArrowExecuteNode *
 garrow_execute_plan_build_node(GArrowExecutePlan *plan,

c_glib/test/test-execute-plan.rb

@@ -28,8 +28,8 @@ def execute(plan)
   end
 
   sub_test_case("aggregate") do
-    def build_plan
-      plan = Arrow::ExecutePlan.new
+    def build_plan(context = nil)
+      plan = Arrow::ExecutePlan.new(context)
 
       record_batch =
         build_record_batch(number: build_int8_array([1, 2, 3, 4, 5]),
@@ -63,6 +63,22 @@ def test_by_string
                      reader.read_all)
       end
     end
+
+    def test_hash_first_with_single_thread
+      thread_pool = Arrow::ThreadPool.new(1)
+      context = Arrow::ExecuteContext.new(thread_pool)
+      plan, reader = build_plan(context) do
+        aggregations = [
+          Arrow::Aggregation.new("hash_first", nil, "number", "first(number)"),
+        ]
+        Arrow::AggregateNodeOptions.new(aggregations, ["string"])
+      end
+      execute(plan) do
+        assert_equal(build_table("string" => build_string_array(["a", "b"]),
+                                 "first(number)" => build_int8_array([1, 2])),
+                     reader.read_all)
+      end
+    end
   end
 
   sub_test_case("hash join") do