| Index: branches/wmf/1.16wmf4/maintenance/refreshCategoryCounts.php |
| — | — | @@ -0,0 +1,102 @@ |
| | 2 | +<?php |
| | 3 | +/** |
| | 4 | + * This script will refresh the cat_pages, cat_subcats and cat_files fields of |
| | 5 | + * the category table, which may be incorrect if the wiki ran the corrupted |
| | 6 | + * version of Article::doDeleteArticle (r40912 --> r47326); see explanation at |
| | 7 | + * [https://bugzilla.wikimedia.org/show_bug.cgi?id=17155]. It will print out |
| | 8 | + * progress indicators every 1000 categories it updates. You may want to use the |
| | 9 | + * throttling options if it's causing too much load; they will not affect |
| | 10 | + * correctness. |
| | 11 | + * |
| | 12 | + * If the script is stopped and later resumed, you can use the --start option |
| | 13 | + * with the last printed progress indicator to pick up where you left off. |
| | 14 | + * This is safe, because any newly-added categories will be added at the end of |
| | 15 | + * the table. |
| | 16 | + * |
| | 17 | + * @file |
| | 18 | + * @ingroup Maintenance |
| | 19 | + * @author Happy-melon, Max Semenik |
| | 20 | + * Based on /maintenance/populateCategory.php by Simetrical. |
| | 21 | + */ |
| | 22 | + |
| | 23 | +require_once( dirname( __FILE__ ) . '/Maintenance.php' ); |
| | 24 | + |
| | 25 | +class RefreshCategoryCounts extends Maintenance { |
| | 26 | + const REPORTING_INTERVAL = 1000; |
| | 27 | + |
| | 28 | + public function __construct() { |
| | 29 | + $this->mDescription = 'Refreshes category counts'; |
| | 30 | + $this->addOption( 'start', 'Start from this category ID', false, true ); |
| | 31 | + $this->addOption( 'maxlag', 'Maximum database slave lag in seconds (5 by default)', false, true ); |
| | 32 | + $this->addOption( 'throttle', 'Optional delay after every processed category in milliseconds', |
| | 33 | + false, true ); |
| | 34 | + } |
| | 35 | + |
| | 36 | + public function execute() { |
| | 37 | + $start = intval( $this->getOption( 'start', 0 ) ); |
| | 38 | + $maxlag = intval( $this->getOption( 'maxlag', 5 ) ); |
| | 39 | + $throttle = intval( $this->getOption( 'throttle', 0 ) ); |
| | 40 | + |
| | 41 | + $this->doRefresh( $start, $maxlag, $throttle ); |
| | 42 | + } |
| | 43 | + |
| | 44 | + protected function doRefresh( $start, $maxlag, $throttle ) { |
| | 45 | + $dbw = wfGetDB( DB_MASTER ); |
| | 46 | + |
| | 47 | + $maxlag = intval( $maxlag ); |
| | 48 | + $throttle = intval( $throttle ); |
| | 49 | + $id = $start; |
| | 50 | + |
| | 51 | + $i = 0; |
| | 52 | + while ( true ) { |
| | 53 | + # Find which category to update |
| | 54 | + $row = $dbw->selectRow( |
| | 55 | + 'category', |
| | 56 | + array( 'cat_id', 'cat_title' ), |
| | 57 | + 'cat_id > ' . $dbw->addQuotes( $id ), |
| | 58 | + __METHOD__, |
| | 59 | + array( 'ORDER BY' => 'cat_id' ) |
| | 60 | + ); |
| | 61 | + if ( !$row ) { |
| | 62 | + # Done, hopefully. |
| | 63 | + break; |
| | 64 | + } |
| | 65 | + $id = $row->cat_id; |
| | 66 | + $name = $row->cat_title; |
| | 67 | + |
| | 68 | + # Use the row to update the category count |
| | 69 | + $cat = Category::newFromName( $name ); |
| | 70 | + if ( !is_object( $cat ) ) { |
| | 71 | + $this->output( "Invalid category name '$name'\n" ); |
| | 72 | + } else { |
| | 73 | + $cat->refreshCounts(); |
| | 74 | + } |
| | 75 | + |
| | 76 | + $i++; |
| | 77 | + if ( !( $i % self::REPORTING_INTERVAL ) ) { |
| | 78 | + $this->output( "$id\n" ); |
| | 79 | + wfWaitForSlaves( $maxlag ); |
| | 80 | + } |
| | 81 | + usleep( $throttle * 1000 ); |
| | 82 | + } |
| | 83 | + |
| | 84 | + /*if ( $dbw->insert( |
| | 85 | + 'updatelog', |
| | 86 | + array( 'ul_key' => 'refresh catgory counts' ), |
| | 87 | + __METHOD__, |
| | 88 | + 'IGNORE' |
| | 89 | + ) |
| | 90 | + ) { |
| | 91 | + $this->output( "Category count refresh complete.\n" ); |
| | 92 | + return true; |
| | 93 | + } else { |
| | 94 | + $this->output( "Could not insert category population row.\n" ); |
| | 95 | + return false; |
| | 96 | + }*/ |
| | 97 | + $this->output( "Category count refresh complete.\n" ); |
| | 98 | + } |
| | 99 | +} |
| | 100 | + |
| | 101 | +$maintClass = "RefreshCategoryCounts"; |
| | 102 | +require_once( DO_MAINTENANCE ); |
| | 103 | + |
| Property changes on: branches/wmf/1.16wmf4/maintenance/refreshCategoryCounts.php |
| ___________________________________________________________________ |
| Name: svn:eol-style |
| 1 | 104 | + native |