PDO: set unicode for mysql and other connection params

[tt-rss.git] / classes / db.php

diff --git a/classes/db.php b/classes/db.php
index 558d3e6b717b8a438ce494b115da7b3eb48e90a4..aba249a50b6779759a2b72fbb8648f7a08c8cc50 100644 (file)
--- a/classes/db.php
+++ b/classes/db.php
@@ -3,21 +3,64 @@ class Db implements IDb {
        private static $instance;
        private $adapter;
        private $link;
+       private $pdo;
 
        private function __construct() {
+
+               $er = error_reporting(E_ALL);
+
                switch (DB_TYPE) {
-               case "mysql":
-                       $this->adapter = new Db_Mysql();
-                       break;
-               case "pgsql":
-                       $this->adapter = new Db_Pgsql();
-                       break;
-               default:
-                       user_error("Unknown DB_TYPE: " . DB_TYPE);
+                       case "mysql":
+                               $this->adapter = new Db_Mysqli();
+                               break;
+                       case "pgsql":
+                               $this->adapter = new Db_Pgsql();
+                               break;
+                       default:
+                               die("Unknown DB_TYPE: " . DB_TYPE);
+               }
+
+               if (!$this->adapter) {
+                       print("Error initializing database adapter for " . DB_TYPE);
+                       exit(100);
+               }
+
+               $db_port = defined(DB_PORT) ? ';port='.DB_PORT : '';
+
+               $this->pdo = new PDO(DB_TYPE . ':dbname='.DB_NAME.';host='.DB_HOST.$db_port,
+                       DB_USER,
+                       DB_PASS);
+
+               $this->pdo->setAttribute( PDO::ATTR_ERRMODE, PDO::ERRMODE_WARNING );
+
+               if (!$this->pdo) {
+                       print("Error connecting via PDO.");
+                       exit(101);
+               }
+
+               if (DB_TYPE == "pgsql") {
+
+                       $this->pdo->query("set client_encoding = 'UTF-8'");
+                       $this->pdo->query("set datestyle = 'ISO, european'");
+                       $this->pdo->query("set TIME ZONE 0");
+                       $this->pdo->query("set cpu_tuple_cost = 0.5");
+
+               } else if (DB_TYPE == "mysql") {
+                       $this->pdo->query("SET time_zone = '+0:0'");
+
+                       if (defined('MYSQL_CHARSET') && MYSQL_CHARSET) {
+                               $this->pdo->query("SET NAMES " . MYSQL_CHARSET);
+                       }
                }
 
-               $this->link = $this->adapter->connect(DB_HOST, DB_USER, DB_PASS, DB_NAME, DB_PORT);
+               $this->link = $this->adapter->connect(DB_HOST, DB_USER, DB_PASS, DB_NAME, defined('DB_PORT') ? DB_PORT : "");
+
+               if (!$this->link) {
+                       print("Error connecting through adapter: " . $this->adapter->last_error());
+                       exit(101);
+               }
 
+               error_reporting($er);
        }
 
        private function __clone() {
@@ -31,17 +74,24 @@ class Db implements IDb {
                return self::$instance;
        }
 
+    public static function pdo() {
+        if (self::$instance == null)
+            self::$instance = new self();
+
+        return self::$instance->pdo;
+    }
+
        static function quote($str){
                return("'$str'");
        }
 
-       function init() {
-               //
+       function reconnect() {
+               $this->link = $this->adapter->connect(DB_HOST, DB_USER, DB_PASS, DB_NAME, defined('DB_PORT') ? DB_PORT : "");
        }
 
        function connect($host, $user, $pass, $db, $port) {
                //return $this->adapter->connect($host, $user, $pass, $db, $port);
-               return $this->link;
+               return ;
        }
 
        function escape_string($s, $strip_tags = true) {
@@ -76,5 +126,7 @@ class Db implements IDb {
                return $this->adapter->last_error();
        }
 
-}
-?>
+       function last_query_error() {
+               return $this->adapter->last_query_error();
+       }
+}
\ No newline at end of file