]> scripts.mit.edu Git - autoinstallsdev/mediawiki.git/blobdiff - maintenance/mcc.php
MediaWiki 1.30.2
[autoinstallsdev/mediawiki.git] / maintenance / mcc.php
index 97a344fb24cfb1e928d7fcf738460c439274134d..784ba0ea357496bee9c3d0118a0331561587e40a 100644 (file)
 /**
  * memcached diagnostic tool
  *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 2 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License along
+ * with this program; if not, write to the Free Software Foundation, Inc.,
+ * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
+ * http://www.gnu.org/copyleft/gpl.html
+ *
+ * @file
  * @todo document
- * @addtogroup Maintenance
+ * @ingroup Maintenance
  */
 
-/** */
-require_once( 'commandLine.inc' );
+$optionsWithArgs = [ 'cache' ];
+$optionsWithoutArgs = [
+       'debug', 'help'
+];
+require_once __DIR__ . '/commandLine.inc';
 
-$mcc = new memcached( array('persistant' => true/*, 'debug' => true*/) );
-$mcc->set_servers( $wgMemCachedServers );
-#$mcc->set_debug( true );
+$debug = isset( $options['debug'] );
+$help = isset( $options['help'] );
+$cache = isset( $options['cache'] ) ? $options['cache'] : null;
 
