DolQueryCollector.php

<?php

use DebugBar\DataCollector\AssetProvider;
use DebugBar\DataCollector\DataCollector;
use DebugBar\DataCollector\Renderable;
use DebugBar\DebugBarException;

dol_include_once('/debugbar/class/TraceableDB.php');

class DolQueryCollector extends DataCollector implements Renderable, AssetProvider
{
  protected $db;

  public function __construct()
  {
    global $db;

    // Replace $db handler with new handler override by TraceableDB
    $db = new TraceableDB($db);

    $this->db = $db;
  }

  public function collect()
  {
    $queries = array();
    $totalExecTime = 0;
    $totalMemoryUsage = 0;
    $totalFailed = 0;
    foreach ($this->db->queries as $query) {
      $queries[] = array(
        'sql' => $query['sql'],
        'duration' => $query['duration'],
        'memory' => $query['memory_usage'],
        'is_success' => $query['is_success'],
        'error_code' => $query['error_code'],
        'error_message' => $query['error_message']
      );
      $totalExecTime += $query['duration'];
      $totalMemoryUsage += $query['memory_usage'];
      if (!$query['is_success']) {
        $totalFailed += 1;
      }
    }

    return array(
      'nb_statements' => count($queries),
      'nb_failed_statements' => $totalFailed,
      'accumulated_duration' => $totalExecTime,
      'memory_usage' => $totalMemoryUsage,
      'statements' => $queries
    );
  }

  public function getName()
  {
    return 'query';
  }

  public function getWidgets()
  {
    global $langs;

    $title = $langs->transnoentities('Database');

    return array(
      "$title" => array(
        "icon" => "arrow-right",
        "widget" => "PhpDebugBar.Widgets.SQLQueriesWidget",
        "map" => "query",
        "default" => "[]"
      ),
      "$title:badge" => array(
        "map" => "query.nb_statements",
        "default" => 0
      )
    );
  }

  public function getAssets()
  {
    return array(
      'css' => 'widgets/sqlqueries/widget.css',
      'js' => 'widgets/sqlqueries/widget.js'
    );
  }
}