Homec4science
Diffusion libphutil (production) Browse

libphutil/src/serviceprofiler/PhutilServiceProfiler.php
7f4334d03989production

Last Change
Raw File

PhutilServiceProfiler.php
View Options

<?php
/*
* Copyright 2012 Facebook, Inc.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
/**
* Simple event store for service calls, so they can be printed to stdout or
* displayed in a debug console.
*
* @group util
*/
final class PhutilServiceProfiler {
private static $instance;
private $listeners = array();
private $events = array();
private $logSize = 0;
private $discardMode = false;
private function __construct() {
}
public function enableDiscardMode() {
$this->discardMode = true;
}
public static function getInstance() {
if (empty(self::$instance)) {
self::$instance = new PhutilServiceProfiler();
}
return self::$instance;
}
public function beginServiceCall(array $data) {
$data['begin'] = microtime(true);
$id = $this->logSize++;
$this->events[$id] = $data;
foreach ($this->listeners as $listener) {
call_user_func($listener, 'begin', $id, $data);
}
return $id;
}
public function endServiceCall($call_id, array $data) {
$data = ($this->events[$call_id] + $data);
$data['end'] = microtime(true);
$data['duration'] = ($data['end'] - $data['begin']);
$this->events[$call_id] = $data;
foreach ($this->listeners as $listener) {
call_user_func($listener, 'end', $call_id, $data);
}
if ($this->discardMode) {
unset($this->events[$call_id]);
}
}
public function getServiceCallLog() {
return $this->events;
}
public function addListener($callback) {
$this->listeners[] = $callback;
}
public static function installEchoListener() {
$instance = PhutilServiceProfiler::getInstance();
$instance->addListener(array('PhutilServiceProfiler', 'echoListener'));
}
public static function echoListener($type, $id, $data) {
$is_begin = false;
$is_end = false;
switch ($type) {
case 'begin':
$is_begin = true;
$mark = '>>>';
break;
case 'end':
$is_end = true;
$mark = '<<<';
break;
default:
$mark = null;
break;
}
$type = idx($data, 'type', 'mystery');
$desc = null;
if ($is_begin) {
switch ($type) {
case 'query':
$desc = substr($data['query'], 0, 512);
break;
case 'exec':
$desc = '$ '.$data['command'];
break;
case 'conduit':
$desc = $data['method'].'()';
break;
case 'lock':
$desc = $data['name'];
break;
case 'event':
$desc = $data['kind'].' <listeners = '.$data['count'].'>';
break;
}
} else if ($is_end) {
$desc = number_format((int)(1000000 * $data['duration'])).' us';
}
$console = PhutilConsole::getConsole();
$console->writeLog(
"%s [%s] <%s> %s\n",
$mark,
$id,
$type,
$desc);
}
}
c4science · Help