-function mccShowHelp($command) {
-
-       if(! $command ) { $command = 'fullhelp'; }
-       $onlyone = true;
-
-       switch ( $command ) {
-
-               case 'fullhelp':
-                       // will show help for all commands
-                       $onlyone = false;
-
-               case 'get':
-                       print "get: grabs something\n";
-               if($onlyone) { break; }
-
-               case 'getsock':
-                       print "getsock: lists sockets\n";
-               if($onlyone) { break; }
+if ( $help ) {
+       mccShowUsage();
+       exit( 0 );
+}
+$mcc = new MemcachedClient( [
+       'persistent' => true,
+       'debug' => $debug,
+] );
+
+if ( $cache ) {
+       if ( !isset( $wgObjectCaches[$cache] ) ) {
+               print "MediaWiki isn't configured with a cache named '$cache'";
+               exit( 1 );
+       }
+       $servers = $wgObjectCaches[$cache]['servers'];
+} elseif ( $wgMainCacheType === CACHE_MEMCACHED ) {
+       $mcc->set_servers( $wgMemCachedServers );
+} elseif ( isset( $wgObjectCaches[$wgMainCacheType]['servers'] ) ) {
+       $mcc->set_servers( $wgObjectCaches[$wgMainCacheType]['servers'] );
+} else {
+       print "MediaWiki isn't configured for Memcached usage\n";
+       exit( 1 );
+}
 
-               case 'set':
-                       print "set: changes something\n";
-               if($onlyone) { break; }
+/**
+ * Show this command line tool usage.
+ */
+function mccShowUsage() {
+       echo <<<EOF
+Usage:
+       mcc.php [--debug]
+       mcc.php --help
 
-               case 'delete':
-                       print "delete: deletes something\n";
-               if($onlyone) { break; }
+MemCached Command (mcc) is an interactive command tool that let you interact
+with the MediaWiki memcached cache.
 
-               case 'history':
-                       print "history: show command line history\n";
-               if($onlyone) { break; }
+Options:
+       --debug Set debug mode on the memcached connection.
+       --help  This help screen.
 
-               case 'server':
-                       print "server: show current memcached server\n";
-               if($onlyone) { break; }
+Interactive commands:
 
-               case 'dumpmcc':
-                       print "dumpmcc: shows the whole thing\n";
-               if($onlyone) { break; }
-
-               case 'exit':
-               case 'quit':
-                       print "exit or quit: exit mcc\n";
-               if($onlyone) { break; }
-
-               case 'help':
-                       print "help: help about a command\n";
-               if($onlyone) { break; }
+EOF;
+       print "\t";
+       print str_replace( "\n", "\n\t", mccGetHelp( false ) );
+       print "\n";
+}
 
-               default:
-                       if($onlyone) {
-                               print "$command: command does not exist or no help for it\n";
-                       }
+function mccGetHelp( $command ) {
+       $output = '';
+       $commandList = [
+               'get' => 'grabs something',
+               'getsock' => 'lists sockets',
+               'set' => 'changes something',
+               'delete' => 'deletes something',
+               'history' => 'show command line history',
+               'server' => 'show current memcached server',
+               'dumpmcc' => 'shows the whole thing',
+               'exit' => 'exit mcc',
+               'quit' => 'exit mcc',
+               'help' => 'help about a command',
+       ];
+       if ( !$command ) {
+               $command = 'fullhelp';
+       }
+       if ( $command === 'fullhelp' ) {
+               $max_cmd_len = max( array_map( 'strlen', array_keys( $commandList ) ) );
+               foreach ( $commandList as $cmd => $desc ) {
+                       $output .= sprintf( "%-{$max_cmd_len}s: %s\n", $cmd, $desc );
+               }
+       } elseif ( isset( $commandList[$command] ) ) {
+               $output .= "$command: $commandList[$command]\n";
+       } else {
+               $output .= "$command: command does not exist or no help for it\n";
        }
+
+       return $output;
 }
 
 do {
@@ -73,8 +116,10 @@ do {
        $showhelp = false;
        $quit = false;
 
-       $line = readconsole( '> ' );
-       if ($line === false) exit;
+       $line = Maintenance::readconsole();
+       if ( $line === false ) {
+               exit;
+       }
 
        $args = explode( ' ', $line );
        $command = array_shift( $args );
@@ -83,24 +128,28 @@ do {
        switch ( $command ) {
                case 'help':
                        // show an help message
-                       mccShowHelp(array_shift($args));
-               break;
+                       print mccGetHelp( array_shift( $args ) );
+                       break;
 
                case 'get':
-                       print "Getting {$args[0]}[{$args[1]}]\n";
+                       $sub = '';
+                       if ( array_key_exists( 1, $args ) ) {
+                               $sub = $args[1];
+                       }
+                       print "Getting {$args[0]}[$sub]\n";
                        $res = $mcc->get( $args[0] );
                        if ( array_key_exists( 1, $args ) ) {
                                $res = $res[$args[1]];
                        }
                        if ( $res === false ) {
-                               #print 'Error: ' . $mcc->error_string() . "\n";
+                               # print 'Error: ' . $mcc->error_string() . "\n";
                                print "MemCached error\n";
                        } elseif ( is_string( $res ) ) {
                                print "$res\n";
                        } else {
                                var_dump( $res );
                        }
-               break;
+                       break;
 
                case 'getsock':
                        $res = $mcc->get( $args[0] );
@@ -109,6 +158,10 @@ do {
                        break;
 
                case 'server':
+                       if ( $mcc->_single_sock !== null ) {
+                               print $mcc->_single_sock . "\n";
+                               break;
+                       }
                        $res = $mcc->get( $args[0] );
                        $hv = $mcc->_hashfunc( $args[0] );
                        for ( $i = 0; $i < 3; $i++ ) {
@@ -125,7 +178,7 @@ do {
                                $value = implode( ' ', $args );
                        }
                        if ( !$mcc->set( $key, $value, 0 ) ) {
-                               #print 'Error: ' . $mcc->error_string() . "\n";
+                               # print 'Error: ' . $mcc->error_string() . "\n";
                                print "MemCached error\n";
                        }
                        break;
@@ -133,14 +186,14 @@ do {
                case 'delete':
                        $key = implode( ' ', $args );
                        if ( !$mcc->delete( $key ) ) {
-                               #print 'Error: ' . $mcc->error_string() . "\n";
+                               # print 'Error: ' . $mcc->error_string() . "\n";
                                print "MemCached error\n";
                        }
                        break;
 
                case 'history':
                        if ( function_exists( 'readline_list_history' ) ) {
-                               foreach( readline_list_history() as $num => $line) {
+                               foreach ( readline_list_history() as $num => $line ) {
                                        print "$num: $line\n";
                                }
                        } else {
@@ -171,5 +224,3 @@ do {
                }
        }
 } while ( !$quit );
-
-