Source of file Driver.php

Size: 3,078 Bytes - Last Modified: 2018-01-25T14:51:55+00:00

src/Drivers/Sqlite/Driver.php

123456789101112131415161718192021222324252627282930313233343536373839404142
Covered by 7 test(s):
  • Query\Tests\ConnectionManagerTest::testConnect
  • Query\Tests\ConnectionManagerTest::testGetConnection
  • Query\Tests\QueryParserTest::testGeneric
  • Query\Tests\QueryParserTest::testGeneric2
  • Query\Tests\QueryParserTest::testWUnderscore
  • Query\Tests\QueryParserTest::testFunction
  • Query\Tests\Drivers\SQLite\SQLiteDriverTest::testConnection
4344
Covered by 3 test(s):
  • Query\Tests\ConnectionManagerTest::testConnect
  • Query\Tests\ConnectionManagerTest::testGetConnection
  • Query\Tests\Drivers\SQLite\SQLiteDriverTest::testConnection
454647
Covered by 7 test(s):
  • Query\Tests\ConnectionManagerTest::testConnect
  • Query\Tests\ConnectionManagerTest::testGetConnection
  • Query\Tests\QueryParserTest::testGeneric
  • Query\Tests\QueryParserTest::testGeneric2
  • Query\Tests\QueryParserTest::testWUnderscore
  • Query\Tests\QueryParserTest::testFunction
  • Query\Tests\Drivers\SQLite\SQLiteDriverTest::testConnection
48495051525354555657
Covered by 3 test(s):
  • Query\Tests\Drivers\SQLite\SQLiteDriverTest::testCreateTable
  • Query\Tests\Drivers\SQLite\SQLiteDriverTest::testDeleteTable
  • Query\Tests\Drivers\SQLite\SQLiteDriverTest::testGetTables
58
Covered by 3 test(s):
  • Query\Tests\Drivers\SQLite\SQLiteDriverTest::testCreateTable
  • Query\Tests\Drivers\SQLite\SQLiteDriverTest::testDeleteTable
  • Query\Tests\Drivers\SQLite\SQLiteDriverTest::testGetTables
59
Covered by 3 test(s):
  • Query\Tests\Drivers\SQLite\SQLiteDriverTest::testCreateTable
  • Query\Tests\Drivers\SQLite\SQLiteDriverTest::testDeleteTable
  • Query\Tests\Drivers\SQLite\SQLiteDriverTest::testGetTables
6061626364656667686970
Covered by 1 test(s):
  • Query\Tests\Drivers\SQLite\SQLiteDriverTest::testGetFKs
7172
Covered by 1 test(s):
  • Query\Tests\Drivers\SQLite\SQLiteDriverTest::testGetFKs
7374
Covered by 1 test(s):
  • Query\Tests\Drivers\SQLite\SQLiteDriverTest::testGetFKs
75
Covered by 1 test(s):
  • Query\Tests\Drivers\SQLite\SQLiteDriverTest::testGetFKs
76
Covered by 1 test(s):
  • Query\Tests\Drivers\SQLite\SQLiteDriverTest::testGetFKs
77
Covered by 1 test(s):
  • Query\Tests\Drivers\SQLite\SQLiteDriverTest::testGetFKs
78
Covered by 1 test(s):
  • Query\Tests\Drivers\SQLite\SQLiteDriverTest::testGetFKs
79
Covered by 1 test(s):
  • Query\Tests\Drivers\SQLite\SQLiteDriverTest::testGetFKs
80818283
Covered by 1 test(s):
  • Query\Tests\Drivers\SQLite\SQLiteDriverTest::testGetFKs
84858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134
<?php declare(strict_types=1);
/**
 * Query
 *
 * SQL Query Builder / Database Abstraction Layer
 *
 * PHP version 7.1
 *
 * @package     Query
 * @author      Timothy J. Warren <tim@timshomepage.net>
 * @copyright   2012 - 2018 Timothy J. Warren
 * @license     http://www.opensource.org/licenses/mit-license.html  MIT License
 * @link        https://git.timshomepage.net/aviat4ion/Query
 */
namespace Query\Drivers\Sqlite;

use PDO;
use Query\Drivers\AbstractDriver;

/**
 * SQLite specific class
 */
class Driver extends AbstractDriver {

	/**
	 * SQLite has a truncate optimization,
	 * but no support for the actual keyword
	 * @var boolean
	 */
	protected $hasTruncate = FALSE;

	/**
	 * Open SQLite Database
	 *
	 * @param string $dsn
	 * @param string $user
	 * @param string $pass
	 * @param array $driverOptions
	 */
	public function __construct(string $dsn, string $user=NULL, string $pass=NULL, array $driverOptions=[])
	{
		if (strpos($dsn, 'sqlite:') === FALSE)
		{
			$dsn = "sqlite:{$dsn}";
		}

		parent::__construct($dsn, $user, $pass);
	}

	/**
	 * List tables for the current database
	 *
	 * @return mixed
	 */
	public function getTables(): array
	{
		$sql = $this->sql->tableList();
		$res = $this->query($sql);
		return dbFilter($res->fetchAll(PDO::FETCH_ASSOC), 'name');
	}

	/**
	 * Retrieve foreign keys for the table
	 *
	 * @param string $table
	 * @return array
	 */
	public function getFks($table): array
	{
		$returnRows = [];

		foreach(parent::getFks($table) as $row)
		{
			$returnRows[] = [
				'child_column' => $row['from'],
				'parent_table' => $row['table'],
				'parent_column' => $row['to'],
				'update' => $row['on_update'],
				'delete' => $row['on_delete']
			];
		}

		return $returnRows;
	}

	/**
	 * Create sql for batch insert
	 *
	 * @codeCoverageIgnore
	 * @param string $table
	 * @param array $data
	 * @return array
	 */
	public function insertBatch(string $table, array $data=[]): array
	{
		// If greater than version 3.7.11, supports the same syntax as
		// MySQL and Postgres
		if (version_compare($this->getAttribute(PDO::ATTR_SERVER_VERSION), '3.7.11', '>='))
		{
			return parent::insertBatch($table, $data);
		}

		// --------------------------------------------------------------------------
		// Otherwise, do a union query as an analogue to a 'proper' batch insert
		// --------------------------------------------------------------------------

		// Each member of the data array needs to be an array
		if ( ! \is_array(current($data)))
		{
			return NULL;
		}

		// Start the block of sql statements
		$table = $this->quoteTable($table);
		$sql = "INSERT INTO {$table} \n";

		// Create a key-value mapping for each field
		$first = array_shift($data);
		$cols = [];
		foreach($first as $colname => $datum)
		{
			$cols[] = $this->_quote($datum) . ' AS ' . $this->quoteIdent($colname);
		}
		$sql .= 'SELECT ' . implode(', ', $cols) . "\n";

		foreach($data as $union)
		{
			$vals = array_map([$this, 'quote'], $union);
			$sql .= 'UNION SELECT ' . implode(',', $vals) . "\n";
		}

		return [$sql, NULL];
	}
}