6 * --maxjobs <num> (default 10000)
9 * This program is free software; you can redistribute it and/or modify
10 * it under the terms of the GNU General Public License as published by
11 * the Free Software Foundation; either version 2 of the License, or
12 * (at your option) any later version.
14 * This program is distributed in the hope that it will be useful,
15 * but WITHOUT ANY WARRANTY; without even the implied warranty of
16 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
17 * GNU General Public License for more details.
19 * You should have received a copy of the GNU General Public License along
20 * with this program; if not, write to the Free Software Foundation, Inc.,
21 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
22 * http://www.gnu.org/copyleft/gpl.html
25 * @ingroup Maintenance
28 require_once __DIR__
. '/Maintenance.php';
31 * Maintenance script that runs pending jobs.
33 * @ingroup Maintenance
35 class RunJobs
extends Maintenance
{
36 public function __construct() {
37 parent
::__construct();
38 $this->mDescription
= "Run pending jobs";
39 $this->addOption( 'maxjobs', 'Maximum number of jobs to run', false, true );
40 $this->addOption( 'maxtime', 'Maximum amount of wall-clock time', false, true );
41 $this->addOption( 'type', 'Type of job to run', false, true );
42 $this->addOption( 'procs', 'Number of processes to use', false, true );
45 public function memoryLimit() {
46 if ( $this->hasOption( 'memory-limit' ) ) {
47 return parent
::memoryLimit();
49 // Don't eat all memory on the machine if we get a bad job.
53 public function execute() {
57 $this->error( "Unable to run jobs; the wiki is in read-only mode.", 1 ); // die
60 if ( $this->hasOption( 'procs' ) ) {
61 $procs = intval( $this->getOption( 'procs' ) );
62 if ( $procs < 1 ||
$procs > 1000 ) {
63 $this->error( "Invalid argument to --procs", true );
64 } elseif ( $procs != 1 ) {
65 $fc = new ForkController( $procs );
66 if ( $fc->start() != 'child' ) {
71 $maxJobs = $this->getOption( 'maxjobs', false );
72 $maxTime = $this->getOption( 'maxtime', false );
74 $type = $this->getOption( 'type', false );
75 $wgTitle = Title
::newFromText( 'RunJobs.php' );
76 $dbw = wfGetDB( DB_MASTER
);
77 $jobsRun = 0; // counter
79 $group = JobQueueGroup
::singleton();
80 // Handle any required periodic queue maintenance
81 $count = $group->executeReadyPeriodicTasks();
83 $this->runJobsLog( "Executed $count periodic queue task(s)." );
86 $flags = JobQueueGroup
::USE_CACHE | JobQueueGroup
::USE_PRIORITY
;
87 $lastTime = time(); // time since last slave check
89 $job = ( $type === false )
90 ?
$group->pop( JobQueueGroup
::TYPE_DEFAULT
, $flags )
91 : $group->pop( $type ); // job from a single queue
92 if ( $job ) { // found a job
94 $this->runJobsLog( $job->toString() . " STARTING" );
96 // Set timer to stop the job if too much CPU time is used
97 set_time_limit( $maxTime ?
: 0 );
99 wfProfileIn( __METHOD__
. '-' . get_class( $job ) );
100 $t = microtime( true );
102 $status = $job->run();
103 $error = $job->getLastError();
104 } catch ( MWException
$e ) {
106 $error = get_class( $e ) . ': ' . $e->getMessage();
107 $e->report(); // write error to STDERR and the log
109 $timeMs = intval( ( microtime( true ) - $t ) * 1000 );
110 wfProfileOut( __METHOD__
. '-' . get_class( $job ) );
114 // Mark the job as done on success or when the job cannot be retried
115 if ( $status !== false ||
!$job->allowRetries() ) {
116 $group->ack( $job ); // done
119 if ( $status === false ) {
120 $this->runJobsLog( $job->toString() . " t=$timeMs error={$error}" );
122 $this->runJobsLog( $job->toString() . " t=$timeMs good" );
125 // Break out if we hit the job count or wall time limits...
126 if ( $maxJobs && $jobsRun >= $maxJobs ) {
128 } elseif ( $maxTime && ( time() - $startTime ) > $maxTime ) {
132 // Don't let any of the main DB slaves get backed up
133 $timePassed = time() - $lastTime;
134 if ( $timePassed >= 5 ||
$timePassed < 0 ) {
138 // Don't let any queue slaves/backups fall behind
139 if ( $jobsRun > 0 && ( $jobsRun %
100 ) == 0 ) {
140 $group->waitForBackups();
143 // Bail if near-OOM instead of in a job
144 $this->assertMemoryOK();
146 } while ( $job ); // stop when there are no jobs
150 * Make sure that this script is not too close to the memory usage limit
151 * @throws MWException
153 private function assertMemoryOK() {
154 static $maxBytes = null;
155 if ( $maxBytes === null ) {
157 if ( preg_match( '!^(\d+)(k|m|g|)$!i', ini_get( 'memory_limit' ), $m ) ) {
158 list( , $num, $unit ) = $m;
159 $conv = array( 'g' => 1024*1024*1024, 'm' => 1024*1024, 'k' => 1024, '' => 1 );
160 $maxBytes = $num * $conv[strtolower( $unit )];
165 $usedBytes = memory_get_usage();
166 if ( $maxBytes && $usedBytes >= .95*$maxBytes ) {
167 throw new MWException( "Detected excessive memory usage ($usedBytes/$maxBytes)." );
172 * Log the job message
173 * @param $msg String The message to log
175 private function runJobsLog( $msg ) {
176 $this->output( wfTimestamp( TS_DB
) . " $msg\n" );
177 wfDebugLog( 'runJobs', $msg );
181 $maintClass = "RunJobs";
182 require_once RUN_MAINTENANCE_IF_MAIN
;