lekoala / silverstripe-devtoolkit

src/Benchmark.php

<?php

namespace LeKoala\DevToolkit;

use LeKoala\Base\Helpers\ClassHelper;
use SilverStripe\Core\Injector\Injector;

class Benchmark
{
    /**
     * @return Monolog\Logger
     */
    public static function getLogger()
    {
        $class = array_pop(explode("\\", get_called_class()));
        return Injector::inst()->get(LoggerInterface::class)->withName(ClassHelper::getClassWithoutNamespace($class));
    }

    /**
     * A dead simple benchmark function
     *
     * Usage : bm(function() { // Insert here the code to benchmark });
     * Alternative usage : bm() ; // Code to test ; bm();
     *
     * @param callable $cb
     * @return void
     */
    public static function run($cb = null)
    {
        $data = self::benchmark($cb);
        if (!$data) {

$data is a non-empty array, thus ! $data is always false.

            return;
        }

        printf("It took %s seconds and used %s memory", $data['time'], $data['memory']);
        die();
    }

    /**
     * @param callable $cb
     * @return bool|array
     */
    protected static function benchmark($cb = null)
    {
        static $data = null;

        // No callback scenario
        if ($cb === null) {
            if ($data === null) {
                $data = [
                    'startTime' => microtime(true),
                    'startMemory' => memory_get_usage(),
                ];
                // Allow another call
                return false;
            } else {
                $startTime = $data['startTime'];
                $startMemory = $data['startMemory'];

                // Clear for future calls
                $data = null;
            }
        } else {
            $startTime = microtime(true);
            $startMemory = memory_get_usage();

            $cb();
        }

        $endTime = microtime(true);
        $endMemory = memory_get_usage();

        $time = sprintf("%.6f", $endTime - $startTime);
        $memory = self::bytesToHuman($endMemory - $startMemory);

        return [
            'time' => $time,
            'memory' => $memory,
        ];
    }

    protected static function bytesToHuman($bytes, $decimals = 2)
    {
        $size   = array('B', 'kB', 'MB', 'GB', 'TB', 'PB', 'EB', 'ZB', 'YB');
        $factor = floor((strlen($bytes) - 1) / 3);

        return sprintf("%.{$decimals}f", $bytes / pow(1024, $factor)) . @$size[$factor];
    }

    public static function log($name, $cb = null)
    {
        $data = self::benchmark($cb);
        if (!$data) {

$data is a non-empty array, thus ! $data is always false.

            return;
        }

        $time = $data['time'];
        $memory = $data['memory'];

        self::getLogger()->debug("$name : $time seconds and $memory memory.");
    }
}