[9.x] Add onFailure method to Pipeline

Author: reziamini

src/Illuminate/Pipeline/Pipeline.php

@@ -24,6 +24,13 @@ class Pipeline implements PipelineContract
      */
     protected $passable;
 
+    /**
+     * The callback to be executed on failure pipeline.
+     *
+     * @var Closure
+     */
+    protected $onFailure;
+
     /**
      * The array of class pipes.
      *
@@ -182,6 +189,10 @@ protected function carry()
 
                     return $this->handleCarry($carry);
                 } catch (Throwable $e) {
+                    if ($this->onFailure) {
+                        return call_user_func($this->onFailure, $pipe);
+                    }
+
                     return $this->handleException($passable, $e);
                 }
             };
@@ -244,6 +255,19 @@ public function setContainer(Container $container)
         return $this;
     }
 
+    /**
+     * Set callback to be executed on failure pipeline.
+     *
+     * @param  Closure  $catch
+     * @return $this
+     */
+    public function onFailure(Closure $callback)
+    {
+        $this->onFailure = $callback;
+
+        return $this;
+    }
+
     /**
      * Handle the value returned from each pipe before passing it to the next.
      *

tests/Pipeline/PipelineTest.php

@@ -231,6 +231,20 @@ public function testPipelineThrowsExceptionOnResolveWithoutContainer()
             });
     }
 
+    public function testExceptionIsHandledByOnFailureMethodInPipeline()
+    {
+        $result = (new Pipeline(new Container))
+            ->send('data')
+            ->through(PipelineWithException::class)
+            ->onFailure(function () {
+                return 'error';
+            })->then(function ($piped) {
+                return $piped;
+            });
+
+        $this->assertEquals('error', $result);
+    }
+
     public function testPipelineThenReturnMethodRunsPipelineThenReturnsPassable()
     {
         $result = (new Pipeline(new Container))
@@ -279,3 +293,11 @@ public function handle($piped, $next, $parameter1 = null, $parameter2 = null)
         return $next($piped);
     }
 }
+
+class PipelineWithException
+{
+    public function handle($piped, $next)
+    {
+        throw new \Exception('Foo');
+    }
+}