DatabaseMysqli.php

Go to the documentation of this file.

<?php
namespace Wikimedia\Rdbms;
 
use mysqli;
use mysqli_result;
use IP;
use stdClass;
 
class DatabaseMysqli extends DatabaseMysqlBase {
    protected function doQuery( $sql ) {
        $conn = $this->getBindingHandle();
 
        if ( $this->bufferResults() ) {
            $ret = $conn->query( $sql );
        } else {
            $ret = $conn->query( $sql, MYSQLI_USE_RESULT );
        }
 
        return $ret;
    }
 
    protected function mysqlConnect( $realServer, $dbName ) {
        # Avoid suppressed fatal error, which is very hard to track down
        if ( !function_exists( 'mysqli_init' ) ) {
            throw new DBConnectionError( $this, "MySQLi functions missing,"
                . " have you compiled PHP with the --with-mysqli option?\n" );
        }
 
        // Other than mysql_connect, mysqli_real_connect expects an explicit port
        // and socket parameters. So we need to parse the port and socket out of
        // $realServer
        $port = null;
        $socket = null;
        $hostAndPort = IP::splitHostAndPort( $realServer );
        if ( $hostAndPort ) {
            $realServer = $hostAndPort[0];
            if ( $hostAndPort[1] ) {
                $port = $hostAndPort[1];
            }
        } elseif ( substr_count( $realServer, ':' ) == 1 ) {
            // If we have a colon and something that's not a port number
            // inside the hostname, assume it's the socket location
            $hostAndSocket = explode( ':', $realServer, 2 );
            $realServer = $hostAndSocket[0];
            $socket = $hostAndSocket[1];
        }
 
        $mysqli = mysqli_init();
 
        $connFlags = 0;
        if ( $this->flags & self::DBO_SSL ) {
            $connFlags |= MYSQLI_CLIENT_SSL;
            $mysqli->ssl_set(
                $this->sslKeyPath,
                $this->sslCertPath,
                $this->sslCAFile,
                $this->sslCAPath,
                $this->sslCiphers
            );
        }
        if ( $this->flags & self::DBO_COMPRESS ) {
            $connFlags |= MYSQLI_CLIENT_COMPRESS;
        }
        if ( $this->flags & self::DBO_PERSISTENT ) {
            $realServer = 'p:' . $realServer;
        }
 
        if ( $this->utf8Mode ) {
            // Tell the server we're communicating with it in UTF-8.
            // This may engage various charset conversions.
            $mysqli->options( MYSQLI_SET_CHARSET_NAME, 'utf8' );
        } else {
            $mysqli->options( MYSQLI_SET_CHARSET_NAME, 'binary' );
        }
        $mysqli->options( MYSQLI_OPT_CONNECT_TIMEOUT, 3 );
 
        if ( $mysqli->real_connect(
            $realServer,
            $this->user,
            $this->password,
            $dbName,
            $port,
            $socket,
            $connFlags
        ) ) {
            return $mysqli;
        }
 
        return false;
    }
 
    protected function connectInitCharset() {
        // already done in mysqlConnect()
        return true;
    }
 
    protected function mysqlSetCharset( $charset ) {
        $conn = $this->getBindingHandle();
 
        if ( method_exists( $conn, 'set_charset' ) ) {
            return $conn->set_charset( $charset );
        } else {
            return $this->query( 'SET NAMES ' . $charset, __METHOD__ );
        }
    }
 
    protected function closeConnection() {
        $conn = $this->getBindingHandle();
 
        return $conn->close();
    }
 
    function insertId() {
        $conn = $this->getBindingHandle();
 
        return (int)$conn->insert_id;
    }
 
    function lastErrno() {
        if ( $this->conn instanceof mysqli ) {
            return $this->conn->errno;
        } else {
            return mysqli_connect_errno();
        }
    }
 
    protected function fetchAffectedRowCount() {
        $conn = $this->getBindingHandle();
 
        return $conn->affected_rows;
    }
 
    function doSelectDomain( DatabaseDomain $domain ) {
        $conn = $this->getBindingHandle();
 
        if ( $domain->getSchema() !== null ) {
            throw new DBExpectedError( $this, __CLASS__ . ": domain schemas are not supported." );
        }
 
        $database = $domain->getDatabase();
        if ( !$conn->select_db( $database ) ) {
            throw new DBExpectedError( $this, "Could not select database '$database'." );
        }
 
        // Update that domain fields on success (no exception thrown)
        $this->currentDomain = $domain;
 
        return true;
    }
 
    protected function mysqlFreeResult( $res ) {
        $res->free_result();
 
        return true;
    }
 
    protected function mysqlFetchObject( $res ) {
        $object = $res->fetch_object();
        if ( $object === null ) {
            return false;
        }
 
        return $object;
    }
 
    protected function mysqlFetchArray( $res ) {
        $array = $res->fetch_array();
        if ( $array === null ) {
            return false;
        }
 
        return $array;
    }
 
    protected function mysqlNumRows( $res ) {
        return $res->num_rows;
    }
 
    protected function mysqlNumFields( $res ) {
        return $res->field_count;
    }
 
    protected function mysqlFetchField( $res, $n ) {
        $field = $res->fetch_field_direct( $n );
 
        // Add missing properties to result (using flags property)
        // which will be part of function mysql-fetch-field for backward compatibility
        $field->not_null = $field->flags & MYSQLI_NOT_NULL_FLAG;
        $field->primary_key = $field->flags & MYSQLI_PRI_KEY_FLAG;
        $field->unique_key = $field->flags & MYSQLI_UNIQUE_KEY_FLAG;
        $field->multiple_key = $field->flags & MYSQLI_MULTIPLE_KEY_FLAG;
        $field->binary = $field->flags & MYSQLI_BINARY_FLAG;
        $field->numeric = $field->flags & MYSQLI_NUM_FLAG;
        $field->blob = $field->flags & MYSQLI_BLOB_FLAG;
        $field->unsigned = $field->flags & MYSQLI_UNSIGNED_FLAG;
        $field->zerofill = $field->flags & MYSQLI_ZEROFILL_FLAG;
 
        return $field;
    }
 
    protected function mysqlFieldName( $res, $n ) {
        $field = $res->fetch_field_direct( $n );
 
        return $field->name;
    }
 
    protected function mysqlFieldType( $res, $n ) {
        $field = $res->fetch_field_direct( $n );
 
        return $field->type;
    }
 
    protected function mysqlDataSeek( $res, $row ) {
        return $res->data_seek( $row );
    }
 
    protected function mysqlError( $conn = null ) {
        if ( $conn === null ) {
            return mysqli_connect_error();
        } else {
            return $conn->error;
        }
    }
 
    protected function mysqlRealEscapeString( $s ) {
        $conn = $this->getBindingHandle();
 
        return $conn->real_escape_string( (string)$s );
    }
 
    public function __toString() {
        if ( $this->conn instanceof mysqli ) {
            return (string)$this->conn->thread_id;
        } else {
            // mConn might be false or something.
            return (string)$this->conn;
        }
    }
 
    protected function getBindingHandle() {
        return parent::getBindingHandle();
    }
}
 
class_alias( DatabaseMysqli::class, 'DatabaseMysqli' );