Fixed bad design

bpolaszek · Mar 8, 2017 · 8c05c62 · 8c05c62
1 parent cb5fb2b
commit 8c05c62
Showing 1 changed file with 1 addition and 10 deletions.
diff --git a/src/Extractor/CallbackExtractor.php b/src/Extractor/CallbackExtractor.php
@@ -35,16 +35,7 @@ public function __invoke($key, $value): ContextElementInterface
     {
         $element = parent::__invoke($key, $value);
         $callback = $this->callback;
-        $result = $callback($element);
-        if (null !== $result) {
-            if (!$result instanceof ContextElementInterface) {
-                throw new \RuntimeException(
-                    sprintf("The callback should either return nothing or a %s class.", ContextElementInterface::class)
-                );
-            } else {
-                $element = $result;
-            }
-        }
+        $callback($element);
         return $element;
     }
 }