MssqlUpdater.php

Go to the documentation of this file.

<?php
class MssqlUpdater extends DatabaseUpdater {

    protected $db;

    protected function getCoreUpdateList() {
        return array(
            // 1.23
            array( 'addField', 'mwuser', 'user_password_expires', 'patch-user_password_expires.sql' ),

            // 1.24
            array( 'addField', 'page', 'page_lang', 'patch-page-page_lang.sql'),
            // Constraint updates
            array( 'updateConstraints', 'category_types', 'categorylinks', 'cl_type' ),
            array( 'updateConstraints', 'major_mime', 'filearchive', 'fa_major_mime' ),
            array( 'updateConstraints', 'media_type', 'filearchive', 'fa_media_type' ),
            array( 'updateConstraints', 'major_mime', 'oldimage', 'oi_major_mime' ),
            array( 'updateConstraints', 'media_type', 'oldimage', 'oi_media_type' ),
            array( 'updateConstraints', 'major_mime', 'image', 'img_major_mime' ),
            array( 'updateConstraints', 'media_type', 'image', 'img_media_type' ),
            array( 'updateConstraints', 'media_type', 'uploadstash', 'us_media_type' ),
            // END: Constraint updates

            array( 'modifyField', 'image', 'img_major_mime',
                'patch-img_major_mime-chemical.sql' ),
            array( 'modifyField', 'oldimage', 'oi_major_mime',
                'patch-oi_major_mime-chemical.sql' ),
            array( 'modifyField', 'filearchive', 'fa_major_mime',
                'patch-fa_major_mime-chemical.sql' ),
        );
    }

    protected function updateConstraints( $constraintType, $table, $field ) {
        global $wgDBname, $wgDBmwschema;

        if ( !$this->doTable( $table ) ) {
            return true;
        }

        $this->output( "...updating constraints on [$table].[$field] ..." );
        $updateKey = "$field-$constraintType-ck";
        if ( !$this->db->tableExists( $table, __METHOD__ ) ) {
            $this->output( "...$table table does not exist, skipping modify field patch.\n" );
            return true;
        } elseif ( !$this->db->fieldExists( $table, $field, __METHOD__ ) ) {
            $this->output( "...$field field does not exist in $table table, " .
                "skipping modify field patch.\n" );
            return true;
        } elseif ( $this->updateRowExists( $updateKey ) ) {
            $this->output( "...$field in table $table already patched.\n" );
            return true;
        }

        # After all checks passed, start the update
        $this->insertUpdateRow( $updateKey );
        $path = 'named_constraints.sql';
        $constraintMap = array(
            'category_types' =>
                "($field in('page', 'subcat', 'file'))",
            'major_mime'     =>
                "($field in('unknown', 'application', 'audio', 'image', 'text', 'video'," .
                " 'message', 'model', 'multipart'))",
            'media_type'     =>
                "($field in('UNKNOWN', 'BITMAP', 'DRAWING', 'AUDIO', 'VIDEO', 'MULTIMEDIA'," .
                "'OFFICE', 'TEXT', 'EXECUTABLE', 'ARCHIVE'))"
        );
        $constraint = $constraintMap[$constraintType];

        # and hack-in those variables that should be replaced
        # in our template file right now
        $this->db->setSchemaVars( array(
            'tableName'       => $table,
            'fieldName'       => $field,
            'checkConstraint' => $constraint,
            'wgDBname'        => $wgDBname,
            'wgDBmwschema'    => $wgDBmwschema,
        ) );

        # Full path from file name
        $path = $this->db->patchPath( $path );

        # No need for a cursor allowing result-iteration; just apply a patch
        # store old value for re-setting later
        $wasScrollable = $this->db->scrollableCursor( false );

        # Apply patch
        $this->db->sourceFile( $path );

        # Reset DB instance to have original state
        $this->db->setSchemaVars( false );
        $this->db->scrollableCursor( $wasScrollable );

        $this->output( "done.\n" );

        return true;
    }
}