00001 <?php
00002
00006 class SiteStats {
00007 static $row, $loaded = false;
00008 static $admins, $jobs;
00009 static $pageCount = array();
00010 static $groupMemberCounts = array();
00011
00012 static function recache() {
00013 self::load( true );
00014 }
00015
00016 static function load( $recache = false ) {
00017 if ( self::$loaded && !$recache ) {
00018 return;
00019 }
00020
00021 self::$row = self::loadAndLazyInit();
00022
00023 # This code is somewhat schema-agnostic, because I'm changing it in a minor release -- TS
00024 if ( !isset( self::$row->ss_total_pages ) && self::$row->ss_total_pages == -1 ) {
00025 # Update schema
00026 $u = new SiteStatsUpdate( 0, 0, 0 );
00027 $u->doUpdate();
00028 $dbr = wfGetDB( DB_SLAVE );
00029 self::$row = $dbr->selectRow( 'site_stats', '*', false, __METHOD__ );
00030 }
00031
00032 self::$loaded = true;
00033 }
00034
00035 static function loadAndLazyInit() {
00036 wfDebug( __METHOD__ . ": reading site_stats from slave\n" );
00037 $row = self::doLoad( wfGetDB( DB_SLAVE ) );
00038
00039 if( !self::isSane( $row ) ) {
00040
00041 wfDebug( __METHOD__ . ": site_stats damaged or missing on slave\n" );
00042 $row = self::doLoad( wfGetDB( DB_MASTER ) );
00043 }
00044
00045 if( !self::isSane( $row ) ) {
00046
00047
00048
00049
00050 wfDebug( __METHOD__ . ": initializing damaged or missing site_stats\n" );
00051
00052 SiteStatsInit::doAllAndCommit( false );
00053
00054 $row = self::doLoad( wfGetDB( DB_MASTER ) );
00055 }
00056
00057 if( !self::isSane( $row ) ) {
00058 wfDebug( __METHOD__ . ": site_stats persistently nonsensical o_O\n" );
00059 }
00060 return $row;
00061 }
00062
00063 static function doLoad( $db ) {
00064 return $db->selectRow( 'site_stats', '*', false, __METHOD__ );
00065 }
00066
00067 static function views() {
00068 self::load();
00069 return self::$row->ss_total_views;
00070 }
00071
00072 static function edits() {
00073 self::load();
00074 return self::$row->ss_total_edits;
00075 }
00076
00077 static function articles() {
00078 self::load();
00079 return self::$row->ss_good_articles;
00080 }
00081
00082 static function pages() {
00083 self::load();
00084 return self::$row->ss_total_pages;
00085 }
00086
00087 static function users() {
00088 self::load();
00089 return self::$row->ss_users;
00090 }
00091
00092 static function activeUsers() {
00093 self::load();
00094 return self::$row->ss_active_users;
00095 }
00096
00097 static function images() {
00098 self::load();
00099 return self::$row->ss_images;
00100 }
00101
00105 static function admins() {
00106 wfDeprecated(__METHOD__);
00107 return self::numberingroup('sysop');
00108 }
00109
00115 static function numberingroup($group) {
00116 if ( !isset( self::$groupMemberCounts[$group] ) ) {
00117 global $wgMemc;
00118 $key = wfMemcKey( 'SiteStats', 'groupcounts', $group );
00119 $hit = $wgMemc->get( $key );
00120 if ( !$hit ) {
00121 $dbr = wfGetDB( DB_SLAVE );
00122 $hit = $dbr->selectField( 'user_groups', 'COUNT(*)',
00123 array( 'ug_group' => $group ), __METHOD__ );
00124 $wgMemc->set( $key, $hit, 3600 );
00125 }
00126 self::$groupMemberCounts[$group] = $hit;
00127 }
00128 return self::$groupMemberCounts[$group];
00129 }
00130
00131 static function jobs() {
00132 if ( !isset( self::$jobs ) ) {
00133 $dbr = wfGetDB( DB_SLAVE );
00134 self::$jobs = $dbr->estimateRowCount('job');
00135
00136 if (self::$jobs == 1) {
00137 self::$jobs = 0;
00138 }
00139 }
00140 return self::$jobs;
00141 }
00142
00143 static function pagesInNs( $ns ) {
00144 wfProfileIn( __METHOD__ );
00145 if( !isset( self::$pageCount[$ns] ) ) {
00146 $dbr = wfGetDB( DB_SLAVE );
00147 $pageCount[$ns] = (int)$dbr->selectField( 'page', 'COUNT(*)', array( 'page_namespace' => $ns ), __METHOD__ );
00148 }
00149 wfProfileOut( __METHOD__ );
00150 return $pageCount[$ns];
00151 }
00152
00154 private static function isSane( $row ) {
00155 if(
00156 $row === false
00157 or $row->ss_total_pages < $row->ss_good_articles
00158 or $row->ss_total_edits < $row->ss_total_pages
00159 or $row->ss_users < $row->ss_admins
00160 ) {
00161 return false;
00162 }
00163
00164 foreach( array( 'total_views', 'total_edits', 'good_articles',
00165 'total_pages', 'users', 'admins', 'images' ) as $member ) {
00166 if(
00167 $row->{"ss_$member"} > 2000000000
00168 or $row->{"ss_$member"} < 0
00169 ) {
00170 return false;
00171 }
00172 }
00173 return true;
00174 }
00175 }
00176
00177
00181 class SiteStatsUpdate {
00182
00183 var $mViews, $mEdits, $mGood, $mPages, $mUsers;
00184
00185 function __construct( $views, $edits, $good, $pages = 0, $users = 0 ) {
00186 $this->mViews = $views;
00187 $this->mEdits = $edits;
00188 $this->mGood = $good;
00189 $this->mPages = $pages;
00190 $this->mUsers = $users;
00191 }
00192
00193 function appendUpdate( &$sql, $field, $delta ) {
00194 if ( $delta ) {
00195 if ( $sql ) {
00196 $sql .= ',';
00197 }
00198 if ( $delta < 0 ) {
00199 $sql .= "$field=$field-1";
00200 } else {
00201 $sql .= "$field=$field+1";
00202 }
00203 }
00204 }
00205
00206 function doUpdate() {
00207 $dbw = wfGetDB( DB_MASTER );
00208
00209 $updates = '';
00210
00211 $this->appendUpdate( $updates, 'ss_total_views', $this->mViews );
00212 $this->appendUpdate( $updates, 'ss_total_edits', $this->mEdits );
00213 $this->appendUpdate( $updates, 'ss_good_articles', $this->mGood );
00214 $this->appendUpdate( $updates, 'ss_total_pages', $this->mPages );
00215 $this->appendUpdate( $updates, 'ss_users', $this->mUsers );
00216
00217 if ( $updates ) {
00218 $site_stats = $dbw->tableName( 'site_stats' );
00219 $sql = "UPDATE $site_stats SET $updates";
00220
00221 # Need a separate transaction because this a global lock
00222 $dbw->begin();
00223 $dbw->query( $sql, __METHOD__ );
00224 $dbw->commit();
00225 }
00226 }
00227
00228 public static function cacheUpdate( $dbw ) {
00229 $dbr = wfGetDB( DB_SLAVE, array( 'SpecialStatistics', 'vslow') );
00230 # Get non-bot users than did some recent action other than making accounts.
00231 # If account creation is included, the number gets inflated ~20+ fold on enwiki.
00232 $activeUsers = $dbr->selectField( 'recentchanges', 'COUNT( DISTINCT rc_user_text )',
00233 array( 'rc_user != 0', 'rc_bot' => 0, "rc_log_type != 'newusers' OR rc_log_type IS NULL" ),
00234 __METHOD__ );
00235 $dbw->update( 'site_stats',
00236 array( 'ss_active_users' => intval($activeUsers) ),
00237 array( 'ss_row_id' => 1 ), __METHOD__
00238 );
00239 return $activeUsers;
00240 }
00241 }
00242
00246 class SiteStatsInit {
00247
00248
00249 private $db;
00250
00251
00252 private $mEdits, $mArticles, $mPages, $mUsers, $mViews, $mFiles = 0;
00253
00258 public function __construct( $useMaster = false ) {
00259 $this->db = wfGetDB( $useMaster ? DB_MASTER : DB_SLAVE );
00260 }
00261
00266 public function edits() {
00267 $this->mEdits = $this->db->selectField( 'revision', 'COUNT(*)', '', __METHOD__ );
00268 $this->mEdits += $this->db->selectField( 'archive', 'COUNT(*)', '', __METHOD__ );
00269 return $this->mEdits;
00270 }
00271
00276 public function articles() {
00277 global $wgContentNamespaces;
00278 $this->mArticles = $this->db->selectField( 'page', 'COUNT(*)', array( 'page_namespace' => $wgContentNamespaces, 'page_is_redirect' => 0, 'page_len > 0' ), __METHOD__ );
00279 return $this->mArticles;
00280 }
00281
00286 public function pages() {
00287 $this->mPages = $this->db->selectField( 'page', 'COUNT(*)', '', __METHOD__ );
00288 return $this->mPages;
00289 }
00290
00295 public function users() {
00296 $this->mUsers = $this->db->selectField( 'user', 'COUNT(*)', '', __METHOD__ );
00297 return $this->mUsers;
00298 }
00299
00304 public function views() {
00305 $this->mViews = $this->db->selectField( 'page', 'SUM(page_counter)', '', __METHOD__ );
00306 return $this->mViews;
00307 }
00308
00313 public function files() {
00314 $this->mFiles = $this->db->selectField( 'image', 'COUNT(*)', '', __METHOD__ );
00315 return $this->mFiles;
00316 }
00317
00325 public static function doAllAndCommit( $update, $noViews = false, $activeUsers = false ) {
00326
00327 $counter = new SiteStatsInit( false );
00328 $counter->edits();
00329 $counter->articles();
00330 $counter->pages();
00331 $counter->users();
00332 $counter->files();
00333
00334
00335 if( !$noViews )
00336 $counter->views();
00337
00338
00339 if( $update )
00340 $counter->update();
00341 else
00342 $counter->refresh();
00343
00344
00345 if( $activeUsers )
00346 SiteStatsUpdate::cacheUpdate( wfGetDB( DB_MASTER ) );
00347 }
00348
00352 public function update() {
00353 list( $values, $conds ) = $this->getDbParams();
00354 $dbw = wfGetDB( DB_MASTER );
00355 $dbw->update( 'site_stats', $values, $conds, __METHOD__ );
00356 }
00357
00362 public function refresh() {
00363 list( $values, $conds, $views ) = $this->getDbParams();
00364 $dbw = wfGetDB( DB_MASTER );
00365 $dbw->delete( 'site_stats', $conds, __METHOD__ );
00366 $dbw->insert( 'site_stats', array_merge( $values, $conds, $views ), __METHOD__ );
00367 }
00368
00373 private function getDbParams() {
00374 $values = array( 'ss_total_edits' => $this->mEdits,
00375 'ss_good_articles' => $this->mArticles,
00376 'ss_total_pages' => $this->mPages,
00377 'ss_users' => $this->mUsers,
00378 'ss_admins' => SiteStats::numberingroup( 'sysop' ),
00379 'ss_images' => $this->mFiles );
00380 $conds = array( 'ss_row_id' => 1 );
00381 $views = array( 'ss_total_views' => $this->mViews );
00382 return array( $values, $conds, $views );
00383 }
00384 }