DatabaseFactory.php

Go to the documentation of this file.

<?php
namespace Wikimedia\Rdbms;
 
use HashBagOStuff;
use InvalidArgumentException;
use Psr\Log\NullLogger;
use Throwable;
 
class DatabaseFactory {
 
    public function create( $type, $params = [], $connect = Database::NEW_CONNECTED ) {
        $class = $this->getClass( $type, $params['driver'] ?? null );
 
        if ( class_exists( $class ) && is_subclass_of( $class, IDatabase::class ) ) {
            $params += [
                // Default configuration
                'host' => null,
                'user' => null,
                'password' => null,
                'dbname' => null,
                'schema' => null,
                'tablePrefix' => '',
                'flags' => 0,
                'variables' => [],
                'lbInfo' => [],
                'cliMode' => ( PHP_SAPI === 'cli' || PHP_SAPI === 'phpdbg' ),
                'agent' => '',
                'serverName' => null,
                'topologyRole' => null,
                // Objects and callbacks
                'topologicalPrimaryConnRef' => $params['topologicalPrimaryConnRef'] ?? null,
                'srvCache' => $params['srvCache'] ?? new HashBagOStuff(),
                'profiler' => $params['profiler'] ?? null,
                'trxProfiler' => $params['trxProfiler'] ?? new TransactionProfiler(),
                'connLogger' => $params['connLogger'] ?? new NullLogger(),
                'queryLogger' => $params['queryLogger'] ?? new NullLogger(),
                'replLogger' => $params['replLogger'] ?? new NullLogger(),
                'errorLogger' => $params['errorLogger'] ?? static function ( Throwable $e ) {
                    trigger_error( get_class( $e ) . ': ' . $e->getMessage(), E_USER_WARNING );
                },
                'deprecationLogger' => $params['deprecationLogger'] ?? static function ( $msg ) {
                    trigger_error( $msg, E_USER_DEPRECATED );
                }
            ];
 
            $conn = new $class( $params );
            if ( $connect === Database::NEW_CONNECTED ) {
                $conn->initConnection();
            }
        } else {
            $conn = null;
        }
 
        return $conn;
    }
 
    public function attributesFromType( $dbType, $driver = null ) {
        static $defaults = [
            Database::ATTR_DB_IS_FILE => false,
            Database::ATTR_DB_LEVEL_LOCKING => false,
            Database::ATTR_SCHEMAS_AS_TABLE_GROUPS => false
        ];
 
        $class = $this->getClass( $dbType, $driver );
        if ( class_exists( $class ) ) {
            return call_user_func( [ $class, 'getAttributes' ] ) + $defaults;
        } else {
            throw new DBUnexpectedError( null, "$dbType is not a supported database type." );
        }
    }
 
    protected function getClass( $dbType, $driver = null ) {
        // For database types with built-in support, the below maps type to IDatabase
        // implementations. For types with multiple driver implementations (PHP extensions),
        // an array can be used, keyed by extension name. In case of an array, the
        // optional 'driver' parameter can be used to force a specific driver. Otherwise,
        // we auto-detect the first available driver. For types without built-in support,
        // an class named "Database<Type>" us used, eg. DatabaseFoo for type 'foo'.
        static $builtinTypes = [
            'mysql' => [ 'mysqli' => DatabaseMysqli::class ],
            'sqlite' => DatabaseSqlite::class,
            'postgres' => DatabasePostgres::class,
        ];
 
        $dbType = strtolower( $dbType );
 
        if ( !isset( $builtinTypes[$dbType] ) ) {
            // Not a built in type, assume standard naming scheme
            return 'Database' . ucfirst( $dbType );
        }
 
        $class = false;
        $possibleDrivers = $builtinTypes[$dbType];
        if ( is_string( $possibleDrivers ) ) {
            $class = $possibleDrivers;
        } elseif ( (string)$driver !== '' ) {
            if ( !isset( $possibleDrivers[$driver] ) ) {
                throw new InvalidArgumentException( __METHOD__ .
                    " type '$dbType' does not support driver '{$driver}'" );
            }
 
            $class = $possibleDrivers[$driver];
        } else {
            foreach ( $possibleDrivers as $posDriver => $possibleClass ) {
                if ( extension_loaded( $posDriver ) ) {
                    $class = $possibleClass;
                    break;
                }
            }
        }
 
        if ( $class === false ) {
            throw new InvalidArgumentException( __METHOD__ .
                " no viable database extension found for type '$dbType'" );
        }
 
        return $class;
    }
}