DatabaseMysqli.php

Go to the documentation of this file.

<?php
namespace Wikimedia\Rdbms;
 
use mysqli;
use mysqli_result;
use Wikimedia\AtEase\AtEase;
use Wikimedia\IPUtils;
 
class DatabaseMysqli extends DatabaseMysqlBase {
    protected function doQuery( $sql ) {
        AtEase::suppressWarnings();
        $res = $this->getBindingHandle()->query( $sql );
        AtEase::restoreWarnings();
 
        return $res instanceof mysqli_result ? new MysqliResultWrapper( $this, $res ) : $res;
    }
 
    protected function mysqlConnect( $server, $user, $password, $db ) {
        if ( !function_exists( 'mysqli_init' ) ) {
            throw $this->newExceptionAfterConnectError(
                "MySQLi functions missing, have you compiled PHP with the --with-mysqli option?"
            );
        }
 
        // PHP 8.1.0+ throws exceptions by default. Turn that off for consistency.
        mysqli_report( MYSQLI_REPORT_OFF );
 
        // Other than mysql_connect, mysqli_real_connect expects an explicit port number
        // e.g. "localhost:1234" or "127.0.0.1:1234"
        // or Unix domain socket path
        // e.g. "localhost:/socket_path" or "localhost:/foo/bar:bar:bar"
        // colons are known to be used by Google AppEngine,
        // see <https://cloud.google.com/sql/docs/mysql/connect-app-engine>
        //
        // We need to parse the port or socket path out of $realServer
        $port = null;
        $socket = null;
        $hostAndPort = IPUtils::splitHostAndPort( $server );
        if ( $hostAndPort ) {
            $realServer = $hostAndPort[0];
            if ( $hostAndPort[1] ) {
                $port = $hostAndPort[1];
            }
        } elseif ( substr_count( $server, ':/' ) == 1 ) {
            // If we have a colon slash instead of a colon and a port number
            // after the ip or hostname, assume it's the Unix domain socket path
            list( $realServer, $socket ) = explode( ':', $server, 2 );
        } else {
            $realServer = $server;
        }
 
        $mysqli = mysqli_init();
        // Make affectedRows() for UPDATE reflect the number of matching rows, regardless
        // of whether any column values changed. This is what callers want to know and is
        // consistent with what Postgres, SQLite, and SQL Server return.
        $flags = MYSQLI_CLIENT_FOUND_ROWS;
        if ( $this->getFlag( self::DBO_SSL ) ) {
            $flags |= MYSQLI_CLIENT_SSL;
            $mysqli->ssl_set(
                $this->sslKeyPath,
                $this->sslCertPath,
                $this->sslCAFile,
                $this->sslCAPath,
                $this->sslCiphers
            );
        }
        if ( $this->getFlag( self::DBO_COMPRESS ) ) {
            $flags |= MYSQLI_CLIENT_COMPRESS;
        }
        if ( $this->getFlag( 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 );
 
        $ok = $mysqli->real_connect( $realServer, $user, $password, $db, $port, $socket, $flags );
 
        return $ok ? $mysqli : null;
    }
 
    protected function closeConnection() {
        $conn = $this->getBindingHandle();
 
        return $conn->close();
    }
 
    public function insertId() {
        $conn = $this->getBindingHandle();
 
        return (int)$conn->insert_id;
    }
 
    public 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;
    }
 
    protected function mysqlFieldType( $res, $n ) {
        $field = $res->fetch_field_direct( $n );
 
        return $field->type;
    }
 
    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 );
    }
 
    protected function getBindingHandle() {
        return parent::getBindingHandle();
    }
}
 
class_alias( DatabaseMysqli::class, 'DatabaseMysqli' );