[4.x] Add afterEach method

src/Benchmark.php

@@ -29,26 +29,28 @@ class Benchmark
 
     protected ?Closure $beforeEach = null;
 
+    protected ?Closure $afterEach = null;
+
     protected array $result = [
         'each'  => [],
         'total' => [],
     ];
 
     public function __construct(
-        protected Runner $runner = new Runner(),
-        protected Transformer $transformer = new Transformer()
+        protected Runner $runner = new Runner,
+        protected Transformer $transformer = new Transformer
     ) {
         $this->view = new View(
             new SymfonyStyle(
-                new ArgvInput(),
-                new ConsoleOutput()
+                new ArgvInput,
+                new ConsoleOutput
             )
         );
     }
 
     public static function start(): static
     {
-        return new static();
+        return new static;
     }
 
     public function beforeEach(callable $callback): self
@@ -58,6 +60,13 @@ public function beforeEach(callable $callback): self
         return $this;
     }
 
+    public function afterEach(callable $callback): self
+    {
+        $this->afterEach = $callback;
+
+        return $this;
+    }
+
     public function iterations(int $count): self
     {
         $this->iterations = max(1, $count);
@@ -114,23 +123,25 @@ protected function each(mixed $name, callable $callback, ProgressBarService $pro
     protected function run(mixed $name, callable $callback, ProgressBarService $progressBar): void
     {
         for ($i = 1; $i <= $this->iterations; ++$i) {
-            $result = $this->runBeforeEach($name, $i);
+            $result = $this->runCallback($this->beforeEach, $name, $i);
 
             [$time, $ram] = $this->call($callback, [$i, $result]);
 
+            $this->runCallback($this->afterEach, $name, $i, $time, $ram);
+
             $this->push($name, $i, $time, $ram);
 
             $progressBar->advance();
         }
     }
 
-    protected function runBeforeEach(mixed $name, int $iteration): mixed
+    protected function runCallback(?Closure $callback, mixed ...$arguments): mixed
     {
-        if ($callback = $this->beforeEach) {
-            return $callback($name, $iteration);
+        if (! $callback) {
+            return null;
         }
 
-        return null;
+        return $callback(...$arguments);
     }
 
     protected function call(callable $callback, array $parameters = []): array