- Added union and unionAll to Kir\MySQL\Builder\Select

Author: rkrx

src/Builder/Select.php

@@ -9,6 +9,7 @@
 use Kir\MySQL\Builder\Traits\JoinBuilder;
 use Kir\MySQL\Builder\Traits\LimitBuilder;
 use Kir\MySQL\Builder\Traits\TableNameBuilder;
+use Kir\MySQL\Builder\Traits\UnionBuilder;
 use Kir\MySQL\Builder\Traits\WhereBuilder;
 
 class Select extends Statement {
@@ -21,18 +22,13 @@ class Select extends Statement {
 	use OrderByBuilder;
 	use LimitBuilder;
 	use OffsetBuilder;
+	use UnionBuilder;
 
-	/**
-	 * @var string[]
-	 */
+	/** @var string[] */
 	private $fields = array();
-	/**
-	 * @var bool
-	 */
+	/** @var bool */
 	private $calcFoundRows = false;
-	/**
-	 * @var bool
-	 */
+	/** @var bool */
 	private $forUpdate = false;
 
 	/**
@@ -137,6 +133,7 @@ public function __toString() {
 		$query = $this->buildOrder($query);
 		$query = $this->buildLimit($query);
 		$query = $this->buildOffset($query);
+		$query = $this->buildUnions($query);
 		$query = $this->buildForUpdate($query);
 		return $query;
 	}

src/Builder/Traits/UnionBuilder.php

@@ -0,0 +1,44 @@
+<?php
+namespace Kir\MySQL\Builder\Traits;
+
+trait UnionBuilder {
+	use AbstractDB;
+
+	/** @var array */
+	private $unions = [];
+
+	/**
+	 * @param string $query
+	 * @return $this
+	 */
+	public function union($query) {
+		$this->unions[] = array('', $query);
+		return $this;
+	}
+
+	/**
+	 * @param string $query
+	 * @return $this
+	 */
+	public function unionAll($query) {
+		$this->unions[] = array('ALL', $query);
+		return $this;
+	}
+
+	/**
+	 * @param string $query
+	 * @return string
+	 */
+	protected function buildUnions($query) {
+		$qaueries = [$query];
+		foreach($this->unions as $unionQuery) {
+			if($unionQuery[0] === 'ALL') {
+				$qaueries[] = 'UNION ALL';
+			} else {
+				$qaueries[] = 'UNION';
+			}
+			$qaueries[] = $unionQuery[1];
+		}
+		return join("\n", $qaueries);
+	}
+}

tests/Builder/SelectTest.php

@@ -174,4 +174,22 @@ public function testSubselectAsField() {
 
 		$this->assertEquals("SELECT\n\t(\n\t\tSELECT\n\t\t\tCOUNT(*)\n\t\tFROM\n\t\t\ttest1 t1\n\t\tINNER JOIN\n\t\t\ttest2 t2 ON t1.id=t2.id\n\t\tWHERE\n\t\t\t(t1.id > 10)\n\t) AS `testfield`\nFROM\n\ttest1 t1\nINNER JOIN\n\ttest2 t2 ON t1.id=t2.id\nWHERE\n\t(t1.id > 10)\n", $query);
 	}
-}
+
+	public function testSubselectUnion() {
+		$query = TestSelect::create()
+		->field('t1.field')
+		->from('t1', 'test1')
+		->joinInner('t2', 'test2', 't1.id=t2.id')
+		->where('t1.id > 10');
+
+		$query = TestSelect::create()
+		->union($query)
+		->field('t1.field')
+		->from('t1', 'test1')
+		->joinInner('t2', 'test2', 't1.id=t2.id')
+		->where('t1.id > 10')
+		->asString();
+
+		$this->assertEquals("SELECT\n\tt1.field\nFROM\n\ttest1 t1\nINNER JOIN\n\ttest2 t2 ON t1.id=t2.id\nWHERE\n\t(t1.id > 10)\n\nUNION\nSELECT\n\tt1.field\nFROM\n\ttest1 t1\nINNER JOIN\n\ttest2 t2 ON t1.id=t2.id\nWHERE\n\t(t1.id > 10)\n", $query);
+	}
+}