Merge pull request #14 from hschimpf/2.x-dev

New option to override awaitTasksCompletion() validation

src/Internals/Commands/Runner/WaitTasksCompletionMessage.php

@@ -2,6 +2,7 @@
 
 namespace HDSSolutions\Console\Parallel\Internals\Commands\Runner;
 
+use Closure;
 use HDSSolutions\Console\Parallel\Internals\Commands\ParallelCommandMessage;
 use HDSSolutions\Console\Parallel\Internals\Runner;
 
@@ -10,8 +11,11 @@
  */
 final class WaitTasksCompletionMessage extends ParallelCommandMessage {
 
-    public function __construct() {
-        parent::__construct('await');
+    /**
+     * @param  Closure  $or_until
+     */
+    public function __construct(Closure $or_until) {
+        parent::__construct('await', [ $or_until ]);
     }
 
 }

src/Internals/Runner.php

@@ -197,9 +197,9 @@ protected function update(): void {
         }
     }
 
-    protected function await(): bool {
+    protected function await(Closure $or_until): bool {
         if (PARALLEL_EXT_LOADED) {
-            return $this->send($this->hasPendingTasks() || $this->hasRunningTasks());
+            return $this->send($or_until() === false && ($this->hasPendingTasks() || $this->hasRunningTasks()));
         }
 
         return true;

src/Scheduler.php

@@ -110,9 +110,13 @@ public static function runTask(mixed ...$data): int {
 
     /**
      * Calling this method will pause execution until all tasks are finished.
+     *
+     * @param  Closure|null  $or_until  Custom validation to stop waiting.
      */
-    public static function awaitTasksCompletion(): bool {
-        $message = new Commands\Runner\WaitTasksCompletionMessage();
+    public static function awaitTasksCompletion(Closure $or_until = null): bool {
+        $message = new Commands\Runner\WaitTasksCompletionMessage(
+            or_until: $or_until ?? static fn() => false,
+        );
 
         if (PARALLEL_EXT_LOADED) {
             $has_pending_tasks = false;