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

}