Changeset 12667

Timestamp:

01/08/2010 07:26:00 PM (15 years ago)

Author:

wpmuguru

Message:

db class compatible with WP, MU & multi site - See #11644

File:

• trunk/wp-includes/wp-db.php (modified) (14 diffs)

trunk/wp-includes/wp-db.php

@@ -136 +136 @@
      */
     var $ready = false;
+    var $blogid = 0;
+    var $siteid = 0;
+    var $blogs;
+    var $signups;
+    var $site;
+    var $sitemeta;
+    var $sitecategories;
+    var $global_tables = array('blogs', 'signups', 'site', 'sitemeta', 'users', 'usermeta', 'sitecategories', 'registration_log', 'blog_versions');
 
     /**
@@ -261 +269 @@
      * @var array
      */
-    var $tables = array('users', 'usermeta', 'posts', 'categories', 'post2cat', 'comments', 'links', 'link2cat', 'options',
+    var $tables = array('posts', 'categories', 'post2cat', 'comments', 'links', 'link2cat', 'options',
             'postmeta', 'terms', 'term_taxonomy', 'term_relationships', 'commentmeta');
 
@@ -338 +346 @@
      */
     function wpdb($dbuser, $dbpassword, $dbname, $dbhost) {
+        if( defined( "WP_USE_MULTIPLE_DB" ) && CONSTANT( "WP_USE_MULTIPLE_DB" ) == true )
+            $this->db_connect();
         return $this->__construct($dbuser, $dbpassword, $dbname, $dbhost);
     }
@@ -360 +370 @@
         if ( WP_DEBUG )
             $this->show_errors();
+
+                if( is_multisite() ) {
+                        $this->charset = 'utf8';
+                        if( defined( 'DB_COLLATE' ) && constant( 'DB_COLLATE' ) != '' ) {
+                                $this->collate = constant( 'DB_COLLATE' );
+                        } else {
+                                $this->collate = 'utf8_general_ci';
+                        }
+                }
 
         if ( defined('DB_CHARSET') )
@@ -398 +417 @@
         }
 
-        $this->select($dbname);
+        $this->select($dbname, $this->dbh);
     }
 
@@ -428 +447 @@
             return new WP_Error('invalid_db_prefix', /*WP_I18N_DB_BAD_PREFIX*/'Invalid database prefix'/*/WP_I18N_DB_BAD_PREFIX*/);
 
-        $old_prefix = $this->prefix;
-        $this->prefix = $prefix;
+                if( is_multisite() ) {
+                        $old_prefix = '';
+                } else {
+                        $old_prefix = $prefix;
+                }
+        if( isset( $this->base_prefix ) )
+            $old_prefix = $this->base_prefix;
+        $this->base_prefix = $prefix;
+        foreach ( $this->global_tables as $table )
+            $this->$table = $prefix . $table;
+
+        if ( defined('VHOST') && empty($this->blogid) )
+            return $old_prefix;
+
+        $this->prefix = $this->get_blog_prefix( $this->blogid );
 
         foreach ( (array) $this->tables as $table )
@@ -443 +475 @@
     }
 
+    function set_blog_id($blog_id, $site_id = '') {
+        if ( !empty($site_id) )
+            $this->siteid = $site_id;
+
+        $old_blog_id = $this->blogid;
+        $this->blogid = $blog_id;
+
+        $this->prefix = $this->get_blog_prefix( $this->blogid );
+
+        foreach ( $this->tables as $table )
+            $this->$table = $this->prefix . $table;
+
+        return $old_blog_id;
+    }
+
+    function get_blog_prefix( $blog_id = '' ) {
+        if ( $blog_id ) {
+                        if( defined('MULTISITE') && ( $blog_id == 0 || $blog_id == 1) ) {
+                    return $this->prefix;
+                        } else {
+                    return $this->base_prefix . $blog_id . '_';
+                        }
+        } else {
+            return $this->base_prefix;
+        }
+    }
+
     /**
      * Selects a database using the current database connection.
@@ -454 +513 @@
      * @return null Always null.
      */
