BatchRowWriter.php

Go to the documentation of this file.

<?php
use MediaWiki\MediaWikiServices;
use Wikimedia\Rdbms\IDatabase;
 
class BatchRowWriter {
    protected $db;
 
    protected $table;
 
    protected $clusterName;
 
    protected $caller;
 
    public function __construct( IDatabase $db, $table, $clusterName = false ) {
        $this->db = $db;
        $this->table = $table;
        $this->clusterName = $clusterName;
    }
 
    public function setCaller( $caller ) {
        $this->caller = $caller;
        return $this;
    }
 
    public function write( array $updates ) {
        $lbFactory = MediaWikiServices::getInstance()->getDBLoadBalancerFactory();
        $ticket = $lbFactory->getEmptyTransactionTicket( __METHOD__ );
 
        $caller = __METHOD__;
        if ( (string)$this->caller !== '' ) {
            $caller .= " (for {$this->caller})";
        }
 
        foreach ( $updates as $update ) {
            $this->db->newUpdateQueryBuilder()
                ->update( $this->table )
                ->set( $update['changes'] )
                ->where( $update['primaryKey'] )
                ->caller( $caller )->execute();
        }
 
        $lbFactory->commitAndWaitForReplication( __METHOD__, $ticket );
    }
}