Expose methods to allow users of the database class to control when transactions occur.

This commit is contained in:
jgen 2014-02-19 02:12:56 -05:00
parent 6bc9315955
commit 2ca74a3b33
2 changed files with 24 additions and 3 deletions

View file

@ -279,6 +279,12 @@ class Database {
*/
public $cache = null;
/**
* A boolean flag to track if we already have an active transaction.
* (ie: True if beginTransaction() already called)
*/
public $transaction = false;
/**
* For now, only connect to the cache, as we will pretty much certainly
* need it. There are some pages where all the data is in cache, so the
@ -327,7 +333,7 @@ class Database {
$this->engine->init($this->db);
// FIXME: this causes problems with running via the PHP CLI
$this->db->beginTransaction();
$this->beginTransaction();
}
private function connect_engine() {
@ -348,12 +354,23 @@ class Database {
}
}
public function beginTransaction() {
if ($this->transaction === false) {
$this->db->beginTransaction();
}
}
public function commit() {
if(!is_null($this->db)) return $this->db->commit();
if(!is_null($this->db) && $this->transaction === true) {
$this->transaction = false;
return $this->db->commit();
}
}
public function rollback() {
if(!is_null($this->db)) return $this->db->rollback();
if(!is_null($this->db) && $this->transaction === true) {
$this->transaction = false;
return $this->db->rollback();
}
public function escape($input) {

View file

@ -54,10 +54,14 @@ send_event(new InitExtEvent());
send_event(new UserCreationEvent("demo", "demo", ""));
$database->commit(); // Need to commit the new user to the database.
$database->beginTransaction();
send_event(new UserCreationEvent("test", "test", ""));
$database->commit(); // Need to commit the new user to the database.
$database->beginTransaction();
// Now we can run all the tests.
$all = new TestFinder("");
$results = $all->run(new TextReporter());