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 );
                }

                foreach ( $servers as $server ) {
                        $this->output( $server . " ", $server );
                        $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 ( !is_null( $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 );