384 lines
9.9 KiB
PHP
384 lines
9.9 KiB
PHP
<?php
|
|
|
|
/**
|
|
* Static accessor class for site_stats and related things
|
|
*/
|
|
class SiteStats {
|
|
static $row, $loaded = false;
|
|
static $admins, $jobs;
|
|
static $pageCount = array();
|
|
static $groupMemberCounts = array();
|
|
|
|
static function recache() {
|
|
self::load( true );
|
|
}
|
|
|
|
static function load( $recache = false ) {
|
|
if ( self::$loaded && !$recache ) {
|
|
return;
|
|
}
|
|
|
|
self::$row = self::loadAndLazyInit();
|
|
|
|
# This code is somewhat schema-agnostic, because I'm changing it in a minor release -- TS
|
|
if ( !isset( self::$row->ss_total_pages ) && self::$row->ss_total_pages == -1 ) {
|
|
# Update schema
|
|
$u = new SiteStatsUpdate( 0, 0, 0 );
|
|
$u->doUpdate();
|
|
$dbr = wfGetDB( DB_SLAVE );
|
|
self::$row = $dbr->selectRow( 'site_stats', '*', false, __METHOD__ );
|
|
}
|
|
|
|
self::$loaded = true;
|
|
}
|
|
|
|
static function loadAndLazyInit() {
|
|
wfDebug( __METHOD__ . ": reading site_stats from slave\n" );
|
|
$row = self::doLoad( wfGetDB( DB_SLAVE ) );
|
|
|
|
if( !self::isSane( $row ) ) {
|
|
// Might have just been initialized during this request? Underflow?
|
|
wfDebug( __METHOD__ . ": site_stats damaged or missing on slave\n" );
|
|
$row = self::doLoad( wfGetDB( DB_MASTER ) );
|
|
}
|
|
|
|
if( !self::isSane( $row ) ) {
|
|
// Normally the site_stats table is initialized at install time.
|
|
// Some manual construction scenarios may leave the table empty or
|
|
// broken, however, for instance when importing from a dump into a
|
|
// clean schema with mwdumper.
|
|
wfDebug( __METHOD__ . ": initializing damaged or missing site_stats\n" );
|
|
|
|
SiteStatsInit::doAllAndCommit( false );
|
|
|
|
$row = self::doLoad( wfGetDB( DB_MASTER ) );
|
|
}
|
|
|
|
if( !self::isSane( $row ) ) {
|
|
wfDebug( __METHOD__ . ": site_stats persistently nonsensical o_O\n" );
|
|
}
|
|
return $row;
|
|
}
|
|
|
|
static function doLoad( $db ) {
|
|
return $db->selectRow( 'site_stats', '*', false, __METHOD__ );
|
|
}
|
|
|
|
static function views() {
|
|
self::load();
|
|
return self::$row->ss_total_views;
|
|
}
|
|
|
|
static function edits() {
|
|
self::load();
|
|
return self::$row->ss_total_edits;
|
|
}
|
|
|
|
static function articles() {
|
|
self::load();
|
|
return self::$row->ss_good_articles;
|
|
}
|
|
|
|
static function pages() {
|
|
self::load();
|
|
return self::$row->ss_total_pages;
|
|
}
|
|
|
|
static function users() {
|
|
self::load();
|
|
return self::$row->ss_users;
|
|
}
|
|
|
|
static function activeUsers() {
|
|
self::load();
|
|
return self::$row->ss_active_users;
|
|
}
|
|
|
|
static function images() {
|
|
self::load();
|
|
return self::$row->ss_images;
|
|
}
|
|
|
|
/**
|
|
* @deprecated Use self::numberingroup('sysop') instead
|
|
*/
|
|
static function admins() {
|
|
wfDeprecated(__METHOD__);
|
|
return self::numberingroup('sysop');
|
|
}
|
|
|
|
/**
|
|
* Find the number of users in a given user group.
|
|
* @param string $group Name of group
|
|
* @return int
|
|
*/
|
|
static function numberingroup($group) {
|
|
if ( !isset( self::$groupMemberCounts[$group] ) ) {
|
|
global $wgMemc;
|
|
$key = wfMemcKey( 'SiteStats', 'groupcounts', $group );
|
|
$hit = $wgMemc->get( $key );
|
|
if ( !$hit ) {
|
|
$dbr = wfGetDB( DB_SLAVE );
|
|
$hit = $dbr->selectField( 'user_groups', 'COUNT(*)',
|
|
array( 'ug_group' => $group ), __METHOD__ );
|
|
$wgMemc->set( $key, $hit, 3600 );
|
|
}
|
|
self::$groupMemberCounts[$group] = $hit;
|
|
}
|
|
return self::$groupMemberCounts[$group];
|
|
}
|
|
|
|
static function jobs() {
|
|
if ( !isset( self::$jobs ) ) {
|
|
$dbr = wfGetDB( DB_SLAVE );
|
|
self::$jobs = $dbr->estimateRowCount('job');
|
|
/* Zero rows still do single row read for row that doesn't exist, but people are annoyed by that */
|
|
if (self::$jobs == 1) {
|
|
self::$jobs = 0;
|
|
}
|
|
}
|
|
return self::$jobs;
|
|
}
|
|
|
|
static function pagesInNs( $ns ) {
|
|
wfProfileIn( __METHOD__ );
|
|
if( !isset( self::$pageCount[$ns] ) ) {
|
|
$dbr = wfGetDB( DB_SLAVE );
|
|
$pageCount[$ns] = (int)$dbr->selectField( 'page', 'COUNT(*)', array( 'page_namespace' => $ns ), __METHOD__ );
|
|
}
|
|
wfProfileOut( __METHOD__ );
|
|
return $pageCount[$ns];
|
|
}
|
|
|
|
/** Is the provided row of site stats sane, or should it be regenerated? */
|
|
private static function isSane( $row ) {
|
|
if(
|
|
$row === false
|
|
or $row->ss_total_pages < $row->ss_good_articles
|
|
or $row->ss_total_edits < $row->ss_total_pages
|
|
or $row->ss_users < $row->ss_admins
|
|
) {
|
|
return false;
|
|
}
|
|
// Now check for underflow/overflow
|
|
foreach( array( 'total_views', 'total_edits', 'good_articles',
|
|
'total_pages', 'users', 'admins', 'images' ) as $member ) {
|
|
if(
|
|
$row->{"ss_$member"} > 2000000000
|
|
or $row->{"ss_$member"} < 0
|
|
) {
|
|
return false;
|
|
}
|
|
}
|
|
return true;
|
|
}
|
|
}
|
|
|
|
|
|
/**
|
|
*
|
|
*/
|
|
class SiteStatsUpdate {
|
|
|
|
var $mViews, $mEdits, $mGood, $mPages, $mUsers;
|
|
|
|
function __construct( $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() {
|
|
$dbw = wfGetDB( DB_MASTER );
|
|
|
|
$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 );
|
|
$this->appendUpdate( $updates, 'ss_total_pages', $this->mPages );
|
|
$this->appendUpdate( $updates, 'ss_users', $this->mUsers );
|
|
|
|
if ( $updates ) {
|
|
$site_stats = $dbw->tableName( 'site_stats' );
|
|
$sql = "UPDATE $site_stats SET $updates";
|
|
|
|
# Need a separate transaction because this a global lock
|
|
$dbw->begin();
|
|
$dbw->query( $sql, __METHOD__ );
|
|
$dbw->commit();
|
|
}
|
|
}
|
|
|
|
public static function cacheUpdate( $dbw ) {
|
|
$dbr = wfGetDB( DB_SLAVE, array( 'SpecialStatistics', 'vslow') );
|
|
# Get non-bot users than did some recent action other than making accounts.
|
|
# If account creation is included, the number gets inflated ~20+ fold on enwiki.
|
|
$activeUsers = $dbr->selectField( 'recentchanges', 'COUNT( DISTINCT rc_user_text )',
|
|
array( 'rc_user != 0', 'rc_bot' => 0, "rc_log_type != 'newusers' OR rc_log_type IS NULL" ),
|
|
__METHOD__ );
|
|
$dbw->update( 'site_stats',
|
|
array( 'ss_active_users' => intval($activeUsers) ),
|
|
array( 'ss_row_id' => 1 ), __METHOD__
|
|
);
|
|
return $activeUsers;
|
|
}
|
|
}
|
|
|
|
/**
|
|
* Class designed for counting of stats.
|
|
*/
|
|
class SiteStatsInit {
|
|
|
|
// Db connection
|
|
private $db;
|
|
|
|
// Various stats
|
|
private $mEdits, $mArticles, $mPages, $mUsers, $mViews, $mFiles = 0;
|
|
|
|
/**
|
|
* Constructor
|
|
* @param $useMaster bool Whether to use the master db
|
|
*/
|
|
public function __construct( $useMaster = false ) {
|
|
$this->db = wfGetDB( $useMaster ? DB_MASTER : DB_SLAVE );
|
|
}
|
|
|
|
/**
|
|
* Count the total number of edits
|
|
* @return int
|
|
*/
|
|
public function edits() {
|
|
$this->mEdits = $this->db->selectField( 'revision', 'COUNT(*)', '', __METHOD__ );
|
|
$this->mEdits += $this->db->selectField( 'archive', 'COUNT(*)', '', __METHOD__ );
|
|
return $this->mEdits;
|
|
}
|
|
|
|
/**
|
|
* Count pages in article space
|
|
* @return int
|
|
*/
|
|
public function articles() {
|
|
global $wgContentNamespaces;
|
|
$this->mArticles = $this->db->selectField( 'page', 'COUNT(*)', array( 'page_namespace' => $wgContentNamespaces, 'page_is_redirect' => 0, 'page_len > 0' ), __METHOD__ );
|
|
return $this->mArticles;
|
|
}
|
|
|
|
/**
|
|
* Count total pages
|
|
* @return int
|
|
*/
|
|
public function pages() {
|
|
$this->mPages = $this->db->selectField( 'page', 'COUNT(*)', '', __METHOD__ );
|
|
return $this->mPages;
|
|
}
|
|
|
|
/**
|
|
* Count total users
|
|
* @return int
|
|
*/
|
|
public function users() {
|
|
$this->mUsers = $this->db->selectField( 'user', 'COUNT(*)', '', __METHOD__ );
|
|
return $this->mUsers;
|
|
}
|
|
|
|
/**
|
|
* Count views
|
|
* @return int
|
|
*/
|
|
public function views() {
|
|
$this->mViews = $this->db->selectField( 'page', 'SUM(page_counter)', '', __METHOD__ );
|
|
return $this->mViews;
|
|
}
|
|
|
|
/**
|
|
* Count total files
|
|
* @return int
|
|
*/
|
|
public function files() {
|
|
$this->mFiles = $this->db->selectField( 'image', 'COUNT(*)', '', __METHOD__ );
|
|
return $this->mFiles;
|
|
}
|
|
|
|
/**
|
|
* Do all updates and commit them. More or less a replacement
|
|
* for the original initStats, but without the calls to wfOut()
|
|
* @param $update bool Whether to update the current stats or write fresh
|
|
* @param $noViews bool When true, do not update the number of page views
|
|
* @param $activeUsers Whether to update the number of active users
|
|
*/
|
|
public static function doAllAndCommit( $update, $noViews = false, $activeUsers = false ) {
|
|
// Grab the object and count everything
|
|
$counter = new SiteStatsInit( false );
|
|
$counter->edits();
|
|
$counter->articles();
|
|
$counter->pages();
|
|
$counter->users();
|
|
$counter->files();
|
|
|
|
// Only do views if we don't want to not count them
|
|
if( !$noViews )
|
|
$counter->views();
|
|
|
|
// Update/refresh
|
|
if( $update )
|
|
$counter->update();
|
|
else
|
|
$counter->refresh();
|
|
|
|
// Count active users if need be
|
|
if( $activeUsers )
|
|
SiteStatsUpdate::cacheUpdate( wfGetDB( DB_MASTER ) );
|
|
}
|
|
|
|
/**
|
|
* Update the current row with the selected values
|
|
*/
|
|
public function update() {
|
|
list( $values, $conds ) = $this->getDbParams();
|
|
$dbw = wfGetDB( DB_MASTER );
|
|
$dbw->update( 'site_stats', $values, $conds, __METHOD__ );
|
|
}
|
|
|
|
/**
|
|
* Refresh site_stats. Erase the current record and save all
|
|
* the new values.
|
|
*/
|
|
public function refresh() {
|
|
list( $values, $conds, $views ) = $this->getDbParams();
|
|
$dbw = wfGetDB( DB_MASTER );
|
|
$dbw->delete( 'site_stats', $conds, __METHOD__ );
|
|
$dbw->insert( 'site_stats', array_merge( $values, $conds, $views ), __METHOD__ );
|
|
}
|
|
|
|
/**
|
|
* Return three arrays of params for the db queries
|
|
* @return array
|
|
*/
|
|
private function getDbParams() {
|
|
$values = array( 'ss_total_edits' => $this->mEdits,
|
|
'ss_good_articles' => $this->mArticles,
|
|
'ss_total_pages' => $this->mPages,
|
|
'ss_users' => $this->mUsers,
|
|
'ss_admins' => SiteStats::numberingroup( 'sysop' ), // @todo make this go away
|
|
'ss_images' => $this->mFiles );
|
|
$conds = array( 'ss_row_id' => 1 );
|
|
$views = array( 'ss_total_views' => $this->mViews );
|
|
return array( $values, $conds, $views );
|
|
}
|
|
}
|