xhprof.drush.inc in XHProf 7

File

xhprof.drush.inc

<?php

/**
 * @file
 */

/**
 * Implements hook_drush_command().
 */
function xhprof_drush_command() {
  $items['xhprof-list'] = array(
    'description' => dt(''),
    'arguments' => array(),
  );
  $items['xhprof-combine'] = array(
    'description' => dt(''),
    'arguments' => array(),
  );
  $items['xhprof-clear'] = array(
    'description' => dt(''),
    'arguments' => array(),
  );
  return $items;
}

/**
 * A command callback.
 */
function drush_xhprof_combine() {
  $run_ids = func_get_args();
  $runs = drush_xhprof_get_runs();
  if (empty($run_ids)) {
    $options = array_keys($runs);
    $options[] = 'All';

    // return dt("You must provide a run id argument!");

    //$choice = drush_choice_multiple($options, FALSE, 'Select xhprof runs to combine.');
    $choice = drush_choice($options, 'Select xhprof runs to combine.');
    drush_print_r($options[$choice]);
    if ($choice !== FALSE) {
      if ($options[$choice] == 'All') {
        $run_ids = array_keys($runs);
      }
      else {

        // TODO: This doesn't work. Probably going to have to use drush_choice_multiple.
        $ids = explode(",", $choice);
        foreach ($ids as $id) {
          $run_ids[] = $options[$id];
        }
      }
    }
  }
  $xhprof = new XHProfRuns_Default();
  $runs = drush_xhprof_get_runs();
  $run_data = array();
  $keys = array();
  $desc = "";
  foreach ($run_ids as $run_id) {
    if (isset($runs[$run_id])) {
      $run = $runs[$run_id];
      if ($data = $xhprof
        ->get_run($run['run_id'], $run['source'], $desc)) {
        $run_data[] = $data;
        $keys = $keys + array_keys($data);
      }
    }
  }
  $agg_run = array();
  $run_count = count($run_data);
  foreach ($keys as $key) {
    $agg_key = array();

    // Check which runs have this parent_child function key, collect metrics if so.
    foreach ($run_data as $data) {
      if (isset($data[$key])) {
        foreach ($data[$key] as $metric => $val) {
          $agg_key[$metric][] = $val;
        }
      }
    }

    // Average each metric for the key into the aggregated run.
    $agg_run[$key] = array();
    foreach ($agg_key as $metric => $vals) {
      $agg_run[$key][$metric] = array_sum($agg_key[$metric]) / count($agg_key[$metric]);
    }
  }
  $namespace = 'drush-' . variable_get('site_name', '');
  drupal_alter('xhprof_namespace', $namespace);
  if ($agg_run_id = $xhprof
    ->save_run($agg_run, $namespace)) {
    drush_print(dt("Aggregated run id: !id", array(
      '!id' => $agg_run_id,
    )));
  }
  else {
    drush_print(dt("Unable to save aggregated xhprof data!"));
  }
}

/**
 * A command callback.
 */
function drush_xhprof_list() {
  $runs = drush_xhprof_get_runs();
  $rows = array();
  $headers = array();
  foreach ($runs as $run) {
    $row = array();
    foreach ($run as $key => $value) {
      if (!in_array($key, $headers)) {
        $headers[] = $key;
      }
      $row[] = $value;
    }
    $rows[] = array_reverse($row);
  }
  $headers = array_reverse($headers);
  array_unshift($rows, $headers);
  drush_print_table($rows, TRUE);
}

/**
 * A command callback.
 */
function drush_xhprof_clear() {
  drush_xhprof_get_dir();
  foreach (drush_xhprof_list_run_files() as $file) {
    drush_register_file_for_deletion($file);
  }
}
function drush_xhprof_get_dir() {
  $dir = ini_get("xhprof.output_dir");
  return !empty($dir) && is_dir($dir) ? $dir = "/tmp" : FALSE;
}
function drush_xhprof_list_run_files() {
  $dir = drush_xhprof_get_dir();
  $files = array();
  foreach (glob("{$dir}/*.xhprof") as $file) {
    $files[] = $file;
  }
  return $files;
}
function drush_xhprof_get_runs() {
  $dir = drush_xhprof_get_dir();
  $run_ids = array();
  foreach (glob("{$dir}/*.xhprof") as $file) {
    $run = array();
    list($run['run_id'], $run['source']) = explode('.', basename($file));
    $runs[$run['run_id']] = $run;
  }
  return $runs;
}

Functions

Name	Description
drush_xhprof_clear	A command callback.
drush_xhprof_combine	A command callback.
drush_xhprof_get_dir
drush_xhprof_get_runs
drush_xhprof_list	A command callback.
drush_xhprof_list_run_files
xhprof_drush_command	Implements hook_drush_command().