StatCounter.php

Go to the documentation of this file.

<?php
class StatCounter {
    protected $deltas = array(); // (key => count)

    protected function __construct() {}

    public static function singleton() {
        static $instance = null;
        if ( !$instance ) {
            $instance = new self();
        }
        return $instance;
    }

    public function incr( $key, $count = 1 ) {
        $this->deltas[$key] = isset( $this->deltas[$key] ) ? $this->deltas[$key] : 0;
        $this->deltas[$key] += $count;
        if ( PHP_SAPI === 'cli' ) {
            $this->flush();
        }
    }

    public function flush() {
        global $wgStatsMethod;

        $deltas = array_filter( $this->deltas ); // remove 0 valued entries
        if ( $wgStatsMethod === 'udp' ) {
            $this->sendDeltasUDP( $deltas );
        } elseif ( $wgStatsMethod === 'cache' ) {
            $this->sendDeltasMemc( $deltas );
        } else {
            // disabled
        }
        $this->deltas = array();
    }

    protected function sendDeltasUDP( array $deltas ) {
        global $wgUDPProfilerHost, $wgUDPProfilerPort, $wgAggregateStatsID,
            $wgStatsFormatString;

        $id = strlen( $wgAggregateStatsID ) ? $wgAggregateStatsID : wfWikiID();

        $lines = array();
        foreach ( $deltas as $key => $count ) {
            $lines[] = sprintf( $wgStatsFormatString, $id, $count, $key );
        }

        if ( count( $lines ) ) {
            static $socket = null;
            if ( !$socket ) {
                $socket = socket_create( AF_INET, SOCK_DGRAM, SOL_UDP );
            }
            $packet = '';
            $packets = array();
            foreach ( $lines as $line ) {
                if ( ( strlen( $packet ) + strlen( $line ) ) > 1450 ) {
                    $packets[] = $packet;
                    $packet = '';
                }
                $packet .= $line;
            }
            if ( $packet != '' ) {
                $packets[] = $packet;
            }
            foreach ( $packets as $packet ) {
                wfSuppressWarnings();
                socket_sendto(
                    $socket,
                    $packet,
                    strlen( $packet ),
                    0,
                    $wgUDPProfilerHost,
                    $wgUDPProfilerPort
                );
                wfRestoreWarnings();
            }
        }
    }

    protected function sendDeltasMemc( array $deltas ) {
        global $wgMemc;

        foreach ( $deltas as $key => $count ) {
            $ckey = wfMemcKey( 'stats', $key );
            if ( $wgMemc->incr( $ckey, $count ) === null ) {
                $wgMemc->add( $ckey, $count );
            }
        }
    }
}