includes/job/jobs/RefreshLinksJob.php

   1 <?php
   2 /**
   3  * Job to update links for a given title.
   4  *
   5  * This program is free software; you can redistribute it and/or modify
   6  * it under the terms of the GNU General Public License as published by
   7  * the Free Software Foundation; either version 2 of the License, or
   8  * (at your option) any later version.
   9  *
  10  * This program is distributed in the hope that it will be useful,
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  13  * GNU General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU General Public License along
  16  * with this program; if not, write to the Free Software Foundation, Inc.,
  17  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
  18  * http://www.gnu.org/copyleft/gpl.html
  19  *
  20  * @file
  21  * @ingroup JobQueue
  22  */
  23
  24 /**
  25  * Background job to update links for a given title.
  26  *
  27  * @ingroup JobQueue
  28  */
  29 class RefreshLinksJob extends Job {
  30         function __construct( $title, $params = '', $id = 0 ) {
  31                 parent::__construct( 'refreshLinks', $title, $params, $id );
  32                 $this->removeDuplicates = true; // job is expensive
  33         }
  34
  35         /**
  36          * Run a refreshLinks job
  37          * @return boolean success
  38          */
  39         function run() {
  40                 wfProfileIn( __METHOD__ );
  41
  42                 $linkCache = LinkCache::singleton();
  43                 $linkCache->clear();
  44
  45                 if ( is_null( $this->title ) ) {
  46                         $this->error = "refreshLinks: Invalid title";
  47                         wfProfileOut( __METHOD__ );
  48                         return false;
  49                 }
  50
  51                 # Wait for the DB of the current/next slave DB handle to catch up to the master.
  52                 # This way, we get the correct page_latest for templates or files that just changed
  53                 # milliseconds ago, having triggered this job to begin with.
  54                 if ( isset( $this->params['masterPos'] ) ) {
  55                         wfGetLB()->waitFor( $this->params['masterPos'] );
  56                 }
  57
  58                 $revision = Revision::newFromTitle( $this->title, false, Revision::READ_NORMAL );
  59                 if ( !$revision ) {
  60                         $this->error = 'refreshLinks: Article not found "' .
  61                                 $this->title->getPrefixedDBkey() . '"';
  62                         wfProfileOut( __METHOD__ );
  63                         return false; // XXX: what if it was just deleted?
  64                 }
  65
  66                 self::runForTitleInternal( $this->title, $revision, __METHOD__ );
  67
  68                 wfProfileOut( __METHOD__ );
  69                 return true;
  70         }
  71
  72         public static function runForTitleInternal( Title $title, Revision $revision, $fname ) {
  73                 wfProfileIn( $fname );
  74                 $content = $revision->getContent( Revision::RAW );
  75
  76                 if ( !$content ) {
  77                         // if there is no content, pretend the content is empty
  78                         $content = $revision->getContentHandler()->makeEmptyContent();
  79                 }
  80
  81                 $updates = $content->getSecondaryDataUpdates( $title, null, false );
  82                 DataUpdate::runUpdates( $updates );
  83                 wfProfileOut( $fname );
  84         }
  85 }
  86
  87 /**
  88  * Background job to update links for a given title.
  89  * Newer version for high use templates.
  90  *
  91  * @ingroup JobQueue
  92  */
  93 class RefreshLinksJob2 extends Job {
  94         const MAX_TITLES_RUN = 10;
  95
  96         function __construct( $title, $params, $id = 0 ) {
  97                 parent::__construct( 'refreshLinks2', $title, $params, $id );
  98         }
  99
 100         /**
 101          * Run a refreshLinks2 job
 102          * @return boolean success
 103          */
 104         function run() {
 105                 wfProfileIn( __METHOD__ );
 106
 107                 $linkCache = LinkCache::singleton();
 108                 $linkCache->clear();
 109
 110                 if ( is_null( $this->title ) ) {
 111                         $this->error = "refreshLinks2: Invalid title";
 112                         wfProfileOut( __METHOD__ );
 113                         return false;
 114                 } elseif ( !isset( $this->params['start'] ) || !isset( $this->params['end'] ) ) {
 115                         $this->error = "refreshLinks2: Invalid params";
 116                         wfProfileOut( __METHOD__ );
 117                         return false;
 118                 }
 119
 120                 // Back compat for pre-r94435 jobs
 121                 $table = isset( $this->params['table'] ) ? $this->params['table'] : 'templatelinks';
 122
 123                 // Avoid slave lag when fetching templates
 124                 if ( isset( $this->params['masterPos'] ) ) {
 125                         $masterPos = $this->params['masterPos'];
 126                 } elseif ( wfGetLB()->getServerCount() > 1  ) {
 127                         $masterPos = wfGetLB()->getMasterPos();
 128                 } else {
 129                         $masterPos = false;
 130                 }
 131
 132                 $titles = $this->title->getBacklinkCache()->getLinks(
 133                         $table, $this->params['start'], $this->params['end'] );
 134
 135                 if ( $titles->count() > self::MAX_TITLES_RUN ) {
 136                         # We don't want to parse too many pages per job as it can starve other jobs.
 137                         # If there are too many pages to parse, break this up into smaller jobs. By passing
 138                         # in the master position here we can cut down on the time spent waiting for slaves to
 139                         # catch up by the runners handling these jobs since time will have passed between now
 140                         # and when they pop these jobs off the queue.
 141                         $start = 0; // batch start
 142                         $end   = 0; // batch end
 143                         $bsize = 0; // batch size
 144                         $first = true; // first of batch
 145                         $jobs  = array();
 146                         foreach ( $titles as $title ) {
 147                                 $start = $first ? $title->getArticleId() : $start;
 148                                 $end   = $title->getArticleId();
 149                                 $first = false;
 150                                 if ( ++$bsize >= self::MAX_TITLES_RUN ) {
 151                                         $jobs[] = new RefreshLinksJob2( $this->title, array(
 152                                                 'table'     => $table,
 153                                                 'start'     => $start,
 154                                                 'end'       => $end,
 155                                                 'masterPos' => $masterPos
 156                                         ) );
 157                                         $first = true;
 158                                         $start = $end = $bsize = 0;
 159                                 }
 160                         }
 161                         if ( $bsize > 0 ) { // group remaining pages into a job
 162                                 $jobs[] = new RefreshLinksJob2( $this->title, array(
 163                                         'table'     => $table,
 164                                         'start'     => $start,
 165                                         'end'       => $end,
 166                                         'masterPos' => $masterPos
 167                                 ) );
 168                         }
 169                         Job::batchInsert( $jobs );
 170                 } elseif ( php_sapi_name() != 'cli' ) {
 171                         # Not suitable for page load triggered job running!
 172                         # Gracefully switch to refreshLinks jobs if this happens.
 173                         $jobs = array();
 174                         foreach ( $titles as $title ) {
 175                                 $jobs[] = new RefreshLinksJob( $title, array( 'masterPos' => $masterPos ) );
 176                         }
 177                         Job::batchInsert( $jobs );
 178                 } else {
 179                         # Wait for the DB of the current/next slave DB handle to catch up to the master.
 180                         # This way, we get the correct page_latest for templates or files that just changed
 181                         # milliseconds ago, having triggered this job to begin with.
 182                         if ( $masterPos ) {
 183                                 wfGetLB()->waitFor( $masterPos );
 184                         }
 185                         # Re-parse each page that transcludes this page and update their tracking links...
 186                         foreach ( $titles as $title ) {
 187                                 $revision = Revision::newFromTitle( $title, false, Revision::READ_NORMAL );
 188                                 if ( !$revision ) {
 189                                         $this->error = 'refreshLinks: Article not found "' .
 190                                                 $title->getPrefixedDBkey() . '"';
 191                                         continue; // skip this page
 192                                 }
 193                                 RefreshLinksJob::runForTitleInternal( $title, $revision, __METHOD__ );
 194                                 wfWaitForSlaves();
 195                         }
 196                 }
 197
 198                 wfProfileOut( __METHOD__ );
 199                 return true;
 200         }
 201 }