Make sure we always have a stack (#134)

* Make sure we always have a stack * cs * style update
php-http · Feb 23, 2017 · de33f9c · de33f9c
1 parent 3384d0b
commit de33f9c
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/Collector/Collector.php b/Collector/Collector.php
@@ -62,7 +62,7 @@ public function getStacks()
     }
 
     /**
-     * @return Stack
+     * @return Stack|bool false if no current stack.
      */
     public function getCurrentStack()
     {

diff --git a/Collector/ProfilePlugin.php b/Collector/ProfilePlugin.php
@@ -58,7 +58,9 @@ public function handleRequest(RequestInterface $request, callable $next, callabl
     {
         $profile = new Profile($this->pluginName, $this->formatter->formatRequest($request));
 
-        $this->collector->getCurrentStack()->addProfile($profile);
+        if ($stack = $this->collector->getCurrentStack()) {
+            $stack->addProfile($profile);
+        }
 
         return $this->plugin->handleRequest($request, $next, $first)->then(function (ResponseInterface $response) use ($profile) {
             $profile->setResponse($this->formatter->formatResponse($response));