CloneDatabase.php

Go to the documentation of this file.

<?php
class CloneDatabase {

        private $newTablePrefix = '';

        private $oldTablePrefix = '';

        private $tablesToClone = array();

        private $dropCurrentTables = true;

        private $useTemporaryTables = true;

        public function __construct( DatabaseBase $db, array $tablesToClone,
                $newTablePrefix, $oldTablePrefix = '', $dropCurrentTables = true )
        {
                $this->db = $db;
                $this->tablesToClone = $tablesToClone;
                $this->newTablePrefix = $newTablePrefix;
                $this->oldTablePrefix = $oldTablePrefix ? $oldTablePrefix : $this->db->tablePrefix();
                $this->dropCurrentTables = $dropCurrentTables;
        }

        public function useTemporaryTables( $u = true ) {
                $this->useTemporaryTables = $u;
        }

        public function cloneTableStructure() {
                foreach( $this->tablesToClone as $tbl ) {
                        # Clean up from previous aborted run.  So that table escaping
                        # works correctly across DB engines, we need to change the pre-
                        # fix back and forth so tableName() works right.

                        self::changePrefix( $this->oldTablePrefix );
                        $oldTableName = $this->db->tableName( $tbl, 'raw' );

                        self::changePrefix( $this->newTablePrefix );
                        $newTableName = $this->db->tableName( $tbl, 'raw' );

                        if( $this->dropCurrentTables && !in_array( $this->db->getType(), array( 'postgres', 'oracle' ) ) ) {
                                $this->db->dropTable( $tbl, __METHOD__ );
                                wfDebug( __METHOD__ . " dropping {$newTableName}\n", true );
                                //Dropping the oldTable because the prefix was changed
                        }

                        # Create new table
                        wfDebug( __METHOD__ . " duplicating $oldTableName to $newTableName\n", true );
                        $this->db->duplicateTableStructure( $oldTableName, $newTableName, $this->useTemporaryTables );
                }
        }

        public function destroy( $dropTables = false ) {
                if( $dropTables ) {
                        self::changePrefix( $this->newTablePrefix );
                        foreach( $this->tablesToClone as $tbl ) {
                                $this->db->dropTable( $tbl );
                        }
                }
                self::changePrefix( $this->oldTablePrefix );
        }

        public static function changePrefix( $prefix ) {
                global $wgDBprefix;
                wfGetLBFactory()->forEachLB( array( 'CloneDatabase', 'changeLBPrefix' ), array( $prefix ) );
                $wgDBprefix = $prefix;
        }

        public static function changeLBPrefix( $lb, $prefix ) {
                $lb->forEachOpenConnection( array( 'CloneDatabase', 'changeDBPrefix' ), array( $prefix ) );
        }

        public static function changeDBPrefix( $db, $prefix ) {
                $db->tablePrefix( $prefix );
        }
}