php-censor/src/B8Framework/Database.php

<?php

namespace b8;

class Database extends \PDO
{
    protected static $initialised = false;
    protected static $servers     = ['read' => [], 'write' => []];
    protected static $connections = ['read' => null, 'write' => null];
    protected static $details     = [];
    protected static $lastUsed    = ['read' => null, 'write' => null];

    protected static function init()
    {
        $config   = Config::getInstance();
        $settings = $config->get('b8.database', []);

        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 = [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'],
                        [
                            \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 = ['read' => null, 'write' => null];
        self::$lastUsed    = ['read' => null, 'write' => null];
        self::$initialised = false;
    }
}