Added Support for Virtual Tables

Author: rkrx

src/Builder/Traits/TableNameBuilder.php

@@ -1,7 +1,7 @@
 <?php
 namespace Kir\MySQL\Builder\Traits;
 
-use Kir\MySQL\Database;
+use Kir\MySQL\Databases\MySQL;
 
 trait TableNameBuilder {
 	use AbstractAliasReplacer;
@@ -32,6 +32,10 @@ protected function buildTableName($alias, $name) {
 			}
 			$name = '(' . join("\n\tUNION\n\t", $parts) . ')';
 		}
+		if($this->db()->getVirtualTables()->has($name)) {
+			$select = (string )$this->db()->getVirtualTables()->get($name);
+			$name = sprintf('(%s)', join("\n\t", explode("\n", trim($select))));
+		}
 		$name = $this->aliasReplacer()->replace($name);
 		if($alias !== null) {
 			return sprintf("%s %s", $name, $alias);
@@ -40,7 +44,7 @@ protected function buildTableName($alias, $name) {
 	}
 
 	/**
-	 * @return Database
+	 * @return MySQL
 	 */
 	abstract protected function db();
 }

src/Databases/MySQL.php

@@ -12,6 +12,7 @@
 use Kir\MySQL\Databases\MySQL\MySQLExceptionInterpreter;
 use Kir\MySQL\QueryLogger\QueryLoggers;
 use Kir\MySQL\Tools\AliasRegistry;
+use Kir\MySQL\Tools\VirtualTables;
 
 /**
  */
@@ -28,6 +29,8 @@ class MySQL implements Database {
 	private $transactionLevel = 0;
 	/** @var QueryLoggers */
 	private $queryLoggers = 0;
+	/** @var VirtualTables */
+	private $virtualTables = null;
 	/** @var MySQLExceptionInterpreter */
 	private $exceptionInterpreter = 0;
 	/** @var array */
@@ -68,6 +71,16 @@ public function getAliasRegistry() {
 		return $this->aliasRegistry;
 	}
 
+	/**
+	 * @return VirtualTables
+	 */
+	public function getVirtualTables() {
+		if($this->virtualTables === null) {
+			$this->virtualTables = new VirtualTables();
+		}
+		return $this->virtualTables;
+	}
+
 	/**
 	 * @param string $query
 	 * @throws Exception
@@ -279,11 +292,9 @@ public function transactionRollback() {
 	 * @return mixed
 	 * @throws \Exception
 	 * @throws \Error
-	 * @throws null
 	 */
 	public function dryRun($callback = null) {
 		$result = null;
-		$exception = null;
 		if(!$this->pdo->inTransaction()) {
 			$this->transactionStart();
 			try {

src/Tools/VirtualTables.php

@@ -0,0 +1,38 @@
+<?php
+namespace Kir\MySQL\Tools;
+
+use Kir\MySQL\Builder\Select;
+
+class VirtualTables {
+	/** @var Select[] */
+	private $virtualTables = [];
+	
+	/**
+	 * @param string $tableName
+	 * @param Select $select
+	 * @return $this
+	 */
+	public function add($tableName, Select $select) {
+		$this->virtualTables[$tableName] = $select;
+		return $this;
+	}
+	
+	/**
+	 * @param string $tableName
+	 * @return bool
+	 */
+	public function has($tableName) {
+		return array_key_exists($tableName, $this->virtualTables);
+	}
+	
+	/**
+	 * @param string $tableName
+	 * @return Select|null
+	 */
+	public function get($tableName) {
+		if($this->has($tableName)) {
+			return $this->virtualTables[$tableName];
+		}
+		return null;
+	}
+}

tests/Builder/SelectTest.php

@@ -341,4 +341,28 @@ public function testSortSpecification() {
 
 		$this->assertEquals("SELECT\n\tt.field1,\n\tt.field2\nFROM\n\ttest t\nORDER BY\n\tREVERSE(t.field2) ASC,\n\tfield1 DESC\n", $query);
 	}
+
+	public function testVirtualTables() {
+		$vt1 = TestSelect::create()
+		->field('a.field1')
+		->from('a', 'tableA');
+		
+		$vt2 = TestSelect::create()
+		->field('a.field1')
+		->from('a', 'tableA');
+		
+		$db = new TestDB();
+		$db->getVirtualTables()->add('virt_table1', $vt1);
+		$db->getVirtualTables()->add('virt_table2', $vt2);
+		
+		$query = TestSelect::create($db)
+		->field('t.field1')
+		->field('t.field2')
+		->from('t', 'test')
+		->joinInner('vt1', 'virt_table1', 'vt1.field1=t.field1')
+		->joinInner('vt2', 'virt_table2', 'vt2.field2=t.field2')
+		->asString();
+
+		$this->assertEquals("SELECT\n\tt.field1,\n\tt.field2\nFROM\n\ttest t\nINNER JOIN\n\t(SELECT\n\t\ta.field1\n\tFROM\n\t\ttableA a) vt1 ON vt1.field1=t.field1\nINNER JOIN\n\t(SELECT\n\t\ta.field1\n\tFROM\n\t\ttableA a) vt2 ON vt2.field2=t.field2\n", $query);
+	}
 }

tests/Builder/SelectTest/TestSelect.php

@@ -2,14 +2,18 @@
 namespace Kir\MySQL\Builder\SelectTest;
 
 use Kir\MySQL\Builder\RunnableSelect;
+use Kir\MySQL\Database;
 use Kir\MySQL\Databases\TestDB;
 
 class TestSelect extends RunnableSelect {
 	/**
+	 * @param Database|null $db
 	 * @return $this
 	 */
-	public static function create() {
-		$db = new TestDB();
+	public static function create(Database $db = null) {
+		if($db === null) {
+			$db = new TestDB();
+		}
 		return new static($db);
 	}
 
@@ -19,4 +23,4 @@ public static function create() {
 	public function asString() {
 		return $this->__toString();
 	}
-}
+}

tests/Builder/UpdateTest.php

@@ -74,10 +74,12 @@ public function testSetAll1() {
 	public function testSetAll2() {
 		$db = Phake::mock('Kir\\MySQL\\Databases\\MySQL');
 		$reg = Phake::mock('Kir\\MySQL\\Tools\\AliasRegistry');
+		$vt = Phake::mock('Kir\\MySQL\\Tools\\VirtualTables');
 		Phake::when($db)->__call('getTableFields', ['test1'])->thenReturn(['field1', 'field2']);
 		Phake::when($db)->__call('quoteField', [Phake::anyParameters()])->thenGetReturnByLambda(function ($fieldName) { return "`{$fieldName}`"; });
 		Phake::when($db)->__call('quote', [Phake::anyParameters()])->thenGetReturnByLambda(function ($value) { return "'{$value}'"; });
 		Phake::when($db)->__call('getAliasRegistry', [])->thenReturn($reg);
+		Phake::when($db)->__call('getVirtualTables', [])->thenReturn($vt);
 		$sql = (new TestUpdate($db))
 		->table('test1')
 		->setAll(['field1' => 1, 'field2' => 2, 'field3' => 3])

tests/Builder/UpdateTest/TestUpdate.php

@@ -9,13 +9,7 @@ class TestUpdate extends Update {
 	 * @return $this
 	 */
 	public static function create() {
-		$functions = array(
-			'getTableFields' => function ($tableName) {
-				return array();
-			}
-		);
-
-		$db = new TestDB($functions);
+		$db = new TestDB();
 		return new static($db);
 	}
 
@@ -25,4 +19,4 @@ public static function create() {
 	public function asString() {
 		return $this->__toString();
 	}
-}
+}