UnionExpression.php 1.21 KB
Newer Older
Ketan's avatar
Ketan committed
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61
<?php
/**
 * Copyright © Magento, Inc. All rights reserved.
 * See COPYING.txt for license details.
 */
namespace Magento\Framework\DB\Sql;

use Magento\Framework\DB\Select;

/**
 * Class UnionExpression
 */
class UnionExpression extends Expression
{
    /**
     * @var Select[]
     */
    protected $parts;

    /**
     * @var string
     */
    protected $type;

    /**
     * @var string
     */
    protected $pattern;

    /**
     * @param Select[] $parts
     * @param string $type (optional)
     * @param string $pattern (optional)
     */
    public function __construct(array $parts, $type = Select::SQL_UNION, $pattern = '')
    {
        $this->parts = $parts;
        $this->type = $type;
        $this->pattern = $pattern;
    }

    /**
     * @inheritdoc
     */
    public function __toString()
    {
        $parts = [];
        foreach ($this->parts as $part) {
            if ($part instanceof Select) {
                $parts[] = sprintf('(%s)', $part->assemble());
            } else {
                $parts[] = $part;
            }
        }
        $sql = implode($parts, $this->type);
        if ($this->pattern) {
            return sprintf($this->pattern, $sql);
        }
        return $sql;
    }
}