split getOrderLimit() into getOrder() and Database::limitResult()
[lhc/web/wiklou.git] / includes / QueryPage.php
index 6ffd116..4952d2a 100644 (file)
@@ -27,7 +27,7 @@ class QueryPage {
        function getSQL() {
                return "SELECT 'sample' as type, 0 as namespace, 'Sample result' as title, 42 as value";
        }
-       
+
        # Override to sort by increasing values
        function sortDescending() {
                return true;
@@ -37,7 +37,12 @@ class QueryPage {
        function getOrderLimit( $offset, $limit ) {
                return " ORDER BY value " .
                        ($this->sortDescending() ? "DESC" : "")
-                       . " LIMIT {$offset}, {$limit}";
+                       . wfLimitResult($limit,$offset);
+       }
+
+       function getOrder() {
+               return " ORDER BY value " .
+                       ($this->sortDescending() ? "DESC" : "");
        }
 
        # Is this query expensive (for some definition of expensive)? Then we
@@ -55,49 +60,79 @@ class QueryPage {
        function formatResult( $skin, $result ) {
                return "";
        }
-       
+
        # This is the actual workhorse. It does everything needed to make a
        # real, honest-to-gosh query page.
-       
+
        function doQuery( $offset, $limit ) {
                global $wgUser, $wgOut, $wgLang, $wgRequest;
                global $wgMiserMode;
 
                $sname = $this->getName();
                $fname = get_class($this) . "::doQuery";
-               $sql = $this->getSQL( $offset, $limit );
+               $sql = $this->getSQL();
+               $dbr =& wfGetDB( DB_SLAVE );
+               $dbw =& wfGetDB( DB_MASTER );
+               $querycache = $dbr->tableName( 'querycache' );
 
                $wgOut->setSyndicated( true );
-               
+               $res = false;
+
                if ( $this->isExpensive() ) {
-                       $type = wfStrencode( $sname );
                        $recache = $wgRequest->getBool( "recache" );
                        if( $recache ) {
                                # Clear out any old cached data
-                               $res = wfQuery( "DELETE FROM querycache WHERE qc_type='$type'", DB_WRITE, $fname );
+                               $dbw->delete( 'querycache', array( 'qc_type' => $sname ), $fname );
 
-                               # Save results into the querycache table
+                               # Do query on the (possibly out of date) slave server
                                $maxstored = 1000;
-                               $res = wfQuery(
-                                       "INSERT INTO querycache(qc_type,qc_namespace,qc_title,qc_value) " .
-                                       $this->getSQL() .
-                                       $this->getOrderLimit( 0, $maxstored ),
-                                       DB_WRITE, $fname );
+                               $res = $dbr->query( $sql . $this->getOrder() . $dbr->limitResult( $maxstored,0 ), $fname );
+
+                               # Fetch results
+                               $insertSql = "INSERT INTO $querycache (qc_type,qc_namespace,qc_title,qc_value) VALUES ";
+                               $first = true;
+                               while ( $row = $dbr->fetchObject( $res ) ) {
+                                       if ( $first ) {
+                                               $first = false;
+                                       } else {
+                                               $insertSql .= ",";
+                                       }
+                                       $insertSql .= "(" .
+                                               $dbw->addQuotes( $row->type ) . "," .
+                                               $dbw->addQuotes( $row->namespace ) . "," .
+                                               $dbw->addQuotes( $row->title ) . "," .
+                                               $dbw->addQuotes( $row->value ) . ")";
+                               }
+
+                               # Save results into the querycache table on the master
+                               $dbw->query( $insertSql, $fname );
+
+                               # Set result pointer to allow reading for display
+                               $numRows = $dbr->numRows( $res );
+                               if ( $numRows <= $offset ) {
+                                       $num = 0;
+                               } else {
+                                       $dbr->dataSeek( $res, $offset );
+                                       $num = max( $limit, $numRows - $offset );
+                               }
                        }
                        if( $wgMiserMode || $recache ) {
+                               $type = $dbr->strencode( $sname );
                                $sql =
                                        "SELECT qc_type as type, qc_namespace as namespace,qc_title as title, qc_value as value
-                                        FROM querycache WHERE qc_type='$type'";
+                                        FROM $querycache WHERE qc_type='$type'";
                        }
                        if( $wgMiserMode ) {
                                $wgOut->addWikiText( wfMsg( "perfcached" ) );
                        }
                }
+               if ( $res === false ) {
+                       $res = $dbr->query( $sql . $this->getOrder() . 
+                                           $dbr->limitResult( $limit,$offset ), $fname );
+                       $num = $dbr->numRows($res);
+               }
+
 
-               $res = wfQuery( $sql . $this->getOrderLimit( $offset, $limit ), DB_READ, $fname );
-               
-               $num = wfNumRows($res);
-               
                $sk = $wgUser->getSkin( );
 
                $top = wfShowingResults( $offset, $num);
@@ -105,21 +140,23 @@ class QueryPage {
 
                # often disable 'next' link when we reach the end
                if($num < $limit) { $atend = true; } else { $atend = false; }
-               
+
                $sl = wfViewPrevNext( $offset, $limit , $wgLang->specialPage( $sname ), "" ,$atend );
                $wgOut->addHTML( "<br />{$sl}</p>\n" );
 
-               $s = "<ol start='" . ( $offset + 1 ) . "'>";
-               while ( $obj = wfFetchObject( $res ) ) {
+               $s = "<ol start='" . ( $offset + 1 ) . "' class='special'>";
+               # Only read at most $num rows, because $res may contain the whole 1000
+               for ( $i = 0; $i < $num && $obj = $dbr->fetchObject( $res ); $i++ ) {
                        $format = $this->formatResult( $sk, $obj );
-                       $s .= "<li>{$format}</li>\n";
+                       $attr = ( $obj->usepatrol && $obj->patrolled == 0 ) ? ' class="not_patrolled"' : '';
+                       $s .= "<li{$attr}>{$format}</li>\n";
                }
-               wfFreeResult( $res );
+               $dbr->freeResult( $res );
                $s .= "</ol>";
                $wgOut->addHTML( $s );
                $wgOut->addHTML( "<p>{$sl}</p>\n" );
        }
-       
+
        # Similar to above, but packaging in a syndicated feed instead of a web page
        function doFeed( $class = "" ) {
                global $wgFeedClasses;
@@ -130,14 +167,15 @@ class QueryPage {
                                $this->feedDesc(),
                                $this->feedUrl() );
                        $feed->outHeader();
-                       
-                       $sql = $this->getSQL( 0, 50 );
-                       $res = wfQuery( $sql, DB_READ, "QueryPage::doFeed" );
-                       while( $obj = wfFetchObject( $res ) ) {
+
+                       $dbr =& wfGetDB( DB_SLAVE );
+                       $sql = $this->getSQL() . $this->getOrder().$dbr->limitResult( 50, 0 );
+                       $res = $dbr->query( $sql, "QueryPage::doFeed" );
+                       while( $obj = $dbr->fetchObject( $res ) ) {
                                $item = $this->feedResult( $obj );
                                if( $item ) $feed->outItem( $item );
                        }
-                       wfFreeResult( $res );
+                       $dbr->freeResult( $res );
 
                        $feed->outFooter();
                        return true;
@@ -158,13 +196,13 @@ class QueryPage {
                        } else {
                                $date = "";
                        }
-                       
+
                        $comments = "";
                        if( $title ) {
                                $talkpage = $title->getTalkPage();
                                $comments = $talkpage->getFullURL();
                        }
-                       
+
                        return new FeedItem(
                                $title->getText(),
                                $this->feedItemDesc( $row ),
@@ -176,7 +214,7 @@ class QueryPage {
                        return NULL;
                }
        }
-       
+
        function feedItemDesc( $row ) {
                $text = "";
                if( isset( $row->comment ) ) {
@@ -184,14 +222,14 @@ class QueryPage {
                } else {
                        $text = "";
                }
-               
+
                if( isset( $row->text ) ) {
                        $text = "<p>" . htmlspecialchars( wfMsg( "summary" ) ) . ": " . $text . "</p>\n<hr />\n<div>" .
                                nl2br( htmlspecialchars( $row->text ) ) . "</div>";;
                }
                return $text;
        }
-       
+
        function feedItemAuthor( $row ) {
                if( isset( $row->user_text ) ) {
                        return $row->user_text;
@@ -199,18 +237,18 @@ class QueryPage {
                        return "";
                }
        }
-       
+
        function feedTitle() {
                global $wgLanguageCode, $wgSitename, $wgLang;
                $page = SpecialPage::getPage( $this->getName() );
                $desc = $page->getDescription();
                return "$wgSitename - $desc [$wgLanguageCode]";
        }
-       
+
        function feedDesc() {
                return wfMsg( "fromwikipedia" );
        }
-       
+
        function feedUrl() {
                global $wgLang;
                $title = Title::MakeTitle( NS_SPECIAL, $this->getName() );