Thijs/wiki.techinc.nl
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Made nextJobDB.php respect $wgJobTypesExcludedFromDefaultQueue.

Browse source 
Change-Id: I22de34fc7be7b701180511fad3c16154ffdf638b
This commit is contained in:
Aaron Schulz 2013-01-30 19:27:53 -08:00 committed by Gerrit Code Review
parent d82616449f
commit 7875479757
1 changed files with 22 additions and 34 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

56
maintenance/nextJobDB.php
View file

@ -45,7 +45,7 @@ class nextJobDB extends Maintenance {
} elseif ( $this->hasOption( 'type' ) ) {
$types = array( $this->getOption( 'type' ) );
} else {
$types = false;
$types = JobQueueGroup::singleton()->getDefaultQueueTypes();
}
$memcKey = 'jobqueue:dbs:v3';
@ -82,24 +82,22 @@ class nextJobDB extends Maintenance {
do {
$again = false;
if ( $types === false ) {
$candidates = call_user_func_array( 'array_merge', $pendingDBs );
} else {
$candidates = array();
$possTypes = array_intersect( $types, array_keys( $pendingDBs ) );
if ( $possTypes ) {
$possTypes = array_values( $possTypes );
$type = $possTypes[ mt_rand( 0, count( $possTypes ) - 1 ) ];
$candidates = $pendingDBs[$type];
$candidates = array(); // list of (type, db)
// Flatten the tree of candidates into a flat list so that a random
// item can be selected, weighing each queue (type/db tuple) equally.
foreach ( $pendingDBs as $type => $dbs ) {
if ( in_array( $type, $types ) ) {
foreach ( $dbs as $db ) {
$candidates[] = array( $type, $db );
}
}
}
if ( !$candidates ) {
if ( !count( $candidates ) ) {
return; // no jobs for this type
}
$candidates = array_values( $candidates );
$db = $candidates[ mt_rand( 0, count( $candidates ) - 1 ) ];
if ( !$this->checkJob( $type, $db ) ) {
list( $type, $db ) = $candidates[ mt_rand( 0, count( $candidates ) - 1 ) ];
if ( !$this->checkJob( $type, $db ) ) { // queue is actually empty?
$pendingDBs = $this->delistDB( $pendingDBs, $db, $type );
// Update the cache to remove the outdated information.
// Make sure that this does not race (especially with full rebuilds).
@ -125,16 +123,16 @@ class nextJobDB extends Maintenance {
}
}
/**
* Remove a type/DB entry from the list of queues with jobs
*
* @param $pendingDBs array
* @param $db string
* @param $type string
* @return Array
*/
private function delistDB( array $pendingDBs, $db, $type ) {
if ( $type === false ) {
// There are no jobs available in the current database
foreach ( $pendingDBs as $type2 => $dbs ) {
$pendingDBs[$type2] = array_diff( $pendingDBs[$type2], array( $db ) );
}
} else {
// There are no jobs of this type available in the current database
$pendingDBs[$type] = array_diff( $pendingDBs[$type], array( $db ) );
}
$pendingDBs[$type] = array_diff( $pendingDBs[$type], array( $db ) );
return $pendingDBs;
}
@ -146,17 +144,7 @@ class nextJobDB extends Maintenance {
* @return bool
*/
private function checkJob( $type, $dbName ) {
$group = JobQueueGroup::singleton( $dbName );
if ( $type === false ) {
foreach ( $group->getDefaultQueueTypes() as $type ) {
if ( !$group->get( $type )->isEmpty() ) {
return true;
}
}
return false;
} else {
return !$group->get( $type )->isEmpty();
}
return !JobQueueGroup::singleton( $dbName )->get( $type )->isEmpty();
}
/**