UnionQueryBuilder.php

Go to the documentation of this file.

<?php
 
namespace Wikimedia\Rdbms;
 
use Wikimedia\Rdbms\Platform\ISQLPlatform;
 
class UnionQueryBuilder {
    private $sqbs = [];
 
    private $db;
 
    private $all = IReadableDatabase::UNION_DISTINCT;
 
    private $options = [];
 
    private $caller = __CLASS__;
 
    public function __construct( IDatabase $db ) {
        $this->db = $db;
    }
 
    public function add( SelectQueryBuilder $selectQueryBuilder ) {
        $this->sqbs[] = $selectQueryBuilder;
        return $this;
    }
 
    public function all() {
        $this->all = $this->db::UNION_ALL;
        return $this;
    }
 
    public function limit( $limit ) {
        if ( !$this->db->unionSupportsOrderAndLimit() ) {
            return $this;
        }
        $this->options['LIMIT'] = $limit;
        return $this;
    }
 
    public function offset( $offset ) {
        if ( !$this->db->unionSupportsOrderAndLimit() ) {
            return $this;
        }
        $this->options['OFFSET'] = $offset;
        return $this;
    }
 
    public function orderBy( $fields, $direction = null ) {
        if ( !$this->db->unionSupportsOrderAndLimit() ) {
            return $this;
        }
        if ( $direction === null ) {
            $this->mergeOption( 'ORDER BY', $fields );
        } elseif ( is_array( $fields ) ) {
            $fieldsWithDirection = [];
            foreach ( $fields as $field ) {
                $fieldsWithDirection[] = "$field $direction";
            }
            $this->mergeOption( 'ORDER BY', $fieldsWithDirection );
        } else {
            $this->mergeOption( 'ORDER BY', "$fields $direction" );
        }
        return $this;
    }
 
    private function mergeOption( $name, $newArrayOrValue ) {
        $value = isset( $this->options[$name] )
            ? (array)$this->options[$name] : [];
        if ( is_array( $newArrayOrValue ) ) {
            $value = array_merge( $value, $newArrayOrValue );
        } else {
            $value[] = $newArrayOrValue;
        }
        $this->options[$name] = $value;
    }
 
    public function caller( $fname ) {
        $this->caller = $fname;
        return $this;
    }
 
    public function fetchResultSet() {
        $sqls = [];
        $tables = [];
        foreach ( $this->sqbs as $sqb ) {
            $sqls[] = $sqb->getSQL();
            $tables = array_merge( $tables, $sqb->getQueryInfo()['tables'] );
        }
        $sql = $this->db->unionQueries( $sqls, $this->all, $this->options );
        $query = new Query( $sql, ISQLPlatform::QUERY_CHANGE_NONE, 'SELECT', $tables );
        return $this->db->query( $query, $this->caller );
    }
}