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 = isset( $wgAlternateMaster['DEFAULT'] )
  25                 ? $wgAlternateMaster['DEFAULT']
  26                 : $wgDBserver;
  27         foreach ( $wgLocalDatabases as $db ) {
  28                 if ( isset( $wgAlternateMaster[$db] ) ) {
  29                         $dbsByMaster[$wgAlternateMaster[$db]][] = $db;
  30                 } else {
  31                         $dbsByMaster[$defaultMaster][] = $db;
  32                 }
  33         }
  34
  35         foreach ( $dbsByMaster as $master => $dbs ) {
  36                 $dbConn = new Database( $master, $wgDBuser, $wgDBpassword, $dbs[0] );
  37                 $stype = $dbConn->addQuotes($type);
  38
  39                 # Padding row for MySQL bug
  40                 $sql = "(SELECT '-------------------------------------------')";
  41                 foreach ( $dbs as $dbName ) {
  42                         if ( $sql != '' ) {
  43                                 $sql .= ' UNION ';
  44                         }
  45                         if ($type === false)
  46                                 $sql .= "(SELECT '$dbName' FROM `$dbName`.job LIMIT 1)";
  47                         else
  48                                 $sql .= "(SELECT '$dbName' FROM `$dbName`.job WHERE job_cmd=$stype LIMIT 1)";
  49                 }
  50                 $res = $dbConn->query( $sql, 'nextJobDB.php' );
  51                 $row = $dbConn->fetchRow( $res ); // discard padding row
  52                 while ( $row = $dbConn->fetchRow( $res ) ) {
  53                         $pendingDBs[] = $row[0];
  54                 }
  55         }
  56
  57         $wgMemc->set( $mckey, $pendingDBs, 300 );
  58 }
  59
  60 if ( $pendingDBs ) {
  61         echo $pendingDBs[mt_rand(0, count( $pendingDBs ) - 1)];
  62 }
  63
  64