Profiler.php

Go to the documentation of this file.

<?php
use Wikimedia\ScopedCallback;
use Wikimedia\Rdbms\TransactionProfiler;

abstract class Profiler {
    protected $profileID = false;
    protected $templated = false;
    protected $params = [];
    protected $context = null;
    protected $trxProfiler;
    private static $instance = null;

    public function __construct( array $params ) {
        if ( isset( $params['profileID'] ) ) {
            $this->profileID = $params['profileID'];
        }
        $this->params = $params;
        $this->trxProfiler = new TransactionProfiler();
    }

    final public static function instance() {
        if ( self::$instance === null ) {
            global $wgProfiler, $wgProfileLimit;

            $params = [
                'class'     => ProfilerStub::class,
                'sampling'  => 1,
                'threshold' => $wgProfileLimit,
                'output'    => [],
            ];
            if ( is_array( $wgProfiler ) ) {
                $params = array_merge( $params, $wgProfiler );
            }

            $inSample = mt_rand( 0, $params['sampling'] - 1 ) === 0;
            // wfIsCLI() is not available yet
            if ( PHP_SAPI === 'cli' || PHP_SAPI === 'phpdbg' || !$inSample ) {
                $params['class'] = ProfilerStub::class;
            }

            if ( !is_array( $params['output'] ) ) {
                $params['output'] = [ $params['output'] ];
            }

            self::$instance = new $params['class']( $params );
        }
        return self::$instance;
    }

    final public static function replaceStubInstance( Profiler $profiler ) {
        if ( self::$instance && !( self::$instance instanceof ProfilerStub ) ) {
            throw new MWException( 'Could not replace non-stub profiler instance.' );
        } else {
            self::$instance = $profiler;
        }
    }

    public function setProfileID( $id ) {
        $this->profileID = $id;
    }

    public function getProfileID() {
        if ( $this->profileID === false ) {
            return WikiMap::getCurrentWikiDbDomain()->getId();
        } else {
            return $this->profileID;
        }
    }

    public function setContext( $context ) {
        $this->context = $context;
    }

    public function getContext() {
        if ( $this->context ) {
            return $this->context;
        } else {
            wfDebug( __METHOD__ . " called and \$context is null. " .
                "Return RequestContext::getMain(); for sanity\n" );
            return RequestContext::getMain();
        }
    }

    // Kept BC for now, remove when possible
    public function profileIn( $functionname ) {
    }

    public function profileOut( $functionname ) {
    }

    abstract public function scopedProfileIn( $section );

    public function scopedProfileOut( SectionProfileCallback &$section = null ) {
        $section = null;
    }

    public function getTransactionProfiler() {
        return $this->trxProfiler;
    }

    abstract public function close();

    private function getOutputs() {
        $outputs = [];
        foreach ( $this->params['output'] as $outputType ) {
            // The class may be specified as either the full class name (for
            // example, 'ProfilerOutputStats') or (for backward compatibility)
            // the trailing portion of the class name (for example, 'stats').
            $outputClass = strpos( $outputType, 'ProfilerOutput' ) === false
                ? 'ProfilerOutput' . ucfirst( $outputType )
                : $outputType;
            if ( !class_exists( $outputClass ) ) {
                throw new MWException( "'$outputType' is an invalid output type" );
            }
            $outputInstance = new $outputClass( $this, $this->params );
            if ( $outputInstance->canUse() ) {
                $outputs[] = $outputInstance;
            }
        }
        return $outputs;
    }

    public function logData() {
        $request = $this->getContext()->getRequest();

        $timeElapsed = $request->getElapsedTime();
        $timeElapsedThreshold = $this->params['threshold'];
        if ( $timeElapsed <= $timeElapsedThreshold ) {
            return;
        }

        $outputs = $this->getOutputs();
        if ( !$outputs ) {
            return;
        }

        $stats = $this->getFunctionStats();
        foreach ( $outputs as $output ) {
            $output->log( $stats );
        }
    }

    public function logDataPageOutputOnly() {
        foreach ( $this->getOutputs() as $output ) {
            if ( $output instanceof ProfilerOutputText ) {
                $stats = $this->getFunctionStats();
                $output->log( $stats );
            }
        }
    }

    public function getContentType() {
        foreach ( headers_list() as $header ) {
            if ( preg_match( '#^content-type: (\w+/\w+);?#i', $header, $m ) ) {
                return $m[1];
            }
        }
        return null;
    }

    public function setTemplated( $t ) {
        $this->templated = $t;
    }

    public function getTemplated() {
        return $this->templated;
    }

    abstract public function getFunctionStats();

    abstract public function getOutput();
}