From 001f90bb99a983ffcfacf1fb8625c91c0b3afff4 Mon Sep 17 00:00:00 2001
From: kaetemi <kaetemi@gmail.com>
Date: Tue, 9 Sep 2014 02:15:25 +0200
Subject: [PATCH] Cache PDO connections and connection failures in a single
 page load

---
 code/web/private_php/ams/autoload/dblayer.php | 38 ++++++++++++-------
 1 file changed, 24 insertions(+), 14 deletions(-)

diff --git a/code/web/private_php/ams/autoload/dblayer.php b/code/web/private_php/ams/autoload/dblayer.php
index da3a757a6..c3ffdb403 100644
--- a/code/web/private_php/ams/autoload/dblayer.php
+++ b/code/web/private_php/ams/autoload/dblayer.php
@@ -29,13 +29,13 @@
  *
  */
 
-// $PDOCache = array();
+$PDOCache = array();
 
 class DBLayer {
 
 	private $PDO;
-	// private $host;
-	// private $dbname;
+	private $host;
+	private $dbname;
 
 	/**
 	* The PDO object, instantiated by the constructor
@@ -54,12 +54,15 @@ class DBLayer {
 		}
 
 		global $cfg;
-		// $this->host = $cfg['db'][$db]['host'];
-		// $this->dbname = $cfg['db'][$db]['name'];
-		/*global $PDOCache;
+		$this->host = $cfg['db'][$db]['host'];
+		$this->dbname = $cfg['db'][$db]['name'];
+		global $PDOCache;
 		if (isset($PDOCache[$this->host])) {
+			if (isset($PDOCache[$this->host]['exception'])) {
+				throw $PDOCache[$this->host]['exception'];
+			}
 			$this->PDO = $PDOCache[$this->host]['pdo'];
-		} else {*/
+		} else {
 			$dsn = "mysql:";
 			$dsn .= "host=" . $cfg['db'][$db]['host'] . ";";
 			$dsn .= "dbname=" . $cfg['db'][$db]['name'] . ";"; // Comment this out when using the cache
@@ -68,14 +71,21 @@ class DBLayer {
 			$opt = array(
 				PDO::ATTR_ERRMODE => PDO::ERRMODE_EXCEPTION,
 				PDO::ATTR_DEFAULT_FETCH_MODE => PDO::FETCH_ASSOC,
-				PDO::ATTR_PERSISTENT => true
+				PDO::ATTR_PERSISTENT => true,
+				PDO::ATTR_TIMEOUT => 5
 			);
-			$this->PDO = new PDO($dsn, $cfg['db'][$db]['user'], $cfg['db'][$db]['pass'], $opt);
-		/*	$PDOCache[$this->host] = array();
+			$PDOCache[$this->host] = array();
+			try {
+				$this->PDO = new PDO($dsn, $cfg['db'][$db]['user'], $cfg['db'][$db]['pass'], $opt);
+			} catch (Exception $e) {
+				$PDOCache[$this->host]['exception'] = $e;
+				throw $e;
+				return;
+			}
 			$PDOCache[$this->host]['pdo'] = $this->PDO;
 			$PDOCache[$this->host]['use'] = $this->dbname;
-		*/	//$this->PDO->query('USE ' . $this->dbname . ';'); // FIXME safety
-		/*}*/
+			$this->PDO->query('USE ' . $this->dbname . ';'); // FIXME safety
+		}
 	}
 
 	function __destruct() {
@@ -83,11 +93,11 @@ class DBLayer {
 	}
 
 	function useDb() {
-		/*global $PDOCache;
+		global $PDOCache;
 		if ($PDOCache[$this->host]['use'] != $this->dbname) {
 			$PDOCache[$this->host]['use'] = $this->dbname;
 			$this->PDO->query('USE ' . $this->dbname . ';'); // FIXME safety
-		}*/
+		}
 	}
 
 	/**