mctest.php

Go to the documentation of this file.

<?php
require_once __DIR__ . '/Maintenance.php';

class mcTest extends Maintenance {
    public function __construct() {
        parent::__construct();
        $this->mDescription = "Makes several 'set', 'incr' and 'get' requests on every"
            . " memcached server and shows a report";
        $this->addOption( 'i', 'Number of iterations', false, true );
        $this->addOption( 'cache', 'Use servers from this $wgObjectCaches store', false, true );
        $this->addArg( 'server[:port]', 'Memcached server to test, with optional port', false );
    }

    public function execute() {
        global $wgMainCacheType, $wgMemCachedTimeout, $wgObjectCaches;

        $cache = $this->getOption( 'cache' );
        $iterations = $this->getOption( 'i', 100 );
        if ( $cache ) {
            if ( !isset( $wgObjectCaches[$cache] ) ) {
                $this->error( "MediaWiki isn't configured with a cache named '$cache'", 1 );
            }
            $servers = $wgObjectCaches[$cache]['servers'];
        } elseif ( $this->hasArg() ) {
            $servers = array( $this->getArg() );
        } elseif ( $wgMainCacheType === CACHE_MEMCACHED ) {
            global $wgMemCachedServers;
            $servers = $wgMemCachedServers;
        } elseif ( isset( $wgObjectCaches[$wgMainCacheType]['servers'] ) ) {
            $servers = $wgObjectCaches[$wgMainCacheType]['servers'];
        } else {
            $this->error( "MediaWiki isn't configured for Memcached usage", 1 );
        }

        # find out the longest server string to nicely align output later on
        $maxSrvLen = $servers ? max( array_map( 'strlen', $servers ) ) : 0;

        foreach ( $servers as $server ) {
            $this->output(
                str_pad( $server, $maxSrvLen ),
                $server  # output channel
            );

            $mcc = new MemCachedClientforWiki( array(
                'persistant' => true,
                'timeout' => $wgMemCachedTimeout
            ) );
            $mcc->set_servers( array( $server ) );
            $set = 0;
            $incr = 0;
            $get = 0;
            $time_start = $this->microtime_float();
            for ( $i = 1; $i <= $iterations; $i++ ) {
                if ( $mcc->set( "test$i", $i ) ) {
                    $set++;
                }
            }
            for ( $i = 1; $i <= $iterations; $i++ ) {
                if ( !is_null( $mcc->incr( "test$i", $i ) ) ) {
                    $incr++;
                }
            }
            for ( $i = 1; $i <= $iterations; $i++ ) {
                $value = $mcc->get( "test$i" );
                if ( $value == $i * 2 ) {
                    $get++;
                }
            }
            $exectime = $this->microtime_float() - $time_start;

            $this->output( " set: $set   incr: $incr   get: $get time: $exectime", $server );
        }
    }

    private function microtime_float() {
        list( $usec, $sec ) = explode( " ", microtime() );
        return ( (float)$usec + (float)$sec );
    }
}

$maintClass = "mcTest";
require_once RUN_MAINTENANCE_IF_MAIN;