maintenance/runBatchedQuery.php

   1 <?php
   2 /**
   3  * Run a database query in batches and wait for slaves. This is used on large
   4  * wikis to prevent replication lag from going through the roof when executing
   5  * large write queries.
   6  *
   7  * This program is free software; you can redistribute it and/or modify
   8  * it under the terms of the GNU General Public License as published by
   9  * the Free Software Foundation; either version 2 of the License, or
  10  * (at your option) any later version.
  11  *
  12  * This program is distributed in the hope that it will be useful,
  13  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  14  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  15  * GNU General Public License for more details.
  16  *
  17  * You should have received a copy of the GNU General Public License along
  18  * with this program; if not, write to the Free Software Foundation, Inc.,
  19  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
  20  * http://www.gnu.org/copyleft/gpl.html
  21  *
  22  * @file
  23  * @ingroup Maintenance
  24  */
  25
  26 require_once( dirname( __FILE__ ) . '/Maintenance.php' );
  27
  28 /**
  29  * Maintenance script to run a database query in batches and wait for slaves.
  30  *
  31  * @ingroup Maintenance
  32  */
  33 class BatchedQueryRunner extends Maintenance {
  34         public function __construct() {
  35                 parent::__construct();
  36                 $this->mDescription = "Run a query repeatedly until it affects 0 rows, and wait for slaves in between.\n" .
  37                                 "NOTE: You need to set a LIMIT clause yourself.";
  38         }
  39
  40         public function execute() {
  41                 if ( !$this->hasArg() )
  42                         $this->error( "No query specified. Specify the query as a command line parameter.", true );
  43
  44                 $query = $this->getArg();
  45                 $n = 1;
  46                 $dbw = wfGetDB( DB_MASTER );
  47                 do {
  48                         $this->output( "Batch $n: " );
  49                         $n++;
  50                         $dbw->query( $query, __METHOD__ );
  51                         $affected = $dbw->affectedRows();
  52                         $this->output( "$affected rows\n" );
  53                         wfWaitForSlaves();
  54                 } while ( $affected > 0 );
  55         }
  56
  57         public function getDbType() {
  58                 return Maintenance::DB_ADMIN;
  59         }
  60 }
  61
  62
  63 $maintClass = "BatchedQueryRunner";
  64 require_once( RUN_MAINTENANCE_IF_MAIN );