blob: e05794cca5b6a66b287e923db5a9fc62c30e74a5 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
|
<?php
$IP = getenv( 'MW_INSTALL_PATH' );
if ( $IP === false ) {
$IP = __DIR__ . '/../../..';
}
require_once( "$IP/maintenance/Maintenance.php" );
/**
* Go through all usernames and calculate and record spoof thingies
*/
class BatchAntiSpoof extends Maintenance {
/**
* @param $items array
*/
protected function batchRecord( $items ) {
SpoofUser::batchRecord( $this->getDB( DB_MASTER ), $items );
}
/**
* @return string
*/
protected function getTableName() {
return 'user';
}
/**
* @return string
*/
protected function getUserColumn() {
return 'user_name';
}
/**
* @param $name string
* @return SpoofUser
*/
protected function makeSpoofUser( $name ) {
return new SpoofUser( $name );
}
/**
* Do the actual work. All child classes will need to implement this
*/
public function execute() {
$dbw = $this->getDB( DB_MASTER );
$batchSize = 1000;
$this->output( "Creating username spoofs...\n" );
$userCol = $this->getUserColumn();
$result = $dbw->select( $this->getTableName(), $userCol, null, __FUNCTION__ );
$n = 0;
$items = array();
foreach( $result as $row ) {
if ( $n++ % $batchSize == 0 ) {
$this->output( "...$n\n" );
}
$items[] = $this->makeSpoofUser( $row->$userCol );
if ( $n % $batchSize == 0 ) {
$this->batchRecord( $items );
$items = array();
wfWaitForSlaves();
}
}
$this->batchRecord( $items );
$this->output( "$n user(s) done.\n" );
}
}
|