lib/bolt/executor.rb in bolt-3.13.0 vs lib/bolt/executor.rb in bolt-3.14.1

@@ -379,14 +379,22 @@
 
     # Call into FiberExecutor to avoid this class getting
     # overloaded while also minimizing the Puppet lookups needed from plan
     # functions
     #
-    def create_future(scope: nil, name: nil, &block)
-      @fiber_executor.create_future(scope: scope, name: name, &block)
+    def create_future(plan_id:, scope: nil, name: nil, &block)
+      @fiber_executor.create_future(scope: scope, name: name, plan_id: plan_id, &block)
     end
 
+    def get_current_future(fiber:)
+      @fiber_executor.get_current_future(fiber: fiber)
+    end
+
+    def get_current_plan_id(fiber:)
+      @fiber_executor.get_current_plan_id(fiber: fiber)
+    end
+
     def plan_complete?
       @fiber_executor.plan_complete?
     end
 
     def round_robin
@@ -399,11 +407,11 @@
 
     def wait(futures, **opts)
       @fiber_executor.wait(futures, **opts)
     end
 
-    def plan_futures
-      @fiber_executor.plan_futures
+    def get_futures_for_plan(plan_id:)
+      @fiber_executor.get_futures_for_plan(plan_id: plan_id)
     end
 
     # Execute a plan function concurrently. This function accepts the executor
     # function to be run and the parameters to pass to it, and returns the
     # result of running the executor function.