X-Git-Url: http://git.heureux-cyclage.org/?a=blobdiff_plain;f=maintenance%2Fsql.php;h=e6a30a38fe5a087ee28c8fe07e898b7ee7b605d9;hb=5820c6dcacb7c144b5ee8c292d12289ac1875df6;hp=a93e51fe5ace013278f008702b46dac388a35c18;hpb=012248fc5a7d6d0257df1a0feadbda5ae4a477e2;p=lhc%2Fweb%2Fwiklou.git diff --git a/maintenance/sql.php b/maintenance/sql.php index a93e51fe5a..e6a30a38fe 100644 --- a/maintenance/sql.php +++ b/maintenance/sql.php @@ -32,12 +32,15 @@ require_once __DIR__ . '/Maintenance.php'; class MwSql extends Maintenance { public function __construct() { parent::__construct(); - $this->mDescription = "Send SQL queries to a MediaWiki database. " . - "Takes a file name containing SQL as argument or runs interactively."; - $this->addOption( 'query', 'Run a single query instead of running interactively', false, true ); + $this->addDescription( 'Send SQL queries to a MediaWiki database. ' . + 'Takes a file name containing SQL as argument or runs interactively.' ); + $this->addOption( 'query', + 'Run a single query instead of running interactively', false, true ); $this->addOption( 'cluster', 'Use an external cluster by name', false, true ); - $this->addOption( 'wikidb', 'The database wiki ID to use if not the current one', false, true ); - $this->addOption( 'slave', 'Use a slave server (either "any" or by name)', false, true ); + $this->addOption( 'wikidb', + 'The database wiki ID to use if not the current one', false, true ); + $this->addOption( 'replicadb', + 'Replica DB server to use instead of the master DB (can be "any")', false, true ); } public function execute() { @@ -50,30 +53,28 @@ class MwSql extends Maintenance { $lb = wfGetLB( $wiki ); } // Figure out which server to use - if ( $this->hasOption( 'slave' ) ) { - $server = $this->getOption( 'slave' ); - if ( $server === 'any' ) { - $index = DB_SLAVE; - } else { - $index = null; - $serverCount = $lb->getServerCount(); - for ( $i = 0; $i < $serverCount; ++$i ) { - if ( $lb->getServerName( $i ) === $server ) { - $index = $i; - break; - } - } - if ( $index === null ) { - $this->error( "No slave server configured with the name '$server'.", 1 ); + $replicaDB = $this->getOption( 'replicadb', $this->getOption( 'slave', '' ) ); + if ( $replicaDB === 'any' ) { + $index = DB_REPLICA; + } elseif ( $replicaDB != '' ) { + $index = null; + $serverCount = $lb->getServerCount(); + for ( $i = 0; $i < $serverCount; ++$i ) { + if ( $lb->getServerName( $i ) === $replicaDB ) { + $index = $i; + break; } } + if ( $index === null ) { + $this->error( "No replica DB server configured with the name '$server'.", 1 ); + } } else { $index = DB_MASTER; } // Get a DB handle (with this wiki's DB selected) from the appropriate load balancer - $db = $lb->getConnection( $index, array(), $wiki ); - if ( $this->hasOption( 'slave' ) && $db->getLBInfo( 'master' ) !== null ) { - $this->error( "The server selected ({$db->getServer()}) is not a slave.", 1 ); + $db = $lb->getConnection( $index, [], $wiki ); + if ( $replicaDB != '' && $db->getLBInfo( 'master' ) !== null ) { + $this->error( "The server selected ({$db->getServer()}) is not a replica DB.", 1 ); } if ( $this->hasArg( 0 ) ) { @@ -82,7 +83,7 @@ class MwSql extends Maintenance { $this->error( "Unable to open input file", true ); } - $error = $db->sourceStream( $file, false, array( $this, 'sqlPrintResult' ) ); + $error = $db->sourceStream( $file, false, [ $this, 'sqlPrintResult' ] ); if ( $error !== true ) { $this->error( $error, true ); } else {