maintenance/updateSpecialPages.php

   1 <?php
   2 /**
   3  * Update for cached special pages.
   4  * Run this script periodically if you have miser mode enabled.
   5  *
   6  * This program is free software; you can redistribute it and/or modify
   7  * it under the terms of the GNU General Public License as published by
   8  * the Free Software Foundation; either version 2 of the License, or
   9  * (at your option) any later version.
  10  *
  11  * This program is distributed in the hope that it will be useful,
  12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  14  * GNU General Public License for more details.
  15  *
  16  * You should have received a copy of the GNU General Public License along
  17  * with this program; if not, write to the Free Software Foundation, Inc.,
  18  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
  19  * http://www.gnu.org/copyleft/gpl.html
  20  *
  21  * @file
  22  * @ingroup Maintenance
  23  */
  24
  25 require_once __DIR__ . '/Maintenance.php';
  26
  27 use MediaWiki\MediaWikiServices;
  28 use Wikimedia\Rdbms\DBReplicationWaitError;
  29
  30 /**
  31  * Maintenance script to update cached special pages.
  32  *
  33  * @ingroup Maintenance
  34  */
  35 class UpdateSpecialPages extends Maintenance {
  36         public function __construct() {
  37                 parent::__construct();
  38                 $this->addOption( 'list', 'List special page names' );
  39                 $this->addOption( 'only', 'Only update "page"; case sensitive, ' .
  40                         'check correct case by calling this script with --list. ' .
  41                         'Ex: --only=BrokenRedirects', false, true );
  42                 $this->addOption( 'override', 'Also update pages that have updates disabled' );
  43         }
  44
  45         public function execute() {
  46                 global $wgQueryCacheLimit, $wgDisableQueryPageUpdate;
  47
  48                 $dbw = $this->getDB( DB_MASTER );
  49
  50                 $this->doSpecialPageCacheUpdates( $dbw );
  51
  52                 foreach ( QueryPage::getPages() as $page ) {
  53                         list( $class, $special ) = $page;
  54                         $limit = isset( $page[2] ) ? $page[2] : null;
  55
  56                         # --list : just show the name of pages
  57                         if ( $this->hasOption( 'list' ) ) {
  58                                 $this->output( "$special [QueryPage]\n" );
  59                                 continue;
  60                         }
  61
  62                         if ( !$this->hasOption( 'override' )
  63                                 && $wgDisableQueryPageUpdate && in_array( $special, $wgDisableQueryPageUpdate )
  64                         ) {
  65                                 $this->output( sprintf( "%-30s [QueryPage] disabled\n", $special ) );
  66                                 continue;
  67                         }
  68
  69                         $specialObj = SpecialPageFactory::getPage( $special );
  70                         if ( !$specialObj ) {
  71                                 $this->output( "No such special page: $special\n" );
  72                                 exit;
  73                         }
  74                         if ( $specialObj instanceof QueryPage ) {
  75                                 $queryPage = $specialObj;
  76                         } else {
  77                                 $class = get_class( $specialObj );
  78                                 $this->fatalError( "$class is not an instance of QueryPage.\n" );
  79                                 die;
  80                         }
  81
  82                         if ( !$this->hasOption( 'only' ) || $this->getOption( 'only' ) == $queryPage->getName() ) {
  83                                 $this->output( sprintf( '%-30s [QueryPage] ', $special ) );
  84                                 if ( $queryPage->isExpensive() ) {
  85                                         $t1 = microtime( true );
  86                                         # Do the query
  87                                         $num = $queryPage->recache( $limit === null ? $wgQueryCacheLimit : $limit );
  88                                         $t2 = microtime( true );
  89                                         if ( $num === false ) {
  90                                                 $this->output( "FAILED: database error\n" );
  91                                         } else {
  92                                                 $this->output( "got $num rows in " );
  93
  94                                                 $elapsed = $t2 - $t1;
  95                                                 $hours = intval( $elapsed / 3600 );
  96                                                 $minutes = intval( $elapsed % 3600 / 60 );
  97                                                 $seconds = $elapsed - $hours * 3600 - $minutes * 60;
  98                                                 if ( $hours ) {
  99                                                         $this->output( $hours . 'h ' );
 100                                                 }
 101                                                 if ( $minutes ) {
 102                                                         $this->output( $minutes . 'm ' );
 103                                                 }
 104                                                 $this->output( sprintf( "%.2fs\n", $seconds ) );
 105                                         }
 106                                         # Reopen any connections that have closed
 107                                         $this->reopenAndWaitForReplicas();
 108                                 } else {
 109                                         $this->output( "cheap, skipped\n" );
 110                                 }
 111                                 if ( $this->hasOption( 'only' ) ) {
 112                                         break;
 113                                 }
 114                         }
 115                 }
 116         }
 117
 118         /**
 119          * Re-open any closed db connection, and wait for replicas
 120          *
 121          * Queries that take a really long time, might cause the
 122          * mysql connection to "go away"
 123          */
 124         private function reopenAndWaitForReplicas() {
 125                 $lbFactory = MediaWikiServices::getInstance()->getDBLoadBalancerFactory();
 126                 $lb = $lbFactory->getMainLB();
 127                 if ( !$lb->pingAll() ) {
 128                         $this->output( "\n" );
 129                         do {
 130                                 $this->error( "Connection failed, reconnecting in 10 seconds..." );
 131                                 sleep( 10 );
 132                         } while ( !$lb->pingAll() );
 133                         $this->output( "Reconnected\n\n" );
 134                 }
 135                 // Wait for the replica DB to catch up
 136                 try {
 137                         $lbFactory->waitForReplication();
 138                 } catch ( DBReplicationWaitError $e ) {
 139                         // ignore
 140                 }
 141         }
 142
 143         public function doSpecialPageCacheUpdates( $dbw ) {
 144                 global $wgSpecialPageCacheUpdates;
 145
 146                 foreach ( $wgSpecialPageCacheUpdates as $special => $call ) {
 147                         # --list : just show the name of pages
 148                         if ( $this->hasOption( 'list' ) ) {
 149                                 $this->output( "$special [callback]\n" );
 150                                 continue;
 151                         }
 152
 153                         if ( !$this->hasOption( 'only' ) || $this->getOption( 'only' ) == $special ) {
 154                                 if ( !is_callable( $call ) ) {
 155                                         $this->error( "Uncallable function $call!" );
 156                                         continue;
 157                                 }
 158                                 $this->output( sprintf( '%-30s [callback] ', $special ) );
 159                                 $t1 = microtime( true );
 160                                 call_user_func( $call, $dbw );
 161                                 $t2 = microtime( true );
 162
 163                                 $this->output( "completed in " );
 164                                 $elapsed = $t2 - $t1;
 165                                 $hours = intval( $elapsed / 3600 );
 166                                 $minutes = intval( $elapsed % 3600 / 60 );
 167                                 $seconds = $elapsed - $hours * 3600 - $minutes * 60;
 168                                 if ( $hours ) {
 169                                         $this->output( $hours . 'h ' );
 170                                 }
 171                                 if ( $minutes ) {
 172                                         $this->output( $minutes . 'm ' );
 173                                 }
 174                                 $this->output( sprintf( "%.2fs\n", $seconds ) );
 175                                 # Wait for the replica DB to catch up
 176                                 $this->reopenAndWaitForReplicas();
 177                         }
 178                 }
 179         }
 180 }
 181
 182 $maintClass = UpdateSpecialPages::class;
 183 require_once RUN_MAINTENANCE_IF_MAIN;