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                         $pageTable = $dbw->tableName( 'page' );
  52                         $dbw->query( "UPDATE $pageTable SET page_counter = page_counter + 1 WHERE page_id = {$this->id}" );
  53                         return;
  54                 }
  55
  56                 # Not important enough to warrant an error page in case of failure
  57                 $oldignore = $dbw->ignoreErrors( true );
  58
  59                 $dbw->insert( 'hitcounter', array( 'hc_id' => $this->id ), __METHOD__ );
  60
  61                 $checkfreq = intval( $wgHitcounterUpdateFreq / 25 + 1 );
  62                 if ( rand() % $checkfreq == 0 && $dbw->lastErrno() == 0 ) {
  63                         $this->collect();
  64                 }
  65
  66                 $dbw->ignoreErrors( $oldignore );
  67         }
  68
  69         protected function collect() {
  70                 global $wgHitcounterUpdateFreq;
  71
  72                 $dbw = wfGetDB( DB_MASTER );
  73
  74                 $hitcounterTable = $dbw->tableName( 'hitcounter' );
  75                 $res = $dbw->query( "SELECT COUNT(*) as n FROM $hitcounterTable" );
  76                 $row = $dbw->fetchObject( $res );
  77                 $rown = intval( $row->n );
  78
  79                 if ( $rown < $wgHitcounterUpdateFreq ) {
  80                         return;
  81                 }
  82
  83                 wfProfileIn( __METHOD__ . '-collect' );
  84                 $old_user_abort = ignore_user_abort( true );
  85
  86                 $dbw->lockTables( array(), array( 'hitcounter' ), __METHOD__, false );
  87
  88                 $dbType = $dbw->getType();
  89                 $tabletype = $dbType == 'mysql' ? "ENGINE=HEAP " : '';
  90                 $acchitsTable = $dbw->tableName( 'acchits' );
  91                 $pageTable = $dbw->tableName( 'page' );
  92
  93                 $dbw->query( "CREATE TEMPORARY TABLE $acchitsTable $tabletype AS " .
  94                         "SELECT hc_id,COUNT(*) AS hc_n FROM $hitcounterTable " .
  95                         'GROUP BY hc_id', __METHOD__ );
  96                 $dbw->delete( 'hitcounter', '*', __METHOD__ );
  97                 $dbw->unlockTables( __METHOD__ );
  98
  99                 if ( $dbType == 'mysql' ) {
 100                         $dbw->query( "UPDATE $pageTable,$acchitsTable SET page_counter=page_counter + hc_n " .
 101                                 'WHERE page_id = hc_id', __METHOD__ );
 102                 } else {
 103                         $dbw->query( "UPDATE $pageTable SET page_counter=page_counter + hc_n " .
 104                                 "FROM $acchitsTable WHERE page_id = hc_id", __METHOD__ );
 105                 }
 106                 $dbw->query( "DROP TABLE $acchitsTable", __METHOD__ );
 107
 108                 ignore_user_abort( $old_user_abort );
 109                 wfProfileOut( __METHOD__ . '-collect' );
 110         }
 111 }