changed contextualinterface and contextualtrait to accept objects

joaojacome · Sep 20, 2018 · b502805 · b502805
1 parent 021b7ba
commit b502805
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 3 deletions.
diff --git a/ContextualInterface.php b/ContextualInterface.php
@@ -12,11 +12,11 @@ interface ContextualInterface
     /**
      * Set the context for the logger
      *
-     * @param array $context
+     * @param mixed $context
      *
      * @return mixed
      */
-    public function setContext(array $context);
+    public function setContext($context);
 
     /**
      * Retrieve the context of this class

diff --git a/Helper/ContextualTrait.php b/Helper/ContextualTrait.php
@@ -2,6 +2,9 @@
 
 namespace KoderHut\OnelogBundle\Helper;
 
+use KoderHut\OnelogBundle\ContextualInterface;
+
+
 /**
  * Trait ContextualTrait
  *
@@ -17,8 +20,14 @@ trait ContextualTrait
     /**
      * @inheritdoc
      */
-    public function setContext(array $context): self
+    public function setContext($context): self
     {
+        if (is_object($context) && $context instanceof ContextualInterface) {
+            $context = $context->getContext();
+        } else if (is_object($context)) {
+            throw new \InvalidArgumentException();
+        }
+
         $this->contextualData = $context;
 
         return $this;