add support for injecting custom handlers

devlop · Feb 10, 2021 · 67c50f9 · 67c50f9
1 parent fe59baa
commit 67c50f9
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/UtmParameters.php b/src/UtmParameters.php
@@ -114,6 +114,10 @@ public static function fake() : UtmParametersInterface
      */
     private static function getRequestHandler($request) : RequestHandlerInterface
     {
+        if ($request instanceof RequestHandlerInterface) {
+            return $request;
+        }
+
         foreach (self::$requestHandlers as $interface => $handler) {
             if ($request instanceof $interface) {
                 return new $handler;
@@ -136,6 +140,10 @@ private static function getRequestHandler($request) : RequestHandlerInterface
      */
     private function getResponseHandler($response) : ResponseHandlerInterface
     {
+        if ($response instanceof ResponseHandlerInterface) {
+            return $response;
+        }
+
         foreach ($this->responseHandlers as $interface => $handler) {
             if ($response instanceof $interface) {
                 return new $handler;