SqlDataUpdate.php

Go to the documentation of this file.

<?php
abstract class SqlDataUpdate extends DataUpdate {
    protected $mDb;

    protected $mOptions;

    private $mHasTransaction;

    protected $mUseTransaction;

    public function __construct( $withTransaction = true ) {
        global $wgAntiLockFlags;

        parent::__construct();

        if ( $wgAntiLockFlags & ALF_NO_LINK_LOCK ) {
            $this->mOptions = array();
        } else {
            $this->mOptions = array( 'FOR UPDATE' );
        }

        // @todo Get connection only when it's needed? Make sure that doesn't
        // break anything, especially transactions!
        $this->mDb = wfGetDB( DB_MASTER );

        $this->mWithTransaction = $withTransaction;
        $this->mHasTransaction = false;
    }

    public function beginTransaction() {
        if ( !$this->mWithTransaction ) {
            return;
        }

        // NOTE: nested transactions are not supported, only start a transaction if none is open
        if ( $this->mDb->trxLevel() === 0 ) {
            $this->mDb->begin( get_class( $this ) . '::beginTransaction' );
            $this->mHasTransaction = true;
        }
    }

    public function commitTransaction() {
        if ( $this->mHasTransaction ) {
            $this->mDb->commit( get_class( $this ) . '::commitTransaction' );
            $this->mHasTransaction = false;
        }
    }

    public function abortTransaction() {
        if ( $this->mHasTransaction ) { //XXX: actually... maybe always?
            $this->mDb->rollback( get_class( $this ) . '::abortTransaction' );
            $this->mHasTransaction = false;
        }
    }

    protected function invalidatePages( $namespace, array $dbkeys ) {
        if ( $dbkeys === array() ) {
            return;
        }

        $now = $this->mDb->timestamp();
        $ids = array();
        $res = $this->mDb->select( 'page', array( 'page_id' ),
            array(
                'page_namespace' => $namespace,
                'page_title' => $dbkeys,
                'page_touched < ' . $this->mDb->addQuotes( $now )
            ), __METHOD__
        );

        foreach ( $res as $row ) {
            $ids[] = $row->page_id;
        }

        if ( $ids === array() ) {
            return;
        }

        $this->mDb->update( 'page', array( 'page_touched' => $now ),
            array(
                'page_id' => $ids,
                'page_touched < ' . $this->mDb->addQuotes( $now )
            ), __METHOD__
        );
    }
}