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__
        );
    }
}