includes/db/LoadMonitor.php

   1 <?php
   2
   3 /**
   4  * An interface for database load monitoring
   5  */
   6
   7 interface LoadMonitor {
   8         /**
   9          * Construct a new LoadMonitor with a given LoadBalancer parent
  10          */
  11         function __construct( $parent );
  12
  13         /**
  14          * Perform pre-connection load ratio adjustment.
  15          * @param array $loads
  16          * @param string $group The selected query group
  17          * @param string $wiki
  18          */
  19         function scaleLoads( &$loads, $group = false, $wiki = false );
  20
  21         /**
  22          * Perform post-connection backoff.
  23          *
  24          * If the connection is in overload, this should return a backoff factor
  25          * which will be used to control polling time. The number of threads
  26          * connected is a good measure.
  27          *
  28          * If there is no overload, zero can be returned.
  29          *
  30          * A threshold thread count is given, the concrete class may compare this
  31          * to the running thread count. The threshold may be false, which indicates
  32          * that the sysadmin has not configured this feature.
  33          *
  34          * @param Database $conn
  35          * @param float $threshold
  36          */
  37         function postConnectionBackoff( $conn, $threshold );
  38
  39         /**
  40          * Return an estimate of replication lag for each server
  41          */
  42         function getLagTimes( $serverIndexes, $wiki );
  43 }
  44
  45
  46 /**
  47  * Basic MySQL load monitor with no external dependencies
  48  * Uses memcached to cache the replication lag for a short time
  49  */
  50
  51 class LoadMonitor_MySQL implements LoadMonitor {
  52         var $parent; // LoadBalancer
  53
  54         function __construct( $parent ) {
  55                 $this->parent = $parent;
  56         }
  57
  58         function scaleLoads( &$loads, $group = false, $wiki = false ) {
  59         }
  60
  61         function getLagTimes( $serverIndexes, $wiki ) {
  62                 wfProfileIn( __METHOD__ );
  63                 $expiry = 5;
  64                 $requestRate = 10;
  65
  66                 global $wgMemc;
  67                 $masterName = $this->parent->getServerName( 0 );
  68                 $memcKey = wfMemcKey( 'lag_times', $masterName );
  69                 $times = $wgMemc->get( $memcKey );
  70                 if ( $times ) {
  71                         # Randomly recache with probability rising over $expiry
  72                         $elapsed = time() - $times['timestamp'];
  73                         $chance = max( 0, ( $expiry - $elapsed ) * $requestRate );
  74                         if ( mt_rand( 0, $chance ) != 0 ) {
  75                                 unset( $times['timestamp'] );
  76                                 wfProfileOut( __METHOD__ );
  77                                 return $times;
  78                         }
  79                         wfIncrStats( 'lag_cache_miss_expired' );
  80                 } else {
  81                         wfIncrStats( 'lag_cache_miss_absent' );
  82                 }
  83
  84                 # Cache key missing or expired
  85
  86                 $times = array();
  87                 foreach ( $serverIndexes as $i ) {
  88                         if ($i == 0) { # Master
  89                                 $times[$i] = 0;
  90                         } elseif ( false !== ( $conn = $this->parent->getAnyOpenConnection( $i ) ) ) {
  91                                 $times[$i] = $conn->getLag();
  92                         } elseif ( false !== ( $conn = $this->parent->openConnection( $i, $wiki ) ) ) {
  93                                 $times[$i] = $conn->getLag();
  94                         }
  95                 }
  96
  97                 # Add a timestamp key so we know when it was cached
  98                 $times['timestamp'] = time();
  99                 $wgMemc->set( $memcKey, $times, $expiry );
 100
 101                 # But don't give the timestamp to the caller
 102                 unset($times['timestamp']);
 103                 $lagTimes = $times;
 104
 105                 wfProfileOut( __METHOD__ );
 106                 return $lagTimes;
 107         }
 108
 109         function postConnectionBackoff( $conn, $threshold ) {
 110                 if ( !$threshold ) {
 111                         return 0;
 112                 }
 113                 $status = $conn->getStatus("Thread%");
 114                 if ( $status['Threads_running'] > $threshold ) {
 115                         return $status['Threads_connected'];
 116                 } else {
 117                         return 0;
 118                 }
 119         }
 120 }
 121