Refactored structure

src/B8Framework/Database.php

@@ -0,0 +1,151 @@
+<?php
+
+namespace b8;
+
+class Database extends \PDO
+{
+	protected static $initialised = false;
+	protected static $servers     = array('read' => array(), 'write' => array());
+	protected static $connections = array('read' => null, 'write' => null);
+	protected static $details     = array();
+    protected static $lastUsed	  = array('read' => null, 'write' => null);
+
+	/**
+	* @deprecated
+	*/
+	public static function setReadServers($read)
+	{
+		$config = Config::getInstance();
+
+		$settings = $config->get('b8.database', array());
+		$settings['servers']['read'] = $read;
+		$config->set('b8.database', $settings);
+	}
+
+	/**
+	* @deprecated
+	*/
+	public static function setWriteServers($write)
+	{
+		$config = Config::getInstance();
+
+		$settings = $config->get('b8.database', array());
+		$settings['servers']['write'] = $write;
+		$config->set('b8.database', $settings);
+	}
+
+	/**
+	* @deprecated
+	*/
+	public static function setDetails($database, $username, $password)
+	{
+		$config               = Config::getInstance();
+		$settings             = $config->get('b8.database', array());
+		$settings['name']     = $database;
+		$settings['username'] = $username;
+		$settings['password'] = $password;
+		$config->set('b8.database', $settings);
+	}
+
+	protected static function init()
+	{
+		$config   = Config::getInstance();
+		$settings = $config->get('b8.database', array());
+		self::$servers['read']  = $settings['servers']['read'];
+		self::$servers['write'] = $settings['servers']['write'];
+		self::$details['db']    = $settings['name'];
+		self::$details['user']  = $settings['username'];
+		self::$details['pass']  = $settings['password'];
+		self::$initialised      = true;
+	}
+
+	/**
+	 * @param string $type
+	 *
+	 * @return \b8\Database
+	 * @throws \Exception
+	 */
+	public static function getConnection($type = 'read')
+	{
+		if (!self::$initialised) {
+			self::init();
+		}
+
+		// If the connection hasn't been used for 5 minutes, force a reconnection:
+		if (!is_null(self::$lastUsed[$type]) && (time() - self::$lastUsed[$type]) > 300) {
+			self::$connections[$type] = null;
+		}
+
+		if(is_null(self::$connections[$type])) {
+			if (is_array(self::$servers[$type])) {
+				// Shuffle, so we pick a random server:
+				$servers = self::$servers[$type];
+				shuffle($servers);
+			} else {
+				// Only one server was specified
+				$servers = array(self::$servers[$type]);
+			}
+
+			$connection = null;
+
+			// Loop until we get a working connection:
+			while(count($servers))
+			{
+				// Pull the next server:
+				$server = array_shift($servers);
+
+				if (stristr($server, ':')) {
+					list($host, $port) = explode(':', $server);
+					$server = $host . ';port=' . $port;
+				}
+
+				// Try to connect:
+				try
+				{
+					$connection = new self('mysql:host=' . $server . ';dbname=' . self::$details['db'],
+						self::$details['user'],
+						self::$details['pass'],
+						array(
+						     \PDO::ATTR_PERSISTENT         => false,
+						     \PDO::ATTR_ERRMODE            => \PDO::ERRMODE_EXCEPTION,
+						     \PDO::ATTR_TIMEOUT            => 2,
+						     \PDO::MYSQL_ATTR_INIT_COMMAND => 'SET NAMES \'UTF8\'',
+						));
+				}
+				catch(\PDOException $ex)
+				{
+					$connection = false;
+				}
+
+				// Opened a connection? Break the loop:
+				if($connection)
+				{
+					break;
+				}
+			}
+
+			// No connection? Oh dear.
+			if(!$connection && $type == 'read')
+			{
+				throw new \Exception('Could not connect to any ' . $type . ' servers.');
+			}
+
+			self::$connections[$type] = $connection;
+		}
+
+		self::$lastUsed[$type] = time();
+		return self::$connections[$type];
+	}
+
+	public function getDetails()
+	{
+		return self::$details;
+	}
+
+    public static function reset()
+    {
+        self::$connections = array('read' => null, 'write' => null);
+        self::$lastUsed = array('read' => null, 'write' => null);
+        self::$initialised = false;
+    }
+}