lib/choria/orchestrator/task/result_set.rb in choria-colt-0.7.0 vs lib/choria/orchestrator/task/result_set.rb in choria-colt-0.8.0

@@ -4,13 +4,16 @@
   class Orchestrator
     class Task
       class ResultSet
         attr_reader :results
 
+        attr_accessor :pending_count
+
         def initialize(on_result:)
           @results = []
           @on_result = on_result
+          @pending_count = 0
         end
 
         def integrate_rpc_error(rpc_error)
           result = rpc_error[:body]
           result[:sender] = rpc_error[:senderid]
@@ -18,10 +21,10 @@
         end
 
         def integrate_result(result)
           structured_result = Choria::Colt::DataStructurer.structure(result).with_indifferent_access
           @results << structured_result
-          @on_result&.call(structured_result)
+          @on_result&.call(structured_result, @results.count, pending_count + @results.count)
         end
       end
     end
   end
 end