ProfilerOutputDb.php

Go to the documentation of this file.

<?php
use Wikimedia\Rdbms\Database;
use Wikimedia\Rdbms\DBError;
 
class ProfilerOutputDb extends ProfilerOutput {
    private $perHost = false;
 
    public function __construct( Profiler $collector, array $params ) {
        parent::__construct( $collector, $params );
 
        // Initialize per-host profiling from config, back-compat if available
        if ( isset( $this->params['perHost'] ) ) {
            $this->perHost = $this->params['perHost'];
        }
    }
 
    public function canUse() {
        # Do not log anything if database is readonly (T7375)
        return !wfReadOnly();
    }
 
    public function log( array $stats ) {
        try {
            $dbw = wfGetDB( DB_MASTER );
        } catch ( DBError $e ) {
            return; // ignore
        }
 
        $fname = __METHOD__;
        $dbw->onTransactionCommitOrIdle( function ( Database $dbw ) use ( $stats, $fname ) {
            $pfhost = $this->perHost ? wfHostname() : '';
            // Sqlite: avoid excess b-tree rebuilds (mostly for non-WAL mode)
            // non-Sqlite: lower contention with small transactions
            $useTrx = ( $dbw->getType() === 'sqlite' );
 
            try {
                $useTrx ? $dbw->startAtomic( $fname ) : null;
 
                foreach ( $stats as $data ) {
                    $name = $data['name'];
                    $eventCount = $data['calls'];
                    $timeSum = (float)$data['real'];
                    $memorySum = (float)$data['memory'];
                    $name = substr( $name, 0, 255 );
 
                    // Kludge
                    $timeSum = $timeSum >= 0 ? $timeSum : 0;
                    $memorySum = $memorySum >= 0 ? $memorySum : 0;
 
                    $dbw->upsert( 'profiling',
                        [
                            'pf_name' => $name,
                            'pf_count' => $eventCount,
                            'pf_time' => $timeSum,
                            'pf_memory' => $memorySum,
                            'pf_server' => $pfhost
                        ],
                        [ [ 'pf_name', 'pf_server' ] ],
                        [
                            "pf_count=pf_count+{$eventCount}",
                            "pf_time=pf_time+{$timeSum}",
                            "pf_memory=pf_memory+{$memorySum}",
                        ],
                        $fname
                    );
                }
            } catch ( DBError $e ) {
                // ignore
            }
 
            try {
                $useTrx ? $dbw->endAtomic( $fname ) : null;
            } catch ( DBError $e ) {
                // ignore
            }
        } );
    }
}