1 | <?php |
||
13 | class PdoDatabase extends PDO |
||
14 | { |
||
15 | /** |
||
16 | * @var PdoDatabase[] |
||
17 | */ |
||
18 | private static $connections = array(); |
||
19 | |||
20 | /** |
||
21 | * @var bool True if a transaction is active |
||
22 | */ |
||
23 | protected $hasActiveTransaction = false; |
||
24 | |||
25 | /** |
||
26 | * Summary of $queryLogStatement |
||
27 | * @var PDOStatement |
||
28 | */ |
||
29 | private $queryLogStatement; |
||
30 | |||
31 | /** |
||
32 | * @param string $connectionName |
||
33 | * @return PdoDatabase |
||
34 | * @throws Exception |
||
35 | */ |
||
36 | 1 | public static function getDatabaseConnection($connectionName) |
|
37 | { |
||
38 | 1 | if (!isset(self::$connections[$connectionName])) { |
|
39 | 1 | global $cDatabaseConfig; |
|
40 | |||
41 | 1 | if (!array_key_exists($connectionName, $cDatabaseConfig)) { |
|
42 | throw new Exception("Database configuration not found for alias $connectionName"); |
||
43 | } |
||
44 | |||
45 | try { |
||
46 | 1 | $databaseObject = new PdoDatabase( |
|
47 | 1 | $cDatabaseConfig[$connectionName]["dsrcname"], |
|
48 | 1 | $cDatabaseConfig[$connectionName]["username"], |
|
49 | 1 | $cDatabaseConfig[$connectionName]["password"], |
|
50 | 1 | $cDatabaseConfig[$connectionName]["options"] |
|
51 | ); |
||
52 | } |
||
53 | catch (PDOException $ex) { |
||
54 | // wrap around any potential stack traces which may include passwords |
||
55 | throw new Exception("Error connecting to database '$connectionName': " . $ex->getMessage()); |
||
56 | } |
||
57 | |||
58 | 1 | $databaseObject->setAttribute(PDO::ATTR_ERRMODE, PDO::ERRMODE_EXCEPTION); |
|
59 | |||
60 | // emulating prepared statements gives a performance boost on MySQL. |
||
61 | // |
||
62 | // however, our version of PDO doesn't seem to understand parameter types when emulating |
||
63 | // the prepared statements, so we're forced to turn this off for now. |
||
64 | // -- stw 2014-02-11 |
||
65 | 1 | $databaseObject->setAttribute(PDO::ATTR_EMULATE_PREPARES, false); |
|
66 | |||
67 | 1 | self::$connections[$connectionName] = $databaseObject; |
|
68 | } |
||
69 | |||
70 | 1 | return self::$connections[$connectionName]; |
|
71 | } |
||
72 | |||
73 | /** |
||
74 | * Determines if this connection has a transaction in progress or not |
||
75 | * @return boolean true if there is a transaction in progress. |
||
76 | */ |
||
77 | public function hasActiveTransaction() |
||
78 | { |
||
79 | return $this->hasActiveTransaction; |
||
80 | } |
||
81 | |||
82 | /** |
||
83 | * Summary of beginTransaction |
||
84 | * @return bool |
||
85 | */ |
||
86 | public function beginTransaction() |
||
87 | { |
||
88 | // Override the pre-existing method, which doesn't stop you from |
||
89 | // starting transactions within transactions - which doesn't work and |
||
90 | // will throw an exception. This eliminates the need to catch exceptions |
||
91 | // all over the rest of the code |
||
92 | if ($this->hasActiveTransaction) { |
||
93 | return false; |
||
94 | } |
||
95 | else { |
||
96 | // set the transaction isolation level for every transaction. |
||
97 | $this->exec("SET TRANSACTION ISOLATION LEVEL SERIALIZABLE;"); |
||
98 | |||
99 | // start a new transaction, and return whether or not the start was |
||
100 | // successful |
||
101 | $this->hasActiveTransaction = parent::beginTransaction(); |
||
102 | return $this->hasActiveTransaction; |
||
103 | } |
||
104 | } |
||
105 | |||
106 | /** |
||
107 | * Commits the active transaction |
||
108 | */ |
||
109 | public function commit() |
||
113 | } |
||
114 | |||
115 | /** |
||
116 | * Rolls back a transaction |
||
117 | */ |
||
118 | public function rollBack() |
||
119 | { |
||
120 | parent::rollback(); |
||
121 | $this->hasActiveTransaction = false; |
||
122 | } |
||
123 | |||
124 | /** |
||
125 | * Summary of transactionally |
||
126 | * @param Closure $method |
||
127 | */ |
||
128 | public function transactionally($method) |
||
129 | { |
||
130 | if (!$this->beginTransaction()) { |
||
131 | BootstrapSkin::displayAlertBox("Error starting database transaction.", "alert-error", "Database transaction error", true, false); |
||
132 | BootstrapSkin::displayInternalFooter(); |
||
133 | die(); |
||
|
|||
134 | } |
||
135 | |||
136 | try { |
||
137 | $method(); |
||
138 | |||
139 | $this->commit(); |
||
140 | } |
||
141 | catch (TransactionException $ex) { |
||
142 | $this->rollBack(); |
||
143 | |||
144 | BootstrapSkin::displayAlertBox($ex->getMessage(), $ex->getAlertType(), $ex->getTitle(), true, false); |
||
145 | |||
146 | // TODO: yuk. |
||
147 | if (defined("PUBLICMODE")) { |
||
148 | BootstrapSkin::displayPublicFooter(); |
||
149 | } |
||
150 | else { |
||
151 | BootstrapSkin::displayInternalFooter(); |
||
152 | } |
||
153 | |||
154 | die(); |
||
155 | } |
||
156 | } |
||
157 | |||
158 | /** |
||
159 | * Prepares a statement for execution. |
||
160 | * @param string $statement |
||
161 | * @param array $driver_options |
||
162 | * @return PDOStatement |
||
163 | */ |
||
164 | 1 | public function prepare($statement, $driver_options = array()) |
|
195 | } |
||
196 | } |
||
197 |
In general, usage of exit should be done with care and only when running in a scripting context like a CLI script.