StatCounter.php

Go to the documentation of this file.

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

    protected $config;

    protected function __construct( Config $config ) {
        $this->config = $config;
    }

    public static function singleton() {
        static $instance = null;
        if ( !$instance ) {
            $instance = new self(
                ConfigFactory::getDefaultInstance()->makeConfig( 'main' )
            );
        }
        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() {
        $statsMethod = $this->config->get( 'StatsMethod' );
        $deltas = array_filter( $this->deltas ); // remove 0 valued entries
        if ( $statsMethod === 'udp' ) {
            $this->sendDeltasUDP( $deltas );
        } elseif ( $statsMethod === 'cache' ) {
            $this->sendDeltasMemc( $deltas );
        } else {
            // disabled
        }
        $this->deltas = array();
    }

    protected function sendDeltasUDP( array $deltas ) {
        $aggregateStatsID = $this->config->get( 'AggregateStatsID' );
        $id = strlen( $aggregateStatsID ) ? $aggregateStatsID : wfWikiID();

        $lines = array();
        foreach ( $deltas as $key => $count ) {
            $lines[] = sprintf( $this->config->get( 'StatsFormatString' ), $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,
                    $this->config->get( 'UDPProfilerHost' ),
                    $this->config->get( 'UDPProfilerPort' )
                );
                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 );
            }
        }
    }
}