3 * Base code for update jobs that put some secondary data extracted
4 * from article content into the database.
6 * This program is free software; you can redistribute it and/or modify
7 * it under the terms of the GNU General Public License as published by
8 * the Free Software Foundation; either version 2 of the License, or
9 * (at your option) any later version.
11 * This program is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 * GNU General Public License for more details.
16 * You should have received a copy of the GNU General Public License along
17 * with this program; if not, write to the Free Software Foundation, Inc.,
18 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
19 * http://www.gnu.org/copyleft/gpl.html
25 * Abstract base class for update jobs that put some secondary data extracted
26 * from article content into the database.
28 * @note: subclasses should NOT start or commit transactions in their doUpdate() method,
29 * a transaction will automatically be wrapped around the update. Starting another
30 * one would break the outer transaction bracket. If need be, subclasses can override
31 * the beginTransaction() and commitTransaction() methods.
33 abstract class SqlDataUpdate
extends DataUpdate
{
34 /** @var DatabaseBase Database connection reference */
37 /** @var array SELECT options to be used (array) */
40 /** @var bool Whether a transaction is open on this object (internal use only!) */
41 private $mHasTransaction;
43 /** @var bool Whether this update should be wrapped in a transaction */
44 protected $mUseTransaction;
49 * @param bool $withTransaction whether this update should be wrapped in a
50 * transaction (default: true). A transaction is only started if no
51 * transaction is already in progress, see beginTransaction() for details.
53 public function __construct( $withTransaction = true ) {
54 global $wgAntiLockFlags;
56 parent
::__construct();
58 if ( $wgAntiLockFlags & ALF_NO_LINK_LOCK
) {
59 $this->mOptions
= array();
61 $this->mOptions
= array( 'FOR UPDATE' );
64 // @todo Get connection only when it's needed? Make sure that doesn't
65 // break anything, especially transactions!
66 $this->mDb
= wfGetDB( DB_MASTER
);
68 $this->mWithTransaction
= $withTransaction;
69 $this->mHasTransaction
= false;
73 * Begin a database transaction, if $withTransaction was given as true in
74 * the constructor for this SqlDataUpdate.
76 * Because nested transactions are not supported by the Database class,
77 * this implementation checks Database::trxLevel() and only opens a
78 * transaction if none is already active.
80 public function beginTransaction() {
81 if ( !$this->mWithTransaction
) {
85 // NOTE: nested transactions are not supported, only start a transaction if none is open
86 if ( $this->mDb
->trxLevel() === 0 ) {
87 $this->mDb
->begin( get_class( $this ) . '::beginTransaction' );
88 $this->mHasTransaction
= true;
93 * Commit the database transaction started via beginTransaction (if any).
95 public function commitTransaction() {
96 if ( $this->mHasTransaction
) {
97 $this->mDb
->commit( get_class( $this ) . '::commitTransaction' );
98 $this->mHasTransaction
= false;
103 * Abort the database transaction started via beginTransaction (if any).
105 public function abortTransaction() {
106 if ( $this->mHasTransaction
) { //XXX: actually... maybe always?
107 $this->mDb
->rollback( get_class( $this ) . '::abortTransaction' );
108 $this->mHasTransaction
= false;
113 * Invalidate the cache of a list of pages from a single namespace.
114 * This is intended for use by subclasses.
116 * @param int $namespace Namespace number
117 * @param array $dbkeys
119 protected function invalidatePages( $namespace, array $dbkeys ) {
120 if ( $dbkeys === array() ) {
125 * Determine which pages need to be updated
126 * This is necessary to prevent the job queue from smashing the DB with
127 * large numbers of concurrent invalidations of the same page
129 $now = $this->mDb
->timestamp();
131 $res = $this->mDb
->select( 'page', array( 'page_id' ),
133 'page_namespace' => $namespace,
134 'page_title' => $dbkeys,
135 'page_touched < ' . $this->mDb
->addQuotes( $now )
139 foreach ( $res as $row ) {
140 $ids[] = $row->page_id
;
143 if ( $ids === array() ) {
149 * We still need the page_touched condition, in case the row has changed since
150 * the non-locking select above.
152 $this->mDb
->update( 'page', array( 'page_touched' => $now ),
155 'page_touched < ' . $this->mDb
->addQuotes( $now )