unioncondition.php

<?php
namespace Bitrix\Main\ORM\Query;
 
use Bitrix\Main\ArgumentException;
use Bitrix\Main\DB\SqlExpression;
 
class UnionCondition
{
    protected $subQuery;
 
    protected $all;
 
    public function __construct($subQuery, $unionAll = false)
    {
        if (!($subQuery instanceof Query) && !($subQuery instanceof SqlExpression))
        {
            throw new ArgumentException("Query or SqlExpression expected, `".gettype($subQuery)."` found.");
        }
 
        $this->subQuery = $subQuery;
        $this->all = $unionAll;
    }
 
    public function getSubQuery()
    {
        return $this->subQuery;
    }
 
    public function setSubQuery($subQuery)
    {
        $this->subQuery = $subQuery;
    }
 
    public function isAll()
    {
        return $this->all;
    }
 
    public function setAll($all)
    {
        $this->all = $all;
    }
 
    public function getSql($forceObjectPrimary = false)
    {
        $sql = "UNION ";
 
        if ($this->all)
        {
            $sql .= "ALL ";
        }
 
        $subQuerySql = $this->getSubQuerySql($forceObjectPrimary);
 
        if (preg_match('/(\sorder\s+by\s|\slimit\s+\d+)/i', $subQuerySql))
        {
            $subQuerySql = "({$subQuerySql})";
        }
 
        return $sql . $subQuerySql;
    }
 
    public function getSubQuerySql($forceObjectPrimary = false)
    {
        if ($this->subQuery instanceof Query)
        {
            return $this->subQuery->getQuery($forceObjectPrimary);
        }
        elseif ($this->subQuery instanceof SqlExpression)
        {
            return $this->subQuery->compile();
        }
 
        return null;
    }
}