includes/ViewCountUpdate.php

   1 <?php
   2 /**
   3  * Update for the 'page_counter' field
   4  *
   5  * This program is free software; you can redistribute it and/or modify
   6  * it under the terms of the GNU General Public License as published by
   7  * the Free Software Foundation; either version 2 of the License, or
   8  * (at your option) any later version.
   9  *
  10  * This program is distributed in the hope that it will be useful,
  11  * but WITHOUT ANY WARRANTY; without even the implied warranty of
  12  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
  13  * GNU General Public License for more details.
  14  *
  15  * You should have received a copy of the GNU General Public License along
  16  * with this program; if not, write to the Free Software Foundation, Inc.,
  17  * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
  18  * http://www.gnu.org/copyleft/gpl.html
  19  *
  20  * @file
  21  */
  22
  23 /**
  24  * Update for the 'page_counter' field, when $wgDisableCounters is false.
  25  *
  26  * Depending on $wgHitcounterUpdateFreq, this will directly increment the
  27  * 'page_counter' field or use the 'hitcounter' table and then collect the data
  28  * from that table to update the 'page_counter' field in a batch operation.
  29  */
  30 class ViewCountUpdate implements DeferrableUpdate {
  31         protected $id;
  32
  33         /**
  34          * Constructor
  35          *
  36          * @param $id Integer: page ID to increment the view count
  37          */
  38         public function __construct( $id ) {
  39                 $this->id = intval( $id );
  40         }
  41
  42         /**
  43          * Run the update
  44          */
  45         public function doUpdate() {
  46                 global $wgHitcounterUpdateFreq;
  47
  48                 $dbw = wfGetDB( DB_MASTER );
  49
  50                 if ( $wgHitcounterUpdateFreq <= 1 || $dbw->getType() == 'sqlite' ) {
  51                         $dbw->update( 'page', array( 'page_counter = page_counter + 1' ), array( 'page_id' => $this->id ), __METHOD__ );
  52                         return;
  53                 }
  54
  55                 # Not important enough to warrant an error page in case of failure
  56                 try {
  57                         $dbw->insert( 'hitcounter', array( 'hc_id' => $this->id ), __METHOD__ );
  58                         $checkfreq = intval( $wgHitcounterUpdateFreq / 25 + 1 );
  59                         if ( rand() % $checkfreq == 0 && $dbw->lastErrno() == 0 ) {
  60                                 $this->collect();
  61                         }
  62                 } catch ( DBError $e ) {}
  63         }
  64
  65         protected function collect() {
  66                 global $wgHitcounterUpdateFreq;
  67
  68                 $dbw = wfGetDB( DB_MASTER );
  69
  70                 $rown = $dbw->selectField( 'hitcounter', 'COUNT(*)', array(), __METHOD__ );
  71
  72                 if ( $rown < $wgHitcounterUpdateFreq ) {
  73                         return;
  74                 }
  75
  76                 wfProfileIn( __METHOD__ . '-collect' );
  77                 $old_user_abort = ignore_user_abort( true );
  78
  79                 $dbw->lockTables( array(), array( 'hitcounter' ), __METHOD__, false );
  80
  81                 $dbType = $dbw->getType();
  82                 $tabletype = $dbType == 'mysql' ? "ENGINE=HEAP " : '';
  83                 $hitcounterTable = $dbw->tableName( 'hitcounter' );
  84                 $acchitsTable = $dbw->tableName( 'acchits' );
  85                 $pageTable = $dbw->tableName( 'page' );
  86
  87                 $dbw->query( "CREATE TEMPORARY TABLE $acchitsTable $tabletype AS " .
  88                         "SELECT hc_id,COUNT(*) AS hc_n FROM $hitcounterTable " .
  89                         'GROUP BY hc_id', __METHOD__ );
  90                 $dbw->delete( 'hitcounter', '*', __METHOD__ );
  91                 $dbw->unlockTables( __METHOD__ );
  92
  93                 if ( $dbType == 'mysql' ) {
  94                         $dbw->query( "UPDATE $pageTable,$acchitsTable SET page_counter=page_counter + hc_n " .
  95                                 'WHERE page_id = hc_id', __METHOD__ );
  96                 } else {
  97                         $dbw->query( "UPDATE $pageTable SET page_counter=page_counter + hc_n " .
  98                                 "FROM $acchitsTable WHERE page_id = hc_id", __METHOD__ );
  99                 }
 100                 $dbw->query( "DROP TABLE $acchitsTable", __METHOD__ );
 101
 102                 ignore_user_abort( $old_user_abort );
 103                 wfProfileOut( __METHOD__ . '-collect' );
 104         }
 105 }