* Use DB_ADMIN for fixSlaveDesync, rebuildtextindex, updateSearchIndex, patchSql

* Duplicate archive() as DatabaseBase::patchPath(), clean up patchSql to use this
This commit is contained in:
Chad Horohoe 2009-08-09 15:22:34 +00:00
parent 3a5a5abaa9
commit a82c01a4b2
5 changed files with 73 additions and 27 deletions

View file

@ -2075,6 +2075,23 @@ abstract class DatabaseBase {
return $error;
}
/**
* Get the full path of a patch file. Originally based on archive()
* from updaters.inc. Keep in mind this always returns a patch, as
* it fails back to MySQL if no DB-specific patch can be found
*
* @param $patch String The name of the patch, like patch-something.sql
* @return String Full path to patch file
*/
public static function patchPatch( $patch ) {
global $wgDBtype, $IP;
if ( file_exists( "$IP/maintenance/$wgDBtype/archives/$name" ) ) {
return "$IP/maintenance/$wgDBtype/archives/$name";
} else {
return "$IP/maintenance/archives/$name";
}
}
/**
* Read and execute commands from an open file handle
* Returns true on success, error string or exception on failure (depending on object's error ignore settings)

View file

@ -32,8 +32,6 @@ class FixSlaveDesync extends Maintenance {
public function execute() {
global $slaveIndexes, $wgDBservers;
$slaveIndexes = array();
for ( $i = 1; $i < count( $wgDBservers ); $i++ ) {
if ( wfGetLB()->isNonZeroLoad( $i ) ) {

View file

@ -3,34 +3,57 @@
* Manually run an SQL patch outside of the general updaters.
* This ensures that the DB options (charset, prefix, engine) are correctly set.
*
* @file
* 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
*
* @ingroup Maintenance
*/
require_once( dirname(__FILE__) . '/commandLine.inc' );
require_once "$IP/maintenance/updaters.inc";
require_once( dirname(__FILE__) . '/Maintenance.php' );
if( $args ) {
foreach( $args as $arg ) {
$files = array(
$arg,
archive( $arg ),
archive( "patch-$arg.sql" ),
);
foreach( $files as $file ) {
if( file_exists( $file ) ) {
echo "$file ...\n";
wfGetDB( DB_MASTER )->fileSource( $file );
continue 2;
}
}
echo "Could not find $arg\n";
class PatchSql extends Maintenance {
public function __construct() {
parent::__construct();
$this->mDescription = "Run an SQL file into the DB, replacing prefix and charset vars";
$this->addArgs( array( 'patch-name' ) );
}
protected function getDbType() {
return Maintenance::DB_ADMIN;
}
public function execute() {
$dbw = wfGetDB( DB_MASTER );
foreach( $this->mArgs as $arg ) {
$files = array(
$arg,
DatabaseBase::patchPatch( $arg ),
DatabaseBase::patchPatch( "patch-$arg.sql" ),
);
foreach( $files as $file ) {
if( file_exists( $file ) ) {
$this->output( "$file ...\n" );
$dbw->fileSource( $file );
continue 2;
}
}
$this->error( "Could not find $arg\n" );
}
$this->output( "done.\n" );
}
echo "done.\n";
} else {
echo "Run an SQL file into the DB, replacing prefix and charset vars.\n";
echo "Usage:\n";
echo " php maintenance/patchSql.php file1.sql file2.sql ...\n";
echo "\n";
echo "Paths in maintenance/archive are automatically expanded if a local file isn't found.\n";
}
$maintClass = "PatchSql";
require_once( DO_MAINTENANCE );

View file

@ -36,6 +36,10 @@ class RebuildTextIndex extends Maintenance {
$this->mDescription = "Rebuild search index table from scratch";
}
protected function getDbType() {
return Maintenance::DB_ADMIN;
}
public function execute() {
global $wgTitle;

View file

@ -40,6 +40,10 @@ class UpdateSearchIndex extends Maintenance {
$this->addOption( 'l', 'How long the searchindex and revision tables will be locked for', false, true );
}
protected function getDbType() {
return Maintenance::DB_ADMIN;
}
public function execute() {
$posFile = $this->getOption( 'p', 'searchUpdate.' . wfWikiId() . '.pos' );
$end = $this->getOption( 'e', wfTimestampNow() );