Database/SQLiteDriver.php at master · slavei/Database · GitHub

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
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
<?php
/**
 * @package    SugiPHP
 * @subpackage Database
 * @author     Plamen Popov <tzappa@gmail.com>
 * @license    http://opensource.org/licenses/mit-license.php (MIT License)
 */

namespace SugiPHP\Database;

class SQLiteDriver implements DriverInterface
{
	/**
	 * SQLite3 handle
	 *
	 * @var object
	 */
	protected $dbHandle;

	/**
	 * Database connection parameters.
	 *
	 * @var array
	 */
	protected $params = array();

	/**
	 * Constructor.
	 *
	 * @param mixed $database
	 */
	public function __construct($database)
	{
		if (is_string($database)) {
			$this->params["database"] = $database;
		} elseif (is_array($database)) {
			$this->params = $database;
		} else {
			$this->setHandle($database);
		}
	}

	/**
	 * {@inheritdoc}
	 */
	public function open()
	{
		// if we have a SQLite database handle (connection)
		if ($this->dbHandle) {
			return ;
		}

		// Database parameter is mandatory
		if (empty($this->params["database"])) {
			throw new Exception("Database parameter is missing", "internal_error");
		}

		// Establish connection
		try {
			$this->dbHandle = new \SQLite3($this->params["database"]);
		} catch (\Exception $e) {
			throw new Exception($e->getMessage(), "connection_error");
		}
	}

	/**
	 * {@inheritdoc}
	 */
	public function close()
	{
		if ($this->dbHandle) {
			$this->dbHandle->close();
			$this->dbHandle = null;
		}
	}

	/**
	 * {@inheritdoc}
	 */
	public function escape($param)
	{
		return $this->dbHandle->escapeString($param);
	}

	/**
	 * {@inheritdoc}
	 */
	public function query($sql)
	{
		// additional warning is triggered
		return @$this->dbHandle->query($sql);
	}

	/**
	 * {@inheritdoc}
	 */
	public function fetch($res)
	{
		return $res->fetchArray(SQLITE3_ASSOC);
	}

	/**
	 * {@inheritdoc}
	 */
	public function affected($res)
	{
		return $this->dbHandle->changes();
	}

	/**
	 * {@inheritdoc}
	 */
	public function lastId()
	{
		return $this->dbHandle->lastInsertRowID();
	}

	/**
	 * {@inheritdoc}
	 */
	public function free($res)
	{
		return $res->finalize();
	}

	/**
	 * {@inheritdoc}
	 */
	public function error()
	{
		return $this->dbHandle->lastErrorMsg();
	}

	/**
	 * {@inheritdoc}
	 */
	public function getHandle()
	{
		return $this->dbHandle;
	}

	/**
	 * {@inheritdoc}
	 */
	public function setHandle($dbHandle)
	{
		if (is_object($dbHandle) and ($dbHandle instanceof \SQLite3)) {
			$this->dbHandle = $dbHandle;
		} else {
			throw new Exception("Handle must be SQLite3 object", "internal_error");
		}
	}
}