62 lines
1.5 KiB
PHP
62 lines
1.5 KiB
PHP
<?php
|
|
|
|
/*
|
|
* Pick a database that has pending jobs
|
|
*/
|
|
|
|
$options = array( 'type' );
|
|
|
|
require_once( 'commandLine.inc' );
|
|
|
|
$type = isset($options['type'])
|
|
? $options['type']
|
|
: false;
|
|
|
|
$mckey = $type === false
|
|
? "jobqueue:dbs"
|
|
: "jobqueue:dbs:$type";
|
|
|
|
$pendingDBs = $wgMemc->get( $mckey );
|
|
if ( !$pendingDBs ) {
|
|
$pendingDBs = array();
|
|
# Cross-reference DBs by master DB server
|
|
$dbsByMaster = array();
|
|
$defaultMaster = $wgAlternateMaster['DEFAULT'];
|
|
foreach ( $wgLocalDatabases as $db ) {
|
|
if ( isset( $wgAlternateMaster[$db] ) ) {
|
|
$dbsByMaster[$wgAlternateMaster[$db]][] = $db;
|
|
} else {
|
|
$dbsByMaster[$defaultMaster][] = $db;
|
|
}
|
|
}
|
|
|
|
foreach ( $dbsByMaster as $master => $dbs ) {
|
|
$dbConn = new Database( $master, $wgDBuser, $wgDBpassword );
|
|
$stype = $dbConn->addQuotes($type);
|
|
|
|
# Padding row for MySQL bug
|
|
$sql = "(SELECT '-------------------------------------------')";
|
|
foreach ( $dbs as $dbName ) {
|
|
if ( $sql != '' ) {
|
|
$sql .= ' UNION ';
|
|
}
|
|
if ($type === false)
|
|
$sql .= "(SELECT '$dbName' FROM `$dbName`.job LIMIT 1)";
|
|
else
|
|
$sql .= "(SELECT '$dbName' FROM `$dbName`.job WHERE job_cmd=$stype LIMIT 1)";
|
|
}
|
|
$res = $dbConn->query( $sql, 'nextJobDB.php' );
|
|
$row = $dbConn->fetchRow( $res ); // discard padding row
|
|
while ( $row = $dbConn->fetchRow( $res ) ) {
|
|
$pendingDBs[] = $row[0];
|
|
}
|
|
}
|
|
|
|
$wgMemc->set( $mckey, $pendingDBs, 300 );
|
|
}
|
|
|
|
if ( $pendingDBs ) {
|
|
echo $pendingDBs[mt_rand(0, count( $pendingDBs ) - 1)];
|
|
}
|
|
|
|
?>
|