BatchRowUpdate.php

Go to the documentation of this file.

<?php
namespace MediaWiki\Utils;
 
class BatchRowUpdate {
    protected $reader;
 
    protected $writer;
 
    protected $generator;
 
    protected $output;
 
    public function __construct(
        BatchRowIterator $reader, BatchRowWriter $writer, RowUpdateGenerator $generator
    ) {
        $this->reader = $reader;
        $this->writer = $writer;
        $this->generator = $generator;
        $this->output = static function ( $text ) {
        }; // nop
    }
 
    public function execute() {
        foreach ( $this->reader as $rows ) {
            $updates = [];
            foreach ( $rows as $row ) {
                $update = $this->generator->update( $row );
                if ( $update ) {
                    $updates[] = [
                        'primaryKey' => $this->reader->extractPrimaryKeys( $row ),
                        'changes' => $update,
                    ];
                }
            }
 
            if ( $updates ) {
                $this->output( "Processing " . count( $updates ) . " rows\n" );
                $this->writer->write( $updates );
            }
        }
 
        $this->output( "Completed\n" );
    }
 
    public function setOutput( callable $output ) {
        $this->output = $output;
    }
 
    protected function output( $text ) {
        ( $this->output )( $text );
    }
}
 
class_alias( BatchRowUpdate::class, 'BatchRowUpdate' );