diff options
author | Andrew Dolgov <[email protected]> | 2013-03-15 03:09:22 -0700 |
---|---|---|
committer | Andrew Dolgov <[email protected]> | 2013-03-15 03:09:22 -0700 |
commit | 07be89952d9e0b16eebd1b1a6a8dc8f7f41fcf3b (patch) | |
tree | 840b4a3237f4dae0a9ae988d7b0691b0fcd38c2d /classes | |
parent | 7d272e5c04141529309fe2924d0aac93f17b56ea (diff) | |
parent | 65d0cc64a2ca1ce259fbba501b13deaf1870e540 (diff) |
Merge pull request #46 from mbirth/master
Core db features in separate classes for Pgsql and Mysql
Diffstat (limited to 'classes')
-rw-r--r-- | classes/Db/Abstract.php | 56 | ||||
-rw-r--r-- | classes/Db/Interface.php | 18 | ||||
-rw-r--r-- | classes/Db/Mysql.php | 71 | ||||
-rw-r--r-- | classes/Db/Pgsql.php | 80 |
4 files changed, 225 insertions, 0 deletions
diff --git a/classes/Db/Abstract.php b/classes/Db/Abstract.php new file mode 100644 index 000000000..d84701fd7 --- /dev/null +++ b/classes/Db/Abstract.php @@ -0,0 +1,56 @@ +<?php + +abstract class Db_Abstract implements Db_Interface +{ + private $dbconn; + protected static $instance; + + private function __construct() { } + + public static function instance() + { + if (is_null(static::$instance)) { + static::$instance = new static(); + } + + return static::$instance; + } + + public function connect($host, $user, $pass, $db) { } + + public function getLink() + { + return $this->dbconn; + } + + public function init() { } + + public function escape_string($s, $strip_tags = true) { } + + public function query($query, $die_on_error = true) { } + + public function fetch_assoc($result) { } + + public function num_rows($result) { } + + public function fetch_result($result, $row, $param) { } + + public function unescape_string($str) + { + $tmp = str_replace("\\\"", "\"", $str); + $tmp = str_replace("\\'", "'", $tmp); + return $tmp; + } + + public function close() { } + + public function affected_rows($result) { } + + public function last_error() { } + + public function quote($str) + { + return("'$str'"); + } + +}
\ No newline at end of file diff --git a/classes/Db/Interface.php b/classes/Db/Interface.php new file mode 100644 index 000000000..c5bfad56f --- /dev/null +++ b/classes/Db/Interface.php @@ -0,0 +1,18 @@ +<?php + +interface Db_Interface +{ + public function connect($host, $user, $pass, $db); + public function getLink(); + public function init(); + public function escape_string($s, $strip_tags = true); + public function query($query, $die_on_error = true); + public function fetch_assoc($result); + public function num_rows($result); + public function fetch_result($result, $row, $param); + public function unescape_string($str); + public function close(); + public function affected_rows($result); + public function last_error(); + public function quote($str); +} diff --git a/classes/Db/Mysql.php b/classes/Db/Mysql.php new file mode 100644 index 000000000..170628304 --- /dev/null +++ b/classes/Db/Mysql.php @@ -0,0 +1,71 @@ +<?php + +class Db_Mysql extends Db_Abstract +{ + public function connect($host, $user, $pass, $db) + { + $link = mysql_connect($host, $user, $pass); + if ($link) { + $result = mysql_select_db($db, $link); + if (!$result) { + die("Can't select DB: " . mysql_error($link)); + } + $this->dbconn = $link; + return $link; + } else { + die("Unable to connect to database (as $user to $host, database $db): " . mysql_error()); + } + } + + public function init() + { + db_query($this->dbconn, "SET time_zone = '+0:0'"); + + if (defined('MYSQL_CHARSET') && MYSQL_CHARSET) { + db_query($this->dbconn, "SET NAMES " . MYSQL_CHARSET); + } + } + + public function escape_string($s, $strip_tags = true) + { + if ($strip_tags) $s = strip_tags($s); + return mysql_real_escape_string($s); + } + + public function query($query, $die_on_error = true) + { + $result = mysql_query($query, $this->dbconn); + if (!$result) { + $query = htmlspecialchars($query); + if ($die_on_error) { + die("Query <i>$query</i> failed: " . ($this->dbconn ? mysql_error($this->dbconn) : "No connection")); + } + } + return $result; + } + + public function fetch_assoc($result) { + return mysql_fetch_assoc($result); + } + + public function num_rows($result) { + return mysql_num_rows($result); + } + + public function fetch_result($result, $row, $param) { + // I hate incoherent naming of PHP functions + return mysql_result($result, $row, $param); + } + + public function close() { + return mysql_close($this->dbconn); + } + + public function affected_rows($result) { + return mysql_affected_rows($this->dbconn); + } + + public function last_error() { + return mysql_error($this->dbconn); + } +} diff --git a/classes/Db/Pgsql.php b/classes/Db/Pgsql.php new file mode 100644 index 000000000..39ed181d7 --- /dev/null +++ b/classes/Db/Pgsql.php @@ -0,0 +1,80 @@ +<?php + +class Db_Pgsql extends Db_Abstract +{ + public function connect($host, $user, $pass, $db) + { + $string = "dbname=$db user=$user"; + + if ($pass) { + $string .= " password=$pass"; + } + + if ($host) { + $string .= " host=$host"; + } + + if (defined('DB_PORT')) { + $string = "$string port=" . DB_PORT; + } + + $link = pg_connect($string); + + if (!$link) { + die("Unable to connect to database (as $user to $host, database $db):" . pg_last_error()); + } + + $this->dbconn = $link; + return $link; + } + + public function init() + { + pg_query($this->dbconn, "set client_encoding = 'UTF-8'"); + pg_set_client_encoding("UNICODE"); + pg_query($this->dbconn, "set datestyle = 'ISO, european'"); + pg_query($this->dbconn, "set TIME ZONE 0"); + } + + public function escape_string($s, $strip_tags = true) + { + if ($strip_tags) $s = strip_tags($s); + return pg_escape_string($s); + } + + public function query($query, $die_on_error = true) + { + $result = pg_query($this->dbconn, $query); + if (!$result) { + $query = htmlspecialchars($query); // just in case + if ($die_on_error) { + die("Query <i>$query</i> failed [$result]: " . ($this->dbconn ? pg_last_error($this->dbconn) : "No connection")); + } + } + return $result; + } + + public function fetch_assoc($result) { + return pg_fetch_assoc($result); + } + + public function num_rows($result) { + return pg_num_rows($result); + } + + public function fetch_result($result, $row, $param) { + return pg_fetch_result($result, $row, $param); + } + + public function close() { + return pg_close($this->dbconn); + } + + public function affected_rows($result) { + return pg_affected_rows($result); + } + + public function last_error() { + return pg_last_error($this->dbconn); + } +} |