TraceableContainer.php

<?php

namespace Drupal\webprofiler\DependencyInjection;

use Drupal\Component\Utility\Timer;
use Drupal\Core\DependencyInjection\Container;

/**
 * Extends the Drupal container class to trace service instantiations.
 */
class TraceableContainer extends Container {
    
    /**
     * @var array
     */
    protected $tracedData;
    
    /**
     * @var \Symfony\Component\Stopwatch\Stopwatch
     */
    private $stopwatch = NULL;
    
    /**
     * @var bool
     */
    private $hasStopwatch = FALSE;
    
    /**
     * @param string $id
     * @param int $invalidBehavior
     *
     * @return object
     */
    public function get($id, $invalidBehavior = self::EXCEPTION_ON_INVALID_REFERENCE) {
        if (!$this->stopwatch && $this->has('stopwatch')) {
            $this->stopwatch = parent::get('stopwatch');
            $this->stopwatch
                ->openSection();
            $this->hasStopwatch = TRUE;
        }
        if ('stopwatch' === $id) {
            return $this->stopwatch;
        }
        Timer::start($id);
        if ($this->hasStopwatch) {
            $e = $this->stopwatch
                ->start($id, 'service');
        }
        $service = parent::get($id, $invalidBehavior);
        $this->tracedData[$id] = Timer::stop($id);
        if ($this->hasStopwatch && $e->isStarted()) {
            $e->stop();
        }
        return $service;
    }
    
    /**
     * @return array
     */
    public function getTracedData() {
        return $this->tracedData;
    }

}