Merge "Less wild whitespace"

[lhc/web/wiklou.git] / includes / ViewCountUpdate.php

diff --git a/includes/ViewCountUpdate.php b/includes/ViewCountUpdate.php
index b2d9b93..28ba341 100644 (file)
--- a/includes/ViewCountUpdate.php
+++ b/includes/ViewCountUpdate.php
@@ -1,37 +1,108 @@
 <?php
 /**
- * See deferred.doc
- * @version # $Id$
- * @package MediaWiki
+ * Update for the 'page_counter' field
+ *
+ * This program is free software; you can redistribute it and/or modify
+ * it under the terms of the GNU General Public License as published by
+ * the Free Software Foundation; either version 2 of the License, or
+ * (at your option) any later version.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License along
+ * with this program; if not, write to the Free Software Foundation, Inc.,
+ * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
+ * http://www.gnu.org/copyleft/gpl.html
+ *
+ * @file
  */
 
 /**
+ * Update for the 'page_counter' field, when $wgDisableCounters is false.
  *
- * @version # $Id$ 
- * @package MediaWiki
+ * Depending on $wgHitcounterUpdateFreq, this will directly increment the
+ * 'page_counter' field or use the 'hitcounter' table and then collect the data
+ * from that table to update the 'page_counter' field in a batch operation.
  */
-class ViewCountUpdate {
-
-       var $mPageID;
+class ViewCountUpdate implements DeferrableUpdate {
+       protected $id;
 
        /**
+        * Constructor
         *
+        * @param $id Integer: page ID to increment the view count
         */
-       function ViewCountUpdate( $pageid ) {
-               $this->mPageID = $pageid;
+       public function __construct( $id ) {
+               $this->id = intval( $id );
        }
 
        /**
-        *
+        * Run the update
         */
-       function doUpdate() {
-               global $wgDisableCounters;
-               if ( $wgDisableCounters ) { return; }
-               $db =& wfGetDB( DB_MASTER );
-               $lowpri = $db->lowPriorityOption();
-               $sql = "UPDATE $lowpri cur SET cur_counter=(1+cur_counter)," .
-                 "cur_timestamp=cur_timestamp WHERE cur_id={$this->mPageID}";
-               $res = $db->query( $sql, "ViewCountUpdate::doUpdate" );
+       public function doUpdate() {
+               global $wgHitcounterUpdateFreq;
+
+               $dbw = wfGetDB( DB_MASTER );
+
+               if ( $wgHitcounterUpdateFreq <= 1 || $dbw->getType() == 'sqlite' ) {
+                       $dbw->update( 'page', array( 'page_counter = page_counter + 1' ), array( 'page_id' => $this->id ), __METHOD__ );
+                       return;
+               }
+
+               # Not important enough to warrant an error page in case of failure
+               $oldignore = $dbw->ignoreErrors( true );
+
+               $dbw->insert( 'hitcounter', array( 'hc_id' => $this->id ), __METHOD__ );
+
+               $checkfreq = intval( $wgHitcounterUpdateFreq / 25 + 1 );
+               if ( rand() % $checkfreq == 0 && $dbw->lastErrno() == 0 ) {
+                       $this->collect();
+               }
+
+               $dbw->ignoreErrors( $oldignore );
+       }
+
+       protected function collect() {
+               global $wgHitcounterUpdateFreq;
+
+               $dbw = wfGetDB( DB_MASTER );
+
+               $rown = $dbw->selectField( 'hitcounter', 'COUNT(*)', array(), __METHOD__ );
+
+               if ( $rown < $wgHitcounterUpdateFreq ) {
+                       return;
+               }
+
+               wfProfileIn( __METHOD__ . '-collect' );
+               $old_user_abort = ignore_user_abort( true );
+
+               $dbw->lockTables( array(), array( 'hitcounter' ), __METHOD__, false );
+
+               $dbType = $dbw->getType();
+               $tabletype = $dbType == 'mysql' ? "ENGINE=HEAP " : '';
+               $hitcounterTable = $dbw->tableName( 'hitcounter' );
+               $acchitsTable = $dbw->tableName( 'acchits' );
+               $pageTable = $dbw->tableName( 'page' );
+
+               $dbw->query( "CREATE TEMPORARY TABLE $acchitsTable $tabletype AS " .
+                       "SELECT hc_id,COUNT(*) AS hc_n FROM $hitcounterTable " .
+                       'GROUP BY hc_id', __METHOD__ );
+               $dbw->delete( 'hitcounter', '*', __METHOD__ );
+               $dbw->unlockTables( __METHOD__ );
+
+               if ( $dbType == 'mysql' ) {
+                       $dbw->query( "UPDATE $pageTable,$acchitsTable SET page_counter=page_counter + hc_n " .
+                               'WHERE page_id = hc_id', __METHOD__ );
+               } else {
+                       $dbw->query( "UPDATE $pageTable SET page_counter=page_counter + hc_n " .
+                               "FROM $acchitsTable WHERE page_id = hc_id", __METHOD__ );
+               }
+               $dbw->query( "DROP TABLE $acchitsTable", __METHOD__ );
+
+               ignore_user_abort( $old_user_abort );
+               wfProfileOut( __METHOD__ . '-collect' );
        }
 }
-?>