Add on rejected callback to downloader in engine

Author: Fadarrizz

src/Core/Engine.php

@@ -72,9 +72,12 @@ public function start(Run $run): void
 
     private function work(Run $run): void
     {
+        $onFulfilled = fn (Response $response) => $this->onFulfilled($response);
+        $onRejected = fn (\Throwable $exception, Request $request) => $this->onRejected($exception, $request);
+
         while (!$this->scheduler->empty()) {
             foreach ($this->scheduler->nextRequests($run->concurrency) as $request) {
-                $this->downloader->prepare($request);
+                $this->downloader->prepare($request, $onRejected);
             }
 
             // It's possible that requests were dropped while sending them through the
@@ -94,15 +97,13 @@ private function work(Run $run): void
                 $difference = $run->concurrency - $scheduledRequests;
 
                 foreach ($this->scheduler->forceNextRequests($difference) as $request) {
-                    $this->downloader->prepare($request);
+                    $this->downloader->prepare($request, $onRejected);
                 }
 
                 $scheduledRequests = $this->downloader->scheduledRequests();
             }
 
-            $this->downloader->flush(
-                fn (Response $response) => $this->onFulfilled($response),
-            );
+            $this->downloader->flush($onFulfilled, $onRejected);
         }
 
         $this->eventDispatcher->dispatch(
@@ -125,6 +126,11 @@ private function onFulfilled(Response $response): void
         }
     }
 
+    private function onRejected(\Throwable $exception, Request $request): void
+    {
+        throw $exception;
+    }
+
     private function scheduleRequest(Request $request): void
     {
         $this->eventDispatcher->dispatch(