includes/db/LoadMonitor.php

   1 <?php
   2 /**
   3  * Database load monitoring.
   4  *
   5  * This program is free software; you can redistribute it and/or modify
   6  * it under the terms of the GNU General Public License as published by
   7  * the Free Software Foundation; either version 2 of the License, or
   8  * (at your option) any later version.
   9  *
  10  * This program is distributed in the hope that it will be useful,
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  13  * GNU General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU General Public License along
  16  * with this program; if not, write to the Free Software Foundation, Inc.,
  17  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
  18  * http://www.gnu.org/copyleft/gpl.html
  19  *
  20  * @file
  21  * @ingroup Database
  22  */
  23
  24 /**
  25  * An interface for database load monitoring
  26  *
  27  * @ingroup Database
  28  */
  29 interface LoadMonitor {
  30         /**
  31          * Construct a new LoadMonitor with a given LoadBalancer parent
  32          *
  33          * @param LoadBalancer $parent
  34          */
  35         public function __construct( $parent );
  36
  37         /**
  38          * Perform pre-connection load ratio adjustment.
  39          * @param array $loads
  40          * @param string|bool $group The selected query group. Default: false
  41          * @param string|bool $wiki Default: false
  42          */
  43         public function scaleLoads( &$loads, $group = false, $wiki = false );
  44
  45         /**
  46          * Return an estimate of replication lag for each server
  47          *
  48          * @param array $serverIndexes
  49          * @param string $wiki
  50          *
  51          * @return array
  52          */
  53         public function getLagTimes( $serverIndexes, $wiki );
  54 }
  55
  56 class LoadMonitorNull implements LoadMonitor {
  57         public function __construct( $parent ) {
  58         }
  59
  60         public function scaleLoads( &$loads, $group = false, $wiki = false ) {
  61         }
  62
  63         public function getLagTimes( $serverIndexes, $wiki ) {
  64                 return array_fill_keys( $serverIndexes, 0 );
  65         }
  66 }
  67
  68 /**
  69  * Basic MySQL load monitor with no external dependencies
  70  * Uses memcached to cache the replication lag for a short time
  71  *
  72  * @ingroup Database
  73  */
  74 class LoadMonitorMySQL implements LoadMonitor {
  75         /** @var LoadBalancer */
  76         public $parent;
  77
  78         public function __construct( $parent ) {
  79                 $this->parent = $parent;
  80         }
  81
  82         public function scaleLoads( &$loads, $group = false, $wiki = false ) {
  83         }
  84
  85         public function getLagTimes( $serverIndexes, $wiki ) {
  86                 if ( count( $serverIndexes ) == 1 && reset( $serverIndexes ) == 0 ) {
  87                         // Single server only, just return zero without caching
  88                         return array( 0 => 0 );
  89                 }
  90
  91                 $section = new ProfileSection( __METHOD__ );
  92
  93                 $expiry = 5;
  94                 $requestRate = 10;
  95
  96                 global $wgMemc;
  97                 if ( empty( $wgMemc ) ) {
  98                         $wgMemc = wfGetMainCache();
  99                 }
 100
 101                 $masterName = $this->parent->getServerName( 0 );
 102                 $memcKey = wfMemcKey( 'lag_times', $masterName );
 103                 $times = $wgMemc->get( $memcKey );
 104                 if ( is_array( $times ) ) {
 105                         # Randomly recache with probability rising over $expiry
 106                         $elapsed = time() - $times['timestamp'];
 107                         $chance = max( 0, ( $expiry - $elapsed ) * $requestRate );
 108                         if ( mt_rand( 0, $chance ) != 0 ) {
 109                                 unset( $times['timestamp'] ); // hide from caller
 110
 111                                 return $times;
 112                         }
 113                         wfIncrStats( 'lag_cache_miss_expired' );
 114                 } else {
 115                         wfIncrStats( 'lag_cache_miss_absent' );
 116                 }
 117
 118                 # Cache key missing or expired
 119                 if ( $wgMemc->add( "$memcKey:lock", 1, 10 ) ) {
 120                         # Let this process alone update the cache value
 121                         $unlocker = new ScopedCallback( function () use ( $wgMemc, $memcKey ) {
 122                                 $wgMemc->delete( $memcKey );
 123                         } );
 124                 } elseif ( is_array( $times ) ) {
 125                         # Could not acquire lock but an old cache exists, so use it
 126                         unset( $times['timestamp'] ); // hide from caller
 127
 128                         return $times;
 129                 }
 130
 131                 $times = array();
 132                 foreach ( $serverIndexes as $i ) {
 133                         if ( $i == 0 ) { # Master
 134                                 $times[$i] = 0;
 135                         } elseif ( false !== ( $conn = $this->parent->getAnyOpenConnection( $i ) ) ) {
 136                                 $times[$i] = $conn->getLag();
 137                         } elseif ( false !== ( $conn = $this->parent->openConnection( $i, $wiki ) ) ) {
 138                                 $times[$i] = $conn->getLag();
 139                         }
 140                 }
 141
 142                 # Add a timestamp key so we know when it was cached
 143                 $times['timestamp'] = time();
 144                 $wgMemc->set( $memcKey, $times, $expiry + 10 );
 145                 unset( $times['timestamp'] ); // hide from caller
 146
 147                 return $times;
 148         }
 149 }