CloneDatabase.php

Go to the documentation of this file.

<?php
use MediaWiki\MediaWikiServices;
use Wikimedia\Rdbms\IMaintainableDatabase;

class CloneDatabase {
    private $newTablePrefix;

    private $oldTablePrefix;

    private $tablesToClone;

    private $dropCurrentTables;

    private $useTemporaryTables = true;

    private $db;

    public function __construct( IMaintainableDatabase $db, array $tablesToClone,
        $newTablePrefix, $oldTablePrefix = null, $dropCurrentTables = true
    ) {
        if ( !$tablesToClone ) {
            throw new InvalidArgumentException( 'Empty list of tables to clone' );
        }
        $this->db = $db;
        $this->tablesToClone = $tablesToClone;
        $this->newTablePrefix = $newTablePrefix;
        $this->oldTablePrefix = $oldTablePrefix ?? $this->db->tablePrefix();
        $this->dropCurrentTables = $dropCurrentTables;
    }

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

    public function cloneTableStructure() {
        global $wgSharedTables, $wgSharedDB;
        foreach ( $this->tablesToClone as $tbl ) {
            if ( $wgSharedDB && in_array( $tbl, $wgSharedTables, true ) ) {
                // Shared tables don't work properly when cloning due to
                // how prefixes are handled (T67654)
                throw new RuntimeException( "Cannot clone shared table $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.

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

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

            // Postgres: Temp tables are automatically deleted upon end of session
            //           Same Temp table name hides existing table for current session
            if ( $this->dropCurrentTables ) {
                if ( $oldTableName === $newTableName ) {
                    // Last ditch check to avoid data loss
                    throw new LogicException( "Not dropping new table, as '$newTableName'"
                        . " is name of both the old and the new table." );
                }
                $this->db->dropTable( $tbl, __METHOD__ );
                wfDebug( __METHOD__ . " dropping {$newTableName}\n" );
                // Dropping the oldTable because the prefix was changed
            }

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

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

    public static function changePrefix( $prefix ) {
        global $wgDBprefix;

        $lbFactory = MediaWikiServices::getInstance()->getDBLoadBalancerFactory();
        $lbFactory->setLocalDomainPrefix( $prefix );
        $wgDBprefix = $prefix;
    }
}