deleteAutoPatrolLogs.php

Go to the documentation of this file.

<?php
require_once __DIR__ . '/Maintenance.php';

class DeleteAutoPatrolLogs extends Maintenance {

    public function __construct() {
        parent::__construct();
        $this->addDescription( 'Remove autopatrol logs in the logging table' );
        $this->addOption( 'dry-run', 'Print debug info instead of actually deleting' );
        $this->addOption(
            'check-old',
            'Check old patrol logs (for deleting old format autopatrols).'
        );
        $this->addOption(
            'before',
            'Timestamp to delete only before that time, all MediaWiki timestamp formats are accepted',
            false,
            true
        );
        $this->addOption(
            'from-id',
            'First row (log id) to start updating from',
            false,
            true
        );
        $this->addOption(
            'sleep',
            'Sleep time (in seconds) between every batch',
            false,
            true
        );
        $this->setBatchSize( 1000 );
    }

    public function execute() {
        $this->setBatchSize( $this->getOption( 'batch-size', $this->getBatchSize() ) );

        $sleep = (int)$this->getOption( 'sleep', 10 );
        $fromId = $this->getOption( 'from-id', null );
        $this->countDown( 5 );
        while ( true ) {
            if ( $this->hasOption( 'check-old' ) ) {
                $rowsData = $this->getRowsOld( $fromId );
                // We reached end of the table
                if ( !$rowsData ) {
                    break;
                }
                $rows = $rowsData['rows'];
                $fromId = $rowsData['lastId'];

                // There is nothing to delete in this batch
                if ( !$rows ) {
                    continue;
                }
            } else {
                $rows = $this->getRows( $fromId );
                if ( !$rows ) {
                    break;
                }
                $fromId = end( $rows );
            }

            if ( $this->hasOption( 'dry-run' ) ) {
                $this->output( 'These rows will get deleted: ' . implode( ', ', $rows ) . "\n" );
            } else {
                $this->deleteRows( $rows );
                $this->output( 'Processed up to row id ' . end( $rows ) . "\n" );
            }

            if ( $sleep > 0 ) {
                sleep( $sleep );
            }
        }
    }

    private function getRows( $fromId ) {
        $dbr = MediaWiki\MediaWikiServices::getInstance()->getDBLoadBalancer()->getConnection(
            DB_REPLICA
        );
        $before = $this->getOption( 'before', false );

        $conds = [
            'log_type' => 'patrol',
            'log_action' => 'autopatrol',
        ];

        if ( $fromId ) {
            $conds[] = 'log_id > ' . $dbr->addQuotes( $fromId );
        }

        if ( $before ) {
            $conds[] = 'log_timestamp < ' . $dbr->addQuotes( $dbr->timestamp( $before ) );
        }

        return $dbr->selectFieldValues(
            'logging',
            'log_id',
            $conds,
            __METHOD__,
            [ 'LIMIT' => $this->getBatchSize() ]
        );
    }

    private function getRowsOld( $fromId ) {
        $dbr = MediaWiki\MediaWikiServices::getInstance()->getDBLoadBalancer()->getConnection(
            DB_REPLICA
        );
        $batchSize = $this->getBatchSize();
        $before = $this->getOption( 'before', false );

        $conds = [
            'log_type' => 'patrol',
            'log_action' => 'patrol',
        ];

        if ( $fromId ) {
            $conds[] = 'log_id > ' . $dbr->addQuotes( $fromId );
        }

        if ( $before ) {
            $conds[] = 'log_timestamp < ' . $dbr->addQuotes( $dbr->timestamp( $before ) );
        }

        $result = $dbr->select(
            'logging',
            [ 'log_id', 'log_params' ],
            $conds,
            __METHOD__,
            [ 'LIMIT' => $batchSize ]
        );

        $last = null;
        $autopatrols = [];
        foreach ( $result as $row ) {
            $last = $row->log_id;
            $logEntry = DatabaseLogEntry::newFromRow( $row );
            $params = $logEntry->getParameters();
            if ( !is_array( $params ) ) {
                continue;
            }

            // This logic belongs to PatrolLogFormatter::getMessageKey
            // and LogFormatter::extractParameters the 'auto' value is logically presented as key [5].
            // For legacy case the logical key is index + 3, meaning [2].
            // For the modern case, the logical key is index - 1 meaning [6].
            if ( array_key_exists( '6::auto', $params ) ) {
                // Between 2011-2016 autopatrol logs
                $auto = $params['6::auto'] === true;
            } elseif ( $logEntry->isLegacy() === true && array_key_exists( 2, $params ) ) {
                // Pre-2011 autopatrol logs
                $auto = $params[2] === '1';
            } else {
                continue;
            }

            if ( $auto === true ) {
                $autopatrols[] = $row->log_id;
            }
        }

        if ( $last === null ) {
            return null;
        }

        return [ 'rows' => $autopatrols, 'lastId' => $last ];
    }

    private function deleteRows( array $rows ) {
        $dbw = MediaWiki\MediaWikiServices::getInstance()->getDBLoadBalancer()->getConnection(
            DB_MASTER
        );

        $dbw->delete(
            'logging',
            [ 'log_id' => $rows ],
            __METHOD__
        );

        MediaWiki\MediaWikiServices::getInstance()->getDBLoadBalancerFactory()->waitForReplication();
    }

}

$maintClass = DeleteAutoPatrolLogs::class;
require_once RUN_MAINTENANCE_IF_MAIN;