ReplaceQueryBuilder.php

Go to the documentation of this file.

<?php
 
namespace Wikimedia\Rdbms;
 
use InvalidArgumentException;
use UnexpectedValueException;
 
class ReplaceQueryBuilder {
    private $table = '';
 
    private $rows = [];
 
    private $caller = __CLASS__;
 
    private $uniqueIndexFields = [];
 
    protected $db;
 
    public function __construct( IDatabase $db ) {
        $this->db = $db;
    }
 
    public function connection( IDatabase $db ) {
        if ( $this->db->getType() !== $db->getType() ) {
            throw new InvalidArgumentException(
                __METHOD__ . ' cannot switch to a database of a different type.'
            );
        }
        $this->db = $db;
        return $this;
    }
 
    public function queryInfo( $info ) {
        if ( isset( $info['table'] ) ) {
            $this->table( $info['table'] );
        }
        if ( isset( $info['rows'] ) ) {
            $this->rows( $info['rows'] );
        }
        if ( isset( $info['uniqueIndexFields'] ) ) {
            $this->uniqueIndexFields( (array)$info['uniqueIndexFields'] );
        }
        if ( isset( $info['caller'] ) ) {
            $this->caller( $info['caller'] );
        }
        return $this;
    }
 
    public function table( $table ) {
        $this->table = $table;
        return $this;
    }
 
    public function replaceInto( string $table ) {
        return $this->table( $table );
    }
 
    public function rows( array $rows ) {
        $this->rows = array_merge( $this->rows, $rows );
        return $this;
    }
 
    public function row( array $row ) {
        $this->rows[] = $row;
        return $this;
    }
 
    public function uniqueIndexFields( $uniqueIndexFields ) {
        if ( is_string( $uniqueIndexFields ) ) {
            $uniqueIndexFields = [ $uniqueIndexFields ];
        }
        $this->uniqueIndexFields = $uniqueIndexFields;
        return $this;
    }
 
    public function caller( $fname ) {
        $this->caller = $fname;
        return $this;
    }
 
    public function execute() {
        if ( !$this->rows ) {
            throw new UnexpectedValueException(
                __METHOD__ . ' can\'t have empty $rows value' );
        }
        if ( $this->table === '' ) {
            throw new UnexpectedValueException(
                __METHOD__ . ' expects table not to be empty' );
        }
        if ( !$this->uniqueIndexFields ) {
            throw new UnexpectedValueException(
                __METHOD__ . ' expects unique key to be provided' );
        }
        $this->db->replace( $this->table, [ $this->uniqueIndexFields ], $this->rows, $this->caller );
    }
 
    public function getQueryInfo() {
        $info = [
            'table' => $this->table,
            'rows' => $this->rows,
            'uniqueIndexFields' => $this->uniqueIndexFields,
        ];
        if ( $this->caller !== __CLASS__ ) {
            $info['caller'] = $this->caller;
        }
        return $info;
    }
}