Fix for using multiple middlware in the same time (#7)

fixing multiple middlewares on key: on_stats
eMAGTechLabs · Dec 17, 2019 · 662beeb · 662beeb
1 parent cd099db
commit 662beeb
Show file tree

Hide file tree

Showing 3 changed files with 15 additions and 3 deletions.
diff --git a/composer.json b/composer.json
@@ -1,7 +1,7 @@
 {
     "name": "emag-tech-labs/guzzle-middleware",
     "description": "Guzzle middlewares",
-    "version": "0.2.7",
+    "version": "0.2.8",
     "license": "GPL-3.0-only",
     "authors": [
         {

diff --git a/src/GuzzleMiddleware/HttpCodeProfiler.php b/src/GuzzleMiddleware/HttpCodeProfiler.php
@@ -26,7 +26,13 @@ public function __construct(StatsDataInterface $statsdService)
     public function __invoke(callable $handler): callable
     {
         return function (RequestInterface $request, array $options) use ($handler) {
-            $options['on_stats'] = function (TransferStats $stats) {
+            $onStats = $options['on_stats'] ?? null;
+
+            $options['on_stats'] = function (TransferStats $stats) use ($onStats) {
+                if (is_callable($onStats)) {
+                    $onStats($stats);
+                }
+
                 $this->startProfiling($stats);
             };
 

diff --git a/src/GuzzleMiddleware/TimingProfiler.php b/src/GuzzleMiddleware/TimingProfiler.php
@@ -25,7 +25,13 @@ public function __construct(StatsDataInterface $statsdService)
     public function __invoke(callable $handler): callable
     {
         return function (RequestInterface $request, array $options) use ($handler) {
-            $options['on_stats'] = function (TransferStats $stats) {
+            $onStats = $options['on_stats'] ?? null;
+
+            $options['on_stats'] = function (TransferStats $stats) use ($onStats) {
+                if (is_callable($onStats)) {
+                    $onStats($stats);
+                }
+
                 $this->startProfiling($stats);
             };