X-Git-Url: http://git.heureux-cyclage.org/?a=blobdiff_plain;f=includes%2FSiteStatsUpdate.php;h=1b6d3804e360044b86eb0c7b169fb637d69b2f9d;hb=09da7fb96c45255899fa0f4c62412d14b8575b3e;hp=f44a3ed9acb479f1ae44e8a5054624b320a3d29f;hpb=ba2afcd9fa9b1f3f6a865da054068466164fd2fa;p=lhc%2Fweb%2Fwiklou.git

diff --git a/includes/SiteStatsUpdate.php b/includes/SiteStatsUpdate.php
index f44a3ed9ac..1b6d3804e3 100644
--- a/includes/SiteStatsUpdate.php
+++ b/includes/SiteStatsUpdate.php
@@ -1,7 +1,6 @@
 <?php
-# $Id$
 /**
- * See deferred.doc
+ * See deferred.txt
  *
  * @package MediaWiki
  */
@@ -12,40 +11,72 @@
  */
 class SiteStatsUpdate {
 
-	var $mViews, $mEdits, $mGood;
+	var $mViews, $mEdits, $mGood, $mPages, $mUsers;
 
-	function SiteStatsUpdate( $views, $edits, $good ) {
+	function SiteStatsUpdate( $views, $edits, $good, $pages = 0, $users = 0 ) {
 		$this->mViews = $views;
 		$this->mEdits = $edits;
 		$this->mGood = $good;
+		$this->mPages = $pages;
+		$this->mUsers = $users;
+	}
+
+	function appendUpdate( &$sql, $field, $delta ) {
+		if ( $delta ) {
+			if ( $sql ) {
+				$sql .= ',';
+			}
+			if ( $delta < 0 ) {
+				$sql .= "$field=$field-1";
+			} else {
+				$sql .= "$field=$field+1";
+			}
+		}
 	}
 
 	function doUpdate() {
-		$a = array();
-
-		if ( $this->mViews < 0 ) { $m = '-1'; }
-		else if ( $this->mViews > 0 ) { $m = '+1'; }
-		else $m = '';
-		array_push( $a, "ss_total_views=(ss_total_views$m)" );
-
-		if ( $this->mEdits < 0 ) { $m = '-1'; }
-		else if ( $this->mEdits > 0 ) { $m = '+1'; }
-		else $m = '';
-		array_push( $a, "ss_total_edits=(ss_total_edits$m)" );
-
-		if ( $this->mGood < 0 ) { $m = '-1'; }
-		else if ( $this->mGood > 0 ) { $m = '+1'; }
-		else $m = '';
-		array_push( $a, "ss_good_articles=(ss_good_articles$m)" );
-
-		$db =& wfGetDB( DB_MASTER );
-		$site_stats = $db->tableName( 'site_stats' );
-		$lowpri = $db->lowPriorityOption();
-
-		$sql = "UPDATE $lowpri $site_stats SET " . implode ( ',', $a ) .
-		  ' WHERE ss_row_id=1';
-		$db->query( $sql, 'SiteStatsUpdate::doUpdate' );
+		$fname = 'SiteStatsUpdate::doUpdate';
+		$dbw =& wfGetDB( DB_MASTER );
+
+		# First retrieve the row just to find out which schema we're in
+		$row = $dbw->selectRow( 'site_stats', '*', false, $fname );
+
+		$updates = '';
+
+		$this->appendUpdate( $updates, 'ss_total_views', $this->mViews );
+		$this->appendUpdate( $updates, 'ss_total_edits', $this->mEdits );
+		$this->appendUpdate( $updates, 'ss_good_articles', $this->mGood );
+
+		if ( isset( $row->ss_total_pages ) ) {
+			# Update schema if required
+			if ( $row->ss_total_pages == -1 && !$this->mViews ) {
+				$dbr =& wfGetDB( DB_SLAVE, array( 'SpecialStatistics', 'vslow') );
+				extract( $dbr->tableNames( 'page', 'user' ) );
+
+				$sql = "SELECT COUNT(page_namespace) AS total FROM $page";
+				$res = $dbr->query( $sql, $fname );
+				$pageRow = $dbr->fetchObject( $res );
+				$pages = $pageRow->total + $this->mPages;
+
+				$sql = "SELECT COUNT(user_id) AS total FROM $user";
+				$res = $dbr->query( $sql, $fname );
+				$userRow = $dbr->fetchObject( $res );
+				$users = $userRow->total + $this->mUsers;
+
+				if ( $updates ) {
+					$updates .= ',';
+				}
+				$updates .= "ss_total_pages=$pages, ss_users=$users";
+			} else {
+				$this->appendUpdate( $updates, 'ss_total_pages', $this->mPages );
+				$this->appendUpdate( $updates, 'ss_users', $this->mUsers );
+			}
+		}
+		if ( $updates ) {
+			$site_stats = $dbw->tableName( 'site_stats' );
+			$sql = $dbw->limitResultForUpdate("UPDATE $site_stats SET $updates", 1);
+			$dbw->query( $sql, $fname );
+		}
 	}
 }
-
 ?>