MediaWiki master
initUserPreference.php
Go to the documentation of this file.
1<?php
2
4
5// @codeCoverageIgnoreStart
6require_once __DIR__ . '/Maintenance.php';
7// @codeCoverageIgnoreEnd
8
20 public function __construct() {
21 parent::__construct();
22 $this->addOption(
23 'target',
24 'Name of the user preference to initialize',
25 true,
26 true,
27 't'
28 );
29 $this->addOption(
30 'source',
31 'Name of the user preference to take the value from',
32 true,
33 true,
34 's'
35 );
36 $this->setBatchSize( 300 );
37 }
38
39 public function execute() {
40 $target = $this->getOption( 'target' );
41 $source = $this->getOption( 'source' );
42 $this->output( "Initializing '$target' based on the value of '$source'\n" );
43
44 $dbr = $this->getReplicaDB();
45 $dbw = $this->getPrimaryDB();
46
47 $iterator = new BatchRowIterator(
48 $dbr,
49 $dbr->newSelectQueryBuilder()
50 ->from( 'user_properties' )
51 ->select( [ 'up_user', 'up_value' ] )
52 ->where( [
53 'up_property' => $source,
54 $dbr->expr( 'up_value', '!=', null ),
55 $dbr->expr( 'up_value', '!=', '0' ),
56 ] )
57 ->caller( __METHOD__ ),
58 [ 'up_user', 'up_property' ],
59 $this->getBatchSize()
60 );
61
62 $processed = 0;
63 foreach ( $iterator as $batch ) {
64 foreach ( $batch as $row ) {
65 $dbw->newInsertQueryBuilder()
66 ->insertInto( 'user_properties' )
67 ->row( [
68 'up_user' => $row->up_user,
69 'up_property' => $target,
70 'up_value' => $row->up_value,
71 ] )
72 ->onDuplicateKeyUpdate()
73 ->uniqueIndexFields( [ 'up_user', 'up_property' ] )
74 ->set( [ 'up_value' => $row->up_value ] )
75 ->caller( __METHOD__ )->execute();
76
77 $processed += $dbw->affectedRows();
78 }
79 }
80
81 $this->output( "Processed $processed user(s)\n" );
82 $this->output( "Finished!\n" );
83 }
84}
85
86// @codeCoverageIgnoreStart
87$maintClass = InitUserPreference::class;
88require_once RUN_MAINTENANCE_IF_MAIN;
89// @codeCoverageIgnoreEnd
Allows iterating a large number of rows in batches transparently.
Maintenance script that initializes a user preference based on the value of another preference.
execute()
Do the actual work.
__construct()
Default constructor.
Abstract maintenance class for quickly writing and churning out maintenance scripts with minimal effo...
output( $out, $channel=null)
Throw some output to the user.
addOption( $name, $description, $required=false, $withArg=false, $shortName=false, $multiOccurrence=false)
Add a parameter to the script.
getOption( $name, $default=null)
Get an option, or return the default.
$source