From 4773d7a35052df4683b0445c08feae609764bfb0 Mon Sep 17 00:00:00 2001
From: nikrou <nikrou@piwigo.org>
Date: Sun, 21 Mar 2010 22:51:36 +0000
Subject: Feature 1255 : only one function use exceptions to deal with
 differents possible errors

git-svn-id: http://piwigo.org/svn/trunk@5236 68402e56-0260-453c-a942-63ccdbb3a9ee
---
 include/dblayer/functions_mysql.inc.php      | 23 ++++++++++++++---------
 include/dblayer/functions_pdo-sqlite.inc.php | 15 +++++----------
 include/dblayer/functions_pgsql.inc.php      | 14 ++++++++++----
 include/dblayer/functions_sqlite.inc.php     | 16 ++++++----------
 4 files changed, 35 insertions(+), 33 deletions(-)

(limited to 'include/dblayer')

diff --git a/include/dblayer/functions_mysql.inc.php b/include/dblayer/functions_mysql.inc.php
index 6ada73a59..c21c697ae 100644
--- a/include/dblayer/functions_mysql.inc.php
+++ b/include/dblayer/functions_mysql.inc.php
@@ -32,16 +32,21 @@ define('DB_RANDOM_FUNCTION', 'RAND');
  *
  */
 
-function pwg_db_connect($host, $user, $password, $database=null, $die=true)
+function pwg_db_connect($host, $user, $password, $database)
 {  
-  $link = @mysql_connect($host, $user, $password) or my_error('mysql_connect', $die);
-
-  return $link;
-}
-
-function pwg_select_db($database, $link, $die=true)
-{
-  return @mysql_select_db($database, $link) or my_error('mysql_select_db', $die);
+  $link = @mysql_connect($host, $user, $password);
+  if (!$link)
+  {
+    throw new Exception("Can't connect to server");
+  }
+  if (mysql_select_db($database, $link))
+  {
+    return $link;
+  }
+  else
+  {
+    throw new Exception('Connection to server succeed, but it was impossible to connect to database');
+  }
 }
 
 function pwg_db_check_charset() 
diff --git a/include/dblayer/functions_pdo-sqlite.inc.php b/include/dblayer/functions_pdo-sqlite.inc.php
index aba554341..33a864980 100644
--- a/include/dblayer/functions_pdo-sqlite.inc.php
+++ b/include/dblayer/functions_pdo-sqlite.inc.php
@@ -33,16 +33,16 @@ define('DB_RANDOM_FUNCTION', 'RANDOM');
  *
  */
 
-function pwg_db_connect($host, $user, $password, $database, $die=true)
+function pwg_db_connect($host, $user, $password, $database)
 {
   global $conf;
 
   $db_file = sprintf('sqlite:%s/%s.db', $conf['local_data_dir'], $database);
 
-  try {
-    $link = new PDO($db_file);
-  } catch (Exception $e) {
-    my_error('sqlite::open', $die);
+  $link = new PDO($db_file);
+  if (!$link)
+  {
+    throw new  Exception('Connection to server succeed, but it was impossible to connect to database');
   }
 
   $link->sqliteCreateFunction('now', 'pwg_now', 0);
@@ -56,11 +56,6 @@ function pwg_db_connect($host, $user, $password, $database, $die=true)
   return $link;
 }
 
-function pwg_select_db($database=null, $link=null, $die=null)
-{
-  return true;
-}
-
 function pwg_db_check_charset() 
 {
   return true;
diff --git a/include/dblayer/functions_pgsql.inc.php b/include/dblayer/functions_pgsql.inc.php
index fe350d090..5c928c53c 100644
--- a/include/dblayer/functions_pgsql.inc.php
+++ b/include/dblayer/functions_pgsql.inc.php
@@ -33,7 +33,7 @@ define('DB_RANDOM_FUNCTION', 'RANDOM');
  *
  */
 
-function pwg_db_connect($host, $user, $password, $database, $die=true)
+function pwg_db_connect($host, $user, $password, $database)
 {
   $connection_string = '';
   if (strpos($host,':') !== false) 
@@ -49,9 +49,15 @@ function pwg_db_connect($host, $user, $password, $database, $die=true)
 				$user, 
 				$password,
 				$database);
-  $link = pg_connect($connection_string) or my_error('pg_connect', $die);  
-
-  return $link;
+  $link = pg_connect($connection_string);
+  if (!$link)
+  {
+    throw new Exception("Can't connect to server");
+  }
+  else
+  {
+    return $link;
+  }
 }
 
 function pwg_select_db($database=null, $link=null, $die=null)
diff --git a/include/dblayer/functions_sqlite.inc.php b/include/dblayer/functions_sqlite.inc.php
index ce426b9e0..7f02a06d1 100644
--- a/include/dblayer/functions_sqlite.inc.php
+++ b/include/dblayer/functions_sqlite.inc.php
@@ -33,7 +33,7 @@ define('DB_RANDOM_FUNCTION', 'RANDOM');
  *
  */
 
-function pwg_db_connect($host, $user, $password, $database, $die=true)
+function pwg_db_connect($host, $user, $password, $database)
 {
   global $conf;
 
@@ -47,10 +47,11 @@ function pwg_db_connect($host, $user, $password, $database, $die=true)
   {
     $sqlite_open_mode = SQLITE3_OPEN_READWRITE;
   }
-  try {
-    $link = new SQLite3($db_file, $sqlite_open_mode);
-  } catch (Exception $e) {
-    my_error('sqlite::open', $die);
+  
+  $link = new SQLite3($db_file, $sqlite_open_mode);
+  if (!$link)
+  {
+    throw new  Exception('Connection to server succeed, but it was impossible to connect to database');
   }
 
   $link->createFunction('now', 'pwg_now', 0);
@@ -64,11 +65,6 @@ function pwg_db_connect($host, $user, $password, $database, $die=true)
   return $link;
 }
 
-function pwg_select_db($database=null, $link=null, $die=null)
-{
-  return true;
-}
-
 function pwg_db_check_charset() 
 {
   return true;
-- 
cgit v1.2.3