3 * This program is free software; you can redistribute it and/or modify
4 * it under the terms of the GNU General Public License as published by
5 * the Free Software Foundation; either version 2 of the License, or
6 * (at your option) any later version.
8 * This program is distributed in the hope that it will be useful,
9 * but WITHOUT ANY WARRANTY; without even the implied warranty of
10 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
11 * GNU General Public License for more details.
13 * You should have received a copy of the GNU General Public License along
14 * with this program; if not, write to the Free Software Foundation, Inc.,
15 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
16 * http://www.gnu.org/copyleft/gpl.html
20 use Wikimedia\Rdbms\IDatabase
;
21 use MediaWiki\MediaWikiServices
;
24 * Class designed for counting of stats.
41 * @param bool|IDatabase $database
42 * - bool: Whether to use the master DB
43 * - IDatabase: Database connection to use
45 public function __construct( $database = false ) {
46 if ( $database instanceof IDatabase
) {
47 $this->dbr
= $database;
48 } elseif ( $database ) {
49 $this->dbr
= self
::getDB( DB_MASTER
);
51 $this->dbr
= self
::getDB( DB_REPLICA
, 'vslow' );
56 * Count the total number of edits
59 public function edits() {
60 $this->edits
= $this->dbr
->selectField( 'revision', 'COUNT(*)', '', __METHOD__
);
61 $this->edits +
= $this->dbr
->selectField( 'archive', 'COUNT(*)', '', __METHOD__
);
67 * Count pages in article space(s)
70 public function articles() {
71 $services = MediaWikiServices
::getInstance();
75 'page_namespace' => $services->getNamespaceInfo()->getContentNamespaces(),
76 'page_is_redirect' => 0,
79 if ( $services->getMainConfig()->get( 'ArticleCountMethod' ) == 'link' ) {
80 $tables[] = 'pagelinks';
81 $conds[] = 'pl_from=page_id';
84 $this->articles
= $this->dbr
->selectField(
86 'COUNT(DISTINCT page_id)',
91 return $this->articles
;
98 public function pages() {
99 $this->pages
= $this->dbr
->selectField( 'page', 'COUNT(*)', '', __METHOD__
);
108 public function users() {
109 $this->users
= $this->dbr
->selectField( 'user', 'COUNT(*)', '', __METHOD__
);
118 public function files() {
119 $this->files
= $this->dbr
->selectField( 'image', 'COUNT(*)', '', __METHOD__
);
125 * Do all updates and commit them. More or less a replacement
126 * for the original initStats, but without output.
128 * @param IDatabase|bool $database
129 * - bool: Whether to use the master DB
130 * - IDatabase: Database connection to use
131 * @param array $options Array of options, may contain the following values
132 * - activeUsers bool: Whether to update the number of active users (default: false)
134 public static function doAllAndCommit( $database, array $options = [] ) {
135 $options +
= [ 'update' => false, 'activeUsers' => false ];
137 // Grab the object and count everything
138 $counter = new self( $database );
141 $counter->articles();
148 // Count active users if need be
149 if ( $options['activeUsers'] ) {
150 SiteStatsUpdate
::cacheUpdate( self
::getDB( DB_MASTER
) );
155 * Insert a dummy row with all zeroes if no row is present
157 public static function doPlaceholderInit() {
158 $dbw = self
::getDB( DB_MASTER
);
159 $exists = $dbw->selectField( 'site_stats', '1', [ 'ss_row_id' => 1 ], __METHOD__
);
160 if ( $exists === false ) {
163 [ 'ss_row_id' => 1 ] +
array_fill_keys( SiteStats
::selectFields(), 0 ),
173 public function refresh() {
176 'ss_total_edits' => $this->edits
=== null ?
$this->edits() : $this->edits
,
177 'ss_good_articles' => $this->articles
=== null ?
$this->articles() : $this->articles
,
178 'ss_total_pages' => $this->pages
=== null ?
$this->pages() : $this->pages
,
179 'ss_users' => $this->users
=== null ?
$this->users() : $this->users
,
180 'ss_images' => $this->files
=== null ?
$this->files() : $this->files
,
183 self
::getDB( DB_MASTER
)->upsert(
194 * @param string[]|string $groups
197 private static function getDB( $index, $groups = [] ) {
198 $lb = MediaWikiServices
::getInstance()->getDBLoadBalancer();
200 return $lb->getConnection( $index, $groups );