includes/SiteStatsUpdate.php

   1 <?php
   2 /**
   3  * See deferred.txt
   4  *
   5  * @package MediaWiki
   6  */
   7
   8 /**
   9  *
  10  * @package MediaWiki
  11  */
  12 class SiteStatsUpdate {
  13
  14         var $mViews, $mEdits, $mGood, $mPages, $mUsers;
  15
  16         function SiteStatsUpdate( $views, $edits, $good, $pages = 0, $users = 0 ) {
  17                 $this->mViews = $views;
  18                 $this->mEdits = $edits;
  19                 $this->mGood = $good;
  20                 $this->mPages = $pages;
  21                 $this->mUsers = $users;
  22         }
  23
  24         function appendUpdate( &$sql, $field, $delta ) {
  25                 if ( $delta ) {
  26                         if ( $sql ) {
  27                                 $sql .= ',';
  28                         }
  29                         if ( $delta < 0 ) {
  30                                 $sql .= "$field=$field-1";
  31                         } else {
  32                                 $sql .= "$field=$field+1";
  33                         }
  34                 }
  35         }
  36
  37         function doUpdate() {
  38                 global $wgDBname;
  39                 $fname = 'SiteStatsUpdate::doUpdate';
  40                 $dbw =& wfGetDB( DB_MASTER );
  41
  42                 # First retrieve the row just to find out which schema we're in
  43                 $row = $dbw->selectRow( 'site_stats', '*', false, $fname );
  44
  45                 $updates = '';
  46
  47                 $this->appendUpdate( $updates, 'ss_total_views', $this->mViews );
  48                 $this->appendUpdate( $updates, 'ss_total_edits', $this->mEdits );
  49                 $this->appendUpdate( $updates, 'ss_good_articles', $this->mGood );
  50
  51                 if ( isset( $row->ss_total_pages ) ) {
  52                         # Update schema if required
  53                         if ( $row->ss_total_pages == -1 && !$this->mViews ) {
  54                                 $dbr =& wfGetDB( DB_SLAVE, array( 'SpecialStatistics', 'vslow') );
  55                                 extract( $dbr->tableNames( 'page', 'user' ) );
  56
  57                                 $sql = "SELECT COUNT(page_namespace) AS total FROM $page";
  58                                 $res = $dbr->query( $sql, $fname );
  59                                 $pageRow = $dbr->fetchObject( $res );
  60                                 $pages = $pageRow->total + $this->mPages;
  61
  62                                 $sql = "SELECT COUNT(user_id) AS total FROM $user";
  63                                 $res = $dbr->query( $sql, $fname );
  64                                 $userRow = $dbr->fetchObject( $res );
  65                                 $users = $userRow->total + $this->mUsers;
  66
  67                                 if ( $updates ) {
  68                                         $updates .= ',';
  69                                 }
  70                                 $updates .= "ss_total_pages=$pages, ss_users=$users";
  71                         } else {
  72                                 $this->appendUpdate( $updates, 'ss_total_pages', $this->mPages );
  73                                 $this->appendUpdate( $updates, 'ss_users', $this->mUsers );
  74                         }
  75                 }
  76                 if ( $updates ) {
  77                         $site_stats = $dbw->tableName( 'site_stats' );
  78                         $sql = "UPDATE $site_stats SET $updates LIMIT 1";
  79                         $dbw->query( $sql, $fname );
  80                 }
  81         }
  82 }
  83 ?>