Database.php

<?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();

    /**
     * @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();
        }

        // Clear the connection and re-connect if we have a non-zero error code:
        if (!is_null(self::$connections[$type]) && self::$connections[$type]->errorCode() !== '00000') {
            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;
        }

        return self::$connections[$type];
    }

    public function getDetails()
    {
        return self::$details;
    }
}