purgeList.php

Go to the documentation of this file.

<?php
require_once( dirname( __FILE__ ) . '/Maintenance.php' );

class PurgeList extends Maintenance {
        public function __construct() {
                parent::__construct();
                $this->mDescription = "Send purge requests for listed pages to squid";
                $this->addOption( 'purge', 'Whether to update page_touched.' , false, false );
                $this->addOption( 'namespace', 'Namespace number', false, true );
                $this->setBatchSize( 100 );
        }

        public function execute() {
                if( $this->hasOption( 'namespace' ) ) {
                        $this->purgeNamespace();
                } else {
                        $this->purgeList();
                }
                $this->output( "Done!\n" );
        }

        private function purgeList() {
                $stdin = $this->getStdin();
                $urls = array();

                while ( !feof( $stdin ) ) {
                        $page = trim( fgets( $stdin ) );
                        if ( preg_match( '%^https?://%', $page ) ) {
                                $urls[] = $page;
                        } elseif ( $page !== '' ) {
                                $title = Title::newFromText( $page );
                                if ( $title ) {
                                        $url = $title->getInternalUrl();
                                        $this->output( "$url\n" );
                                        $urls[] = $url;
                                        if ( $this->getOption( 'purge' ) ) {
                                                $title->invalidateCache();
                                        }
                                } else {
                                        $this->output( "(Invalid title '$page')\n" );
                                }
                        }
                }
                $this->sendPurgeRequest( $urls );
        }

        private function purgeNamespace() {
                $dbr = wfGetDB( DB_SLAVE );
                $ns = $dbr->addQuotes( $this->getOption( 'namespace') );

                $result = $dbr->select(
                        array( 'page' ),
                        array( 'page_namespace', 'page_title' ),
                        array( "page_namespace = $ns" ),
                        __METHOD__,
                        array( 'ORDER BY' => 'page_id' )
                );

                $start   = 0;
                $end = $dbr->numRows( $result );
                $this->output( "Will purge $end pages from namespace $ns\n" );

                # Do remaining chunk
                $end += $this->mBatchSize - 1;
                $blockStart = $start;
                $blockEnd = $start + $this->mBatchSize - 1;

                while( $blockEnd <= $end ) {
                        # Select pages we will purge:
                        $result = $dbr->select(
                                array( 'page' ),
                                array( 'page_namespace', 'page_title' ),
                                array( "page_namespace = $ns" ),
                                __METHOD__,
                                array( # conditions
                                        'ORDER BY' => 'page_id',
                                        'LIMIT'    => $this->mBatchSize,
                                        'OFFSET'   => $blockStart,
                                )
                        );
                        # Initialize/reset URLs to be purged
                        $urls = array();
                        foreach( $result as $row ) {
                                $title = Title::makeTitle( $row->page_namespace, $row->page_title );
                                $url = $title->getInternalUrl();
                                $urls[] = $url;
                        }

                        $this->sendPurgeRequest( $urls );

                        $blockStart += $this->mBatchSize;
                        $blockEnd   += $this->mBatchSize;
                }
        }

        private function sendPurgeRequest( $urls ) {
                $this->output( "Purging " . count( $urls ). " urls\n" );
                $u = new SquidUpdate( $urls );
                $u->doUpdate();
        }

}

$maintClass = "PurgeList";
require_once( RUN_MAINTENANCE_IF_MAIN );