-    function select($db) {
-        if (!@mysql_select_db($db, $this->dbh)) {
+    function select($db, &$dbh) {
+        if (!@mysql_select_db($db, $dbh)) {
             $this->ready = false;
             $this->bail(sprintf(/*WP_I18N_DB_SELECT_DB*/'
@@ -606 +665 @@
             return false;
 
-        $str = htmlspecialchars($str, ENT_QUOTES);
-        $query = htmlspecialchars($this->last_query, ENT_QUOTES);
-
-        // If there is an error then take note of it
-        print "<div id='error'>
-        <p class='wpdberror'><strong>WordPress database error:</strong> [$str]<br />
-        <code>$query</code></p>
-        </div>";
+                // If there is an error then take note of it
+                if( is_multisite() ) {
+                        $msg = "WordPress database error: [$str]\n{$this->query}\n";
+                        if( defined( 'ERRORLOGFILE' ) )
+                                error_log( $msg, 3, CONSTANT( 'ERRORLOGFILE' ) );
+                        if( defined( 'DIEONDBERROR' ) )
+                                die( $msg );
+                } else {
+                        $str = htmlspecialchars($str, ENT_QUOTES);
+                        $query = htmlspecialchars($this->last_query, ENT_QUOTES);
+
+                        print "<div id='error'>
+                        <p class='wpdberror'><strong>WordPress database error:</strong> [$str]<br />
+                        <code>$query</code></p>
+                        </div>";
+                }
     }
 
@@ -671 +738 @@
     }
 
+    function db_connect( $query = "SELECT" ) {
+        global $db_list, $global_db_list;
+        if( is_array( $db_list ) == false )
+            return true;
+
+        if( $this->blogs != '' && preg_match("/(" . $this->blogs . "|" . $this->users . "|" . $this->usermeta . "|" . $this->site . "|" . $this->sitemeta . "|" . $this->sitecategories . ")/i",$query) ) {
+            $action = 'global';
+            $details = $global_db_list[ mt_rand( 0, count( $global_db_list ) -1 ) ];
+            $this->db_global = $details;
+        } elseif ( preg_match("/^\\s*(alter table|create|insert|delete|update|replace) /i",$query) ) {
+            $action = 'write';
+            $details = $db_list[ 'write' ][ mt_rand( 0, count( $db_list[ 'write' ] ) -1 ) ];
+            $this->db_write = $details;
+        } else {
+            $action = '';
+            $details = $db_list[ 'read' ][ mt_rand( 0, count( $db_list[ 'read' ] ) -1 ) ];
+            $this->db_read = $details;
+        }
+
+        $dbhname = "dbh" . $action;
+        $this->$dbhname = @mysql_connect( $details[ 'db_host' ], $details[ 'db_user' ], $details[ 'db_password' ] );
+        if (!$this->$dbhname ) {
+            $this->bail("
+<h1>Error establishing a database connection</h1>
+<p>This either means that the username and password information in your <code>wp-config.php</code> file is incorrect or we can't contact the database server at <code>$dbhost</code>. This could mean your host's database server is down.</p>
+<ul>
+    <li>Are you sure you have the correct username and password?</li>
+    <li>Are you sure that you have typed the correct hostname?</li>
+    <li>Are you sure that the database server is running?</li>
+</ul>
+<p>If you're unsure what these terms mean you should probably contact your host. If you still need help you can always visit the <a href='http://wordpress.org/support/'>WordPress Support Forums</a>.</p>
+");
+        }
+        $this->select( $details[ 'db_name' ], $this->$dbhname );
+    }
+
     /**
      * Perform a MySQL database query, using current database connection.
@@ -703 +806 @@
         if ( defined('SAVEQUERIES') && SAVEQUERIES )
             $this->timer_start();
-
-        $this->result = @mysql_query($query, $this->dbh);
+        
+        // use $this->dbh for read ops, and $this->dbhwrite for write ops
+        // use $this->dbhglobal for gloal table ops
+        unset( $dbh );
+        if( defined( "WP_USE_MULTIPLE_DB" ) && CONSTANT( "WP_USE_MULTIPLE_DB" ) == true ) {
+            if( $this->blogs != '' && preg_match("/(" . $this->blogs . "|" . $this->users . "|" . $this->usermeta . "|" . $this->site . "|" . $this->sitemeta . "|" . $this->sitecategories . ")/i",$query) ) {
+                if( false == isset( $this->dbhglobal ) ) {
+                    $this->db_connect( $query );
+                }
+                $dbh =& $this->dbhglobal;
+                $this->last_db_used = "global";
+            } elseif ( preg_match("/^\\s*(alter table|create|insert|delete|update|replace) /i",$query) ) {
+                if( false == isset( $this->dbhwrite ) ) {
+                    $this->db_connect( $query );
+                }
+                $dbh =& $this->dbhwrite;
+                $this->last_db_used = "write";
+            } else {
+                $dbh =& $this->dbh;
+                $this->last_db_used = "read";
+            }
+        } else {
+            $dbh =& $this->dbh;
+            $this->last_db_used = "other/read";
+        }
+
+        $this->result = @mysql_query($query, $dbh);
         ++$this->num_queries;
 
@@ -711 +839 @@
 
         // If there is an error then take note of it..
-        if ( $this->last_error = mysql_error($this->dbh) ) {
+        if ( $this->last_error = mysql_error($dbh) ) {
             $this->print_error();
             return false;
@@ -717 +845 @@
 
         if ( preg_match("/^\\s*(insert|delete|update|replace|alter) /i",$query) ) {
-            $this->rows_affected = mysql_affected_rows($this->dbh);
+            $this->rows_affected = mysql_affected_rows($dbh);
             // Take note of the insert_id
             if ( preg_match("/^\\s*(insert|replace) /i",$query) ) {
-                $this->insert_id = mysql_insert_id($this->dbh);
+                $this->insert_id = mysql_insert_id($dbh);
             }
             // Return number of rows affected
@@ -1045 +1173 @@
      * @since 2.5.0
      * @uses $wp_version
-     * @uses $required_mysql_version
      *
      * @return WP_Error