ViewCountUpdate.php

Go to the documentation of this file.

<?php
class ViewCountUpdate implements DeferrableUpdate {
        protected $id;

        public function __construct( $id ) {
                $this->id = intval( $id );
        }

        public function doUpdate() {
                global $wgHitcounterUpdateFreq;

                $dbw = wfGetDB( DB_MASTER );

                if ( $wgHitcounterUpdateFreq <= 1 || $dbw->getType() == 'sqlite' ) {
                        $pageTable = $dbw->tableName( 'page' );
                        $dbw->query( "UPDATE $pageTable SET page_counter = page_counter + 1 WHERE page_id = {$this->id}" );
                        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 );

                $hitcounterTable = $dbw->tableName( 'hitcounter' );
                $res = $dbw->query( "SELECT COUNT(*) as n FROM $hitcounterTable" );
                $row = $dbw->fetchObject( $res );
                $rown = intval( $row->n );

                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 " : '';
                $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' );
        }
}