Code Coverage
 
Lines
Functions and Methods
Classes and Traits
Total
0.00% covered (danger)
0.00%
0 / 67
0.00% covered (danger)
0.00%
0 / 2
CRAP
0.00% covered (danger)
0.00%
0 / 1
InitEditCount
0.00% covered (danger)
0.00%
0 / 64
0.00% covered (danger)
0.00%
0 / 2
72
0.00% covered (danger)
0.00%
0 / 1
 __construct
0.00% covered (danger)
0.00%
0 / 7
0.00% covered (danger)
0.00%
0 / 1
2
 execute
0.00% covered (danger)
0.00%
0 / 57
0.00% covered (danger)
0.00%
0 / 1
56
1<?php
2/**
3 * Init the user_editcount database field based on the number of rows in the
4 * revision table.
5 *
6 * This program is free software; you can redistribute it and/or modify
7 * it under the terms of the GNU General Public License as published by
8 * the Free Software Foundation; either version 2 of the License, or
9 * (at your option) any later version.
10 *
11 * This program is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 * GNU General Public License for more details.
15 *
16 * You should have received a copy of the GNU General Public License along
17 * with this program; if not, write to the Free Software Foundation, Inc.,
18 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
19 * http://www.gnu.org/copyleft/gpl.html
20 *
21 * @file
22 * @ingroup Maintenance
23 */
24
25require_once __DIR__ . '/Maintenance.php';
26
27use MediaWiki\WikiMap\WikiMap;
28use Wikimedia\Rdbms\IDatabase;
29use Wikimedia\Rdbms\RawSQLValue;
30
31class InitEditCount extends Maintenance {
32    public function __construct() {
33        parent::__construct();
34        $this->addOption( 'quick', 'Force the update to be done in a single query' );
35        $this->addOption( 'background', 'Force replication-friendly mode; may be inefficient but avoids'
36        . 'locking tables or lagging replica DBs with large updates; calculates counts on a replica DB'
37        . 'if possible. Background mode will be automatically used if multiple servers are listed in the'
38        . 'load balancer, usually indicating a replication environment.' );
39        $this->addDescription( 'Batch-recalculate user_editcount fields from the revision table' );
40    }
41
42    public function execute() {
43        $dbw = $this->getPrimaryDB();
44
45        // Autodetect mode...
46        if ( $this->hasOption( 'background' ) ) {
47            $backgroundMode = true;
48        } elseif ( $this->hasOption( 'quick' ) ) {
49            $backgroundMode = false;
50        } else {
51            $lb = $this->getServiceContainer()->getDBLoadBalancer();
52            $backgroundMode = $lb->hasReplicaServers();
53        }
54
55        if ( $backgroundMode ) {
56            $this->output( "Using replication-friendly background mode...\n" );
57
58            $dbr = $this->getReplicaDB();
59            $chunkSize = 100;
60            $lastUser = $dbr->newSelectQueryBuilder()
61                ->select( 'MAX(user_id)' )
62                ->from( 'user' )
63                ->caller( __METHOD__ )->fetchField();
64
65            $start = microtime( true );
66            $migrated = 0;
67            for ( $min = 0; $min <= $lastUser; $min += $chunkSize ) {
68                $max = $min + $chunkSize;
69
70                $result = $dbr->newSelectQueryBuilder()
71                    ->select( [ 'user_id', 'user_editcount' => "COUNT(actor_rev_user.actor_user)" ] )
72                    ->from( 'user' )
73                    ->leftJoin( 'revision', 'rev', "user_id = actor_rev_user.actor_user" )
74                    ->join( 'actor', 'actor_rev_user', 'actor_rev_user.actor_id = rev_actor' )
75                    ->where( "user_id > $min AND user_id <= $max" )
76                    ->groupBy( 'user_id' )
77                    ->caller( __METHOD__ )->fetchResultSet();
78
79                foreach ( $result as $row ) {
80                    $dbw->newUpdateQueryBuilder()
81                        ->update( 'user' )
82                        ->set( [ 'user_editcount' => $row->user_editcount ] )
83                        ->where( [ 'user_id' => $row->user_id ] )
84                        ->caller( __METHOD__ )->execute();
85                    ++$migrated;
86                }
87
88                $delta = microtime( true ) - $start;
89                $rate = ( $delta == 0.0 ) ? 0.0 : $migrated / $delta;
90                $this->output( sprintf( "%s %d (%0.1f%%) done in %0.1f secs (%0.3f accounts/sec).\n",
91                    WikiMap::getCurrentWikiDbDomain()->getId(),
92                    $migrated,
93                    min( $max, $lastUser ) / $lastUser * 100.0,
94                    $delta,
95                    $rate ) );
96
97                $this->waitForReplication();
98            }
99        } else {
100            $this->output( "Using single-query mode...\n" );
101
102            $subquery = $dbw->newSelectQueryBuilder()
103                ->select( 'COUNT(*)' )
104                ->from( 'revision' )
105                ->join( 'actor', 'actor_rev_user', 'actor_rev_user.actor_id = rev_actor' )
106                ->where( 'user_id = actor_rev_user.actor_user' )
107                ->caller( __METHOD__ )->getSQL();
108
109            $dbw->newUpdateQueryBuilder()
110                ->table( 'user' )
111                ->set( [ 'user_editcount' => new RawSQLValue( "($subquery)" ) ] )
112                ->where( IDatabase::ALL_ROWS )
113                ->caller( __METHOD__ )
114                ->execute();
115        }
116
117        $this->output( "Done!\n" );
118    }
119}
120
121$maintClass = InitEditCount::class;
122require_once RUN_MAINTENANCE_IF_MAIN;