1 files changed, 61 insertions, 0 deletions
diff --git a/vendor/ipl/orm/src/UnionQuery.php b/vendor/ipl/orm/src/UnionQuery.php
new file mode 100644
index 0000000..6f3823d
--- /dev/null
+++ b/vendor/ipl/orm/src/UnionQuery.php
@@ -0,0 +1,61 @@
+<?php
+
+namespace ipl\Orm;
+
+use ipl\Sql\Select;
+
+class UnionQuery extends Query
+{
+    /** @var Query[] Underlying queries */
+    private $unions;
+
+    /**
+     * Get the underlying queries
+     *
+     * @return Query[]
+     */
+    public function getUnions()
+    {
+        if ($this->unions === null) {
+            $this->unions = [];
+
+            /** @var UnionModel $model */
+            $model = $this->getModel();
+            foreach ($model->getUnions() as list($target, $relations, $columns)) {
+                $query = (new Query())
+                    ->setDb($this->getDb())
+                    ->setModel(new $target())
+                    ->columns($columns)
+                    ->disableDefaultSort()
+                    ->with($relations);
+
+                $this->unions[] = $query;
+            }
+        }
+
+        return $this->unions;
+    }
+
+    public function getSelectBase()
+    {
+        if ($this->selectBase === null) {
+            $this->selectBase = new Select();
+        }
+
+        $union = new Select();
+
+        foreach ($this->getUnions() as $query) {
+            $select = $query->assembleSelect();
+            $columns = $select->getColumns();
+            $select->resetColumns();
+            ksort($columns);
+            $select->columns($columns);
+
+            $union->unionAll($select);
+        }
+
+        $this->selectBase->from([$this->getModel()->getTableName() => $union]);
+
+        return $this->selectBase;
+    }
+}