union.php

<?php
namespace Bitrix\Main\ORM\Query;
use Bitrix\Main\ArgumentException;
use Bitrix\Main\DB\Connection;
 
class Union
{
    protected $queries;
 
    protected $order;
 
    protected $limit;
 
    protected $offset;
 
    protected $connection;
 
    public function __construct(Connection $connection)
    {
        $this->connection = $connection;
    }
 
    public function addQuery(UnionCondition $query)
    {
        $this->queries[] = $query;
        return $this;
    }
 
    public function getQueries()
    {
        return $this->queries;
    }
 
    public function setOrder($order)
    {
        $this->order = array();
 
        if (!is_array($order))
        {
            $order = array($order);
        }
 
        foreach ($order as $k => $v)
        {
            if (is_numeric($k))
            {
                $this->addOrder($v);
            }
            else
            {
                $this->addOrder($k, $v);
            }
        }
 
        return $this;
    }
 
    public function addOrder($definition, $order = 'ASC')
    {
        $order = strtoupper($order);
 
        if (!in_array($order, array('ASC', 'DESC'), true))
        {
            throw new ArgumentException(sprintf('Invalid order "%s"', $order));
        }
 
        $helper = $this->connection->getSqlHelper();
 
        if ($order == 'ASC')
        {
            $order = $helper->getAscendingOrder();
        }
        else
        {
            $order = $helper->getDescendingOrder();
        }
 
        $this->order[$definition] = $order;
 
        return $this;
    }
 
    public function getOrder()
    {
        return $this->order;
    }
 
    public function setLimit($limit)
    {
        $this->limit = $limit;
        return $this;
    }
 
    public function getLimit()
    {
        return $this->limit;
    }
 
    public function setOffset($offset)
    {
        $this->offset = $offset;
        return $this;
    }
 
    public function getOffset()
    {
        return $this->offset;
    }
}