@@ -15,33 +15,48 @@ discard block |
||
15 | 15 | |
16 | 16 | public function query(\Maphper\Lib\Query $query) { |
17 | 17 | $queryId = md5($query->getSql()); |
18 | - if (isset($this->queryCache[$queryId])) $stmt = $this->queryCache[$queryId]; |
|
19 | - else { |
|
18 | + if (isset($this->queryCache[$queryId])) { |
|
19 | + $stmt = $this->queryCache[$queryId]; |
|
20 | + } else { |
|
20 | 21 | $stmt = $this->pdo->prepare($query->getSql(), [\PDO::ATTR_CURSOR => \PDO::CURSOR_FWDONLY]); |
21 | - if ($stmt) $this->queryCache[$queryId] = $stmt; |
|
22 | + if ($stmt) { |
|
23 | + $this->queryCache[$queryId] = $stmt; |
|
24 | + } |
|
22 | 25 | } |
23 | 26 | |
24 | 27 | $args = $query->getArgs(); |
25 | - foreach ($args as &$arg) if ($arg instanceof \DateTime) { |
|
28 | + foreach ($args as &$arg) { |
|
29 | + if ($arg instanceof \DateTime) { |
|
26 | 30 | if ($arg->format('H:i:s') == '00:00:00') $arg = $arg->format('Y-m-d'); |
27 | - else $arg = $arg->format('Y-m-d H:i:s'); |
|
31 | + } |
|
32 | + else { |
|
33 | + $arg = $arg->format('Y-m-d H:i:s'); |
|
34 | + } |
|
28 | 35 | } |
29 | 36 | |
30 | 37 | if ($stmt !== false) { |
31 | 38 | try { |
32 | - if (count($args) > 0) $res = $stmt->execute($args); |
|
33 | - else $res = $stmt->execute(); |
|
34 | - if ($stmt->errorCode() !== '00000') throw new \Exception('Invalid query'); |
|
35 | - if (substr($query->getSql(), 0, 6) === 'SELECT') return $stmt->fetchAll(\PDO::FETCH_OBJ); |
|
36 | - else return $stmt; |
|
37 | - } |
|
38 | - catch (\Exception $e) { |
|
39 | + if (count($args) > 0) { |
|
40 | + $res = $stmt->execute($args); |
|
41 | + } else { |
|
42 | + $res = $stmt->execute(); |
|
43 | + } |
|
44 | + if ($stmt->errorCode() !== '00000') { |
|
45 | + throw new \Exception('Invalid query'); |
|
46 | + } |
|
47 | + if (substr($query->getSql(), 0, 6) === 'SELECT') { |
|
48 | + return $stmt->fetchAll(\PDO::FETCH_OBJ); |
|
49 | + } else { |
|
50 | + return $stmt; |
|
51 | + } |
|
52 | + } catch (\Exception $e) { |
|
39 | 53 | //SQLite causes an error if when the DB schema changes, rebuild $stmt and try again. |
40 | 54 | if ($stmt->errorInfo()[2] == 'database schema has changed') { |
41 | 55 | unset($this->queryCache[$queryId]); |
42 | 56 | return $this->query($query); |
57 | + } else { |
|
58 | + return $stmt; |
|
43 | 59 | } |
44 | - else return $stmt; |
|
45 | 60 | } |
46 | 61 | } |
47 | 62 | //Handle SQLite when PDO_ERRMODE is set to SILENT |
@@ -55,12 +70,19 @@ discard block |
||
55 | 70 | } |
56 | 71 | |
57 | 72 | private function getType($val) { |
58 | - if ($val instanceof \DateTime) return 'DATETIME'; |
|
59 | - else if (is_int($val)) return 'INTEGER'; |
|
60 | - else if (is_double($val)) return 'DECIMAL(9,' . strlen($val) - strrpos($val, '.') - 1 . ')'; |
|
61 | - else if (is_string($val) && strlen($val) < 256) return 'VARCHAR(255)'; |
|
62 | - else if (is_string($val) && strlen($val) > 256) return 'LONGBLOG'; |
|
63 | - else return 'VARCHAR(255)'; |
|
73 | + if ($val instanceof \DateTime) { |
|
74 | + return 'DATETIME'; |
|
75 | + } else if (is_int($val)) { |
|
76 | + return 'INTEGER'; |
|
77 | + } else if (is_double($val)) { |
|
78 | + return 'DECIMAL(9,' . strlen($val) - strrpos($val, '.') - 1 . ')'; |
|
79 | + } else if (is_string($val) && strlen($val) < 256) { |
|
80 | + return 'VARCHAR(255)'; |
|
81 | + } else if (is_string($val) && strlen($val) > 256) { |
|
82 | + return 'LONGBLOG'; |
|
83 | + } else { |
|
84 | + return 'VARCHAR(255)'; |
|
85 | + } |
|
64 | 86 | } |
65 | 87 | |
66 | 88 | private function tableExists($name) { |
@@ -94,8 +116,7 @@ discard block |
||
94 | 116 | $this->pdo->query('INSERT INTO ' . $this->quote($table . $affix) . '(' . $columns . ') SELECT ' . $columns . ' FROM ' . $this->quote($table)); |
95 | 117 | $this->pdo->query('DROP TABLE IF EXISTS ' . $table ); |
96 | 118 | } |
97 | - } |
|
98 | - catch (\PDOException $e) { |
|
119 | + } catch (\PDOException $e) { |
|
99 | 120 | // No data to copy |
100 | 121 | echo $e->getMessage(); |
101 | 122 | } |
@@ -109,8 +130,11 @@ discard block |
||
109 | 130 | $parts = []; |
110 | 131 | foreach ($primaryKey as $key) { |
111 | 132 | $pk = $data->$key; |
112 | - if ($pk == null) $parts[] = $key . ' INTEGER'; |
|
113 | - else $parts[] = $key . ' ' . $this->getType($pk) . ' NOT NULL'; |
|
133 | + if ($pk == null) { |
|
134 | + $parts[] = $key . ' INTEGER'; |
|
135 | + } else { |
|
136 | + $parts[] = $key . ' ' . $this->getType($pk) . ' NOT NULL'; |
|
137 | + } |
|
114 | 138 | } |
115 | 139 | |
116 | 140 | $pkField = implode(', ', $parts) . ', PRIMARY KEY(' . implode(', ', $primaryKey) . ')'; |
@@ -119,8 +143,12 @@ discard block |
||
119 | 143 | $this->pdo->query('CREATE TABLE ' . $table . ' (' . $pkField . ')'); |
120 | 144 | |
121 | 145 | foreach ($data as $key => $value) { |
122 | - if (is_array($value) || (is_object($value) && !($value instanceof \DateTime))) continue; |
|
123 | - if (in_array($key, $primaryKey)) continue; |
|
146 | + if (is_array($value) || (is_object($value) && !($value instanceof \DateTime))) { |
|
147 | + continue; |
|
148 | + } |
|
149 | + if (in_array($key, $primaryKey)) { |
|
150 | + continue; |
|
151 | + } |
|
124 | 152 | |
125 | 153 | $type = $this->getType($value); |
126 | 154 | |
@@ -130,10 +158,14 @@ discard block |
||
130 | 158 | |
131 | 159 | |
132 | 160 | public function addIndex($table, array $fields) { |
133 | - if (empty($fields)) return false; |
|
161 | + if (empty($fields)) { |
|
162 | + return false; |
|
163 | + } |
|
134 | 164 | |
135 | 165 | //SQLite doesn't support ASC/DESC indexes, remove the keywords |
136 | - foreach ($fields as &$field) $field = str_ireplace([' desc', ' asc'], '', $field); |
|
166 | + foreach ($fields as &$field) { |
|
167 | + $field = str_ireplace([' desc', ' asc'], '', $field); |
|
168 | + } |
|
137 | 169 | sort($fields); |
138 | 170 | $fields = array_map('strtolower', $fields); |
139 | 171 | $fields = array_map('trim', $fields); |
@@ -142,8 +174,7 @@ discard block |
||
142 | 174 | |
143 | 175 | try { |
144 | 176 | $this->pdo->query('CREATE INDEX IF NOT EXISTS ' . $keyName . ' ON ' . $table . ' (' . implode(', ', $fields) . ')'); |
145 | - } |
|
146 | - catch (\Exception $e) { |
|
177 | + } catch (\Exception $e) { |
|
147 | 178 | |
148 | 179 | } |
149 | 180 | } |