DatabaseMysqli.php

Go to the documentation of this file.

<?php
class DatabaseMysqli extends DatabaseMysqlBase {
    protected function doQuery( $sql ) {
        if ( $this->bufferResults() ) {
            $ret = $this->mConn->query( $sql );
        } else {
            $ret = $this->mConn->query( $sql, MYSQLI_USE_RESULT );
        }

        return $ret;
    }

    protected function mysqlConnect( $realServer ) {
        global $wgDBmysql5;
        # Fail now
        # Otherwise we get a 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 );
            $realServer = $hostAndSocket[0];
            $socket = $hostAndSocket[1];
        }

        $connFlags = 0;
        if ( $this->mFlags & DBO_SSL ) {
            $connFlags |= MYSQLI_CLIENT_SSL;
        }
        if ( $this->mFlags & DBO_COMPRESS ) {
            $connFlags |= MYSQLI_CLIENT_COMPRESS;
        }
        if ( $this->mFlags & DBO_PERSISTENT ) {
            $realServer = 'p:' . $realServer;
        }

        $mysqli = mysqli_init();
        if ( $wgDBmysql5 ) {
            // 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' );
        }

        $numAttempts = 2;
        for ( $i = 0; $i < $numAttempts; $i++ ) {
            if ( $i > 1 ) {
                usleep( 1000 );
            }
            if ( $mysqli->real_connect( $realServer, $this->mUser,
                $this->mPassword, $this->mDBname, $port, $socket, $connFlags )
            ) {
                return $mysqli;
            }
        }

        return false;
    }

    protected function connectInitCharset() {
        // already done in mysqlConnect()
        return true;
    }

    protected function mysqlSetCharset( $charset ) {
        if ( method_exists( $this->mConn, 'set_charset' ) ) {
            return $this->mConn->set_charset( $charset );
        } else {
            return $this->query( 'SET NAMES ' . $charset, __METHOD__ );
        }
    }

    protected function closeConnection() {
        return $this->mConn->close();
    }

    function insertId() {
        return $this->mConn->insert_id;
    }

    function lastErrno() {
        if ( $this->mConn ) {
            return $this->mConn->errno;
        } else {
            return mysqli_connect_errno();
        }
    }

    function affectedRows() {
        return $this->mConn->affected_rows;
    }

    function selectDB( $db ) {
        $this->mDBname = $db;

        return $this->mConn->select_db( $db );
    }

    function getServerVersion() {
        return $this->mConn->server_info;
    }

    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 );
        $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;

        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 ) {
        return $this->mConn->real_escape_string( $s );
    }

    protected function mysqlPing() {
        return $this->mConn->ping();
    }

    public function __toString() {
        if ( $this->mConn instanceof Mysqli ) {
            return (string)$this->mConn->thread_id;
        } else {
            // mConn might be false or something.
            return (string)$this->mConn;
        }
    }
}