maintenance/nextJobDB.php

   1 <?php
   2
   3 /*
   4  * Pick a database that has pending jobs
   5  */
   6
   7 $options = array( 'type'  );
   8
   9 require_once( 'commandLine.inc' );
  10
  11 $type = isset($options['type'])
  12                 ? $options['type']
  13                 : false;
  14
  15 $mckey = $type === false
  16             ? "jobqueue:dbs"
  17             : "jobqueue:dbs:$type";
  18
  19 $pendingDBs = $wgMemc->get( $mckey );
  20 if ( !$pendingDBs ) {
  21         $pendingDBs = array();
  22         # Cross-reference DBs by master DB server
  23         $dbsByMaster = array();
  24         $defaultMaster = $wgAlternateMaster['DEFAULT'];
  25         foreach ( $wgLocalDatabases as $db ) {
  26                 if ( isset( $wgAlternateMaster[$db] ) ) {
  27                         $dbsByMaster[$wgAlternateMaster[$db]][] = $db;
  28                 } else {
  29                         $dbsByMaster[$defaultMaster][] = $db;
  30                 }
  31         }
  32
  33         foreach ( $dbsByMaster as $master => $dbs ) {
  34                 $dbConn = new Database( $master, $wgDBuser, $wgDBpassword, $dbs[0] );
  35                 $stype = $dbConn->addQuotes($type);
  36
  37                 # Padding row for MySQL bug
  38                 $sql = "(SELECT '-------------------------------------------')";
  39                 foreach ( $dbs as $dbName ) {
  40                         if ( $sql != '' ) {
  41                                 $sql .= ' UNION ';
  42                         }
  43                         if ($type === false)
  44                                 $sql .= "(SELECT '$dbName' FROM `$dbName`.job LIMIT 1)";
  45                         else
  46                                 $sql .= "(SELECT '$dbName' FROM `$dbName`.job WHERE job_cmd=$stype LIMIT 1)";
  47                 }
  48                 $res = $dbConn->query( $sql, 'nextJobDB.php' );
  49                 $row = $dbConn->fetchRow( $res ); // discard padding row
  50                 while ( $row = $dbConn->fetchRow( $res ) ) {
  51                         $pendingDBs[] = $row[0];
  52                 }
  53         }
  54
  55         $wgMemc->set( $mckey, $pendingDBs, 300 );
  56 }
  57
  58 if ( $pendingDBs ) {
  59         echo $pendingDBs[mt_rand(0, count( $pendingDBs ) - 1)];
  60 }
  61
  62