UpdateQueryBuilder.php

Go to the documentation of this file.

<?php
 
namespace Wikimedia\Rdbms;
 
use InvalidArgumentException;
use UnexpectedValueException;
 
// Very long type annotations :(
// phpcs:disable Generic.Files.LineLength
 
class UpdateQueryBuilder {
    private $table = '';
 
    private $set = [];
 
    private $conds = [];
 
    private $caller = __CLASS__;
 
    protected $options = [];
 
    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['set'] ) ) {
            $this->set( $info['set'] );
        }
        if ( isset( $info['conds'] ) ) {
            $this->where( $info['conds'] );
        }
        if ( isset( $info['options'] ) ) {
            $this->options( (array)$info['options'] );
        }
        if ( isset( $info['caller'] ) ) {
            $this->caller( $info['caller'] );
        }
        return $this;
    }
 
    public function table( $table ) {
        $this->table = $table;
        return $this;
    }
 
    public function update( string $table ) {
        return $this->table( $table );
    }
 
    public function option( $name, $value = null ) {
        if ( $value === null ) {
            $this->options[] = $name;
        } else {
            $this->options[$name] = $value;
        }
        return $this;
    }
 
    public function options( array $options ) {
        $this->options = array_merge( $this->options, $options );
        return $this;
    }
 
    public function where( $conds ) {
        if ( is_array( $conds ) ) {
            foreach ( $conds as $key => $cond ) {
                if ( is_int( $key ) ) {
                    $this->conds[] = $cond;
                } elseif ( isset( $this->conds[$key] ) ) {
                    // @phan-suppress-previous-line PhanTypeMismatchDimFetch
                    // T288882
                    $this->conds[] = $this->db->makeList(
                        [ $key => $cond ], IDatabase::LIST_AND );
                } else {
                    $this->conds[$key] = $cond;
                }
            }
        } else {
            $this->conds[] = $conds;
        }
        return $this;
    }
 
    public function andWhere( $conds ) {
        return $this->where( $conds );
    }
 
    public function conds( $conds ) {
        return $this->where( $conds );
    }
 
    public function set( $set ) {
        if ( is_array( $set ) ) {
            foreach ( $set as $key => $value ) {
                if ( is_int( $key ) ) {
                    $this->set[] = $value;
                } else {
                    $this->set[$key] = $value;
                }
            }
        } else {
            $this->set[] = $set;
        }
        return $this;
    }
 
    public function andSet( $set ) {
        return $this->set( $set );
    }
 
    public function ignore() {
        $this->options[] = 'IGNORE';
        return $this;
    }
 
    public function caller( $fname ) {
        $this->caller = $fname;
        return $this;
    }
 
    public function execute(): void {
        if ( !$this->conds ) {
            throw new UnexpectedValueException(
                __METHOD__ . ' expects at least one condition to be set' );
        }
        if ( !$this->set ) {
            throw new UnexpectedValueException(
                __METHOD__ . ' can\t have empty $set value' );
        }
        if ( $this->table === '' ) {
            throw new UnexpectedValueException(
                __METHOD__ . ' expects table not to be empty' );
        }
        $this->db->update( $this->table, $this->set, $this->conds, $this->caller, $this->options );
    }
 
    public function getQueryInfo() {
        $info = [
            'table' => $this->table,
            'set' => $this->set,
            'conds' => $this->conds,
            'options' => $this->options,
        ];
        if ( $this->caller !== __CLASS__ ) {
            $info['caller'] = $this->caller;
        }
        return $info;
    }
}