* (bug 1754) Patch by Anders Wegge Jakobsen: Both Atom and RSS were using the
[lhc/web/wiklou.git] / includes / LinksUpdate.php
index 98bc427..c161a33 100644 (file)
@@ -1,39 +1,52 @@
-<?
-# See deferred.doc
-
+<?php
+/**
+ * See deferred.txt
+ * @package MediaWiki
+ */
+
+/**
+ * @todo document
+ * @package MediaWiki
+ */
 class LinksUpdate {
 
-       /* private */ var $mId, $mTitle;
-
-       function LinksUpdate( $id, $title )
-       {
+       /**#@+
+        * @access private
+        */
+       var $mId, $mTitle;
+       /**#@-*/
+
+       /**
+        * Constructor
+        * Initialize private variables
+        * @param integer $id
+        * @param string $title
+        */
+       function LinksUpdate( $id, $title ) {
                $this->mId = $id;
                $this->mTitle = $title;
-               $this->mTitleEnc = wfStrencode( $title );
        }
 
-       function doUpdate()
-       {
+       /**
+        * Update link tables with outgoing links from an updated article
+        * Relies on the 'link cache' to be filled out.
+        */
+       
+       function doUpdate() {
                global $wgUseBetterLinksUpdate, $wgLinkCache, $wgDBtransactions;
-               
-               /* Update link tables with outgoing links from an updated article */
-               /* Relies on the 'link cache' to be filled out */
-
-               if ( !$wgUseBetterLinksUpdate ) {
-                       $this->doDumbUpdate();
-                       return;
-               }
+               global $wgEnablePersistentLC, $wgUseCategoryMagic;
 
-               $fname = "LinksUpdate::doUpdate";
+               $fname = 'LinksUpdate::doUpdate';
                wfProfileIn( $fname );
 
                $del = array();
                $add = array();
 
-               if( $wgDBtransactions ) {
-                       $sql = "BEGIN";
-                       wfQuery( $sql, DB_WRITE, $fname );
-               }
+               $dbw =& wfGetDB( DB_MASTER );
+               $links = $dbw->tableName( 'links' );
+               $brokenlinks = $dbw->tableName( 'brokenlinks' );
+               $imagelinks = $dbw->tableName( 'imagelinks' );
+               $categorylinks = $dbw->tableName( 'categorylinks' );
                
                #------------------------------------------------------------------------------
                # Good links
@@ -41,34 +54,29 @@ class LinksUpdate {
                if ( $wgLinkCache->incrementalSetup( LINKCACHE_GOOD, $del, $add ) ) {
                        # Delete where necessary
                        if ( count( $del ) ) {
-                               $sql = "DELETE FROM links WHERE l_from='{$this->mTitleEnc}' AND l_to IN(".
-                                       implode( ",", $del ) . ")";
-                               wfQuery( $sql, DB_WRITE, $fname );
+                               $sql = "DELETE FROM $links WHERE l_from={$this->mId} AND l_to IN(".
+                                       implode( ',', $del ) . ')';
+                               $dbw->query( $sql, $fname );
                        }
                } else {
                        # Delete everything
-                       $sql = "DELETE FROM links WHERE l_from='{$this->mTitleEnc}'";
-                       wfQuery( $sql, DB_WRITE, $fname );
-                       
+                       $dbw->delete( 'links', array( 'l_from' => $this->mId ), $fname );
+                                               
                        # Get the addition list
                        $add = $wgLinkCache->getGoodLinks();
                }
 
                # Do the insertion
-               $sql = "";
                if ( 0 != count( $add ) ) {
-                       $sql = "INSERT INTO links (l_from,l_to) VALUES ";
-                       $first = true;
-                       foreach( $add as $lt => $lid ) {
-                               
-                               if ( ! $first ) { $sql .= ","; }
-                               $first = false;
-
-                               $sql .= "('{$this->mTitleEnc}',{$lid})";
-                       }
-               }
-               if ( "" != $sql ) { 
-                       wfQuery( $sql, DB_WRITE, $fname ); 
+                       $arr=array();
+                       foreach($add as $lt=>$lid) 
+                               array_push( $arr, array(
+                                                       'l_from' => $this->mId,
+                                                       'l_to'   => $lid ) );
+                       # The link cache was constructed without FOR UPDATE, so there may be collisions
+                       # Ignoring for now, I'm not sure if that causes problems or not, but I'm fairly
+                       # sure it's better than without IGNORE
+                       $dbw->insert( 'links', $arr, $fname, array( 'IGNORE' ) );
                }
 
                #------------------------------------------------------------------------------
@@ -77,171 +85,215 @@ class LinksUpdate {
                if ( $wgLinkCache->incrementalSetup( LINKCACHE_BAD, $del, $add ) ) {
                        # Delete where necessary
                        if ( count( $del ) ) {
-                               $sql = "DELETE FROM brokenlinks WHERE bl_from={$this->mId} AND bl_to IN('" .    
-                                       implode( "','", $del ) . "')";
-                               wfQuery( $sql, DB_WRITE, $fname );
+                               $sql = "DELETE FROM $brokenlinks WHERE bl_from={$this->mId} AND bl_to IN(";
+                               $first = true;
+                               foreach( $del as $badTitle ) {
+                                       if ( $first ) {
+                                               $first = false;
+                                       } else {
+                                               $sql .= ',';
+                                       }
+                                       $sql .= $dbw->addQuotes( $badTitle );
+                               }
+                               $sql .= ')';
+                               $dbw->query( $sql, $fname );
                        }
                } else {
                        # Delete all
-                       $sql = "DELETE FROM brokenlinks WHERE bl_from={$this->mId}";
-                       wfQuery( $sql, DB_WRITE, $fname );
+                       $dbw->delete( 'brokenlinks', array( 'bl_from' => $this->mId ) );
                        
                        # Get addition list
                        $add = $wgLinkCache->getBadLinks();
                }
 
                # Do additions
-               $sql = "";
+               $sql = '';
                if ( 0 != count ( $add ) ) {
-                       $sql = "INSERT INTO brokenlinks (bl_from,bl_to) VALUES ";
-                       $first = true;
+                       $arr = array();
                        foreach( $add as $blt ) {
-                               $blt = wfStrencode( $blt );
-                               if ( ! $first ) { $sql .= ","; }
-                               $first = false;
-
-                               $sql .= "({$this->mId},'{$blt}')";
+                               array_push( $arr, array(
+                                       'bl_from' => $this->mId,
+                                       'bl_to'   => $blt ) );
                        }
-               }
-               if ( "" != $sql ) { 
-                       wfQuery( $sql, DB_WRITE, $fname );
+                       $dbw->insert( 'brokenlinks', $arr, $fname, array( 'IGNORE' ) );
                }
 
                #------------------------------------------------------------------------------
                # Image links
-               $sql = "DELETE FROM imagelinks WHERE il_from='{$this->mTitleEnc}'";
-               wfQuery( $sql, DB_WRITE, $fname );
+               $sql = "DELETE FROM $imagelinks WHERE il_from='{$this->mId}'";
+               $dbw->query( $sql, $fname );
                
                # Get addition list
                $add = $wgLinkCache->getImageLinks();
                
                # Do the insertion
-               $sql = "";
+               $sql = '';
+               $image = NS_IMAGE;
                if ( 0 != count ( $add ) ) {
-                       $sql = "INSERT INTO imagelinks (il_from,il_to) VALUES ";
-                       $first = true;
-                       foreach( $add as $iname => $val ) {
-                               $iname = wfStrencode( $iname );
-                               if ( ! $first ) { $sql .= ","; }
-                               $first = false;
-
-                               $sql .= "('{$this->mTitleEnc}','{$iname}')";
+                       $arr = array();
+                       foreach ($add as $iname => $val ) {
+                               $nt = Title::makeTitle( $image, $iname );
+                               if( !$nt ) continue;
+                               $nt->invalidateCache();
+                               array_push( $arr, array(
+                                       'il_from' => $this->mId,
+                                       'il_to'   => $iname ) );
                        }
+                       $dbw->insert('imagelinks', $arr, $fname, array('IGNORE'));
                }
-               if ( "" != $sql ) { wfQuery( $sql, DB_WRITE, $fname ); }
 
+               #------------------------------------------------------------------------------
+               # Category links
+               if( $wgUseCategoryMagic ) {
+                       $sql = "DELETE FROM $categorylinks WHERE cl_from='{$this->mId}'";
+                       $dbw->query( $sql, $fname );
+                       
+                       # Get addition list
+                       $add = $wgLinkCache->getCategoryLinks();
+                       
+                       # Do the insertion
+                       $sql = '';
+                       if ( 0 != count ( $add ) ) {
+                               $arr = array();
+                               foreach( $add as $cname => $sortkey ) {
+                                       $nt = Title::makeTitle( NS_CATEGORY, $cname );
+                                       if( !$nt ) continue;
+                                       $nt->invalidateCache();
+                                       array_push( $arr, array(
+                                               'cl_from'    => $this->mId,
+                                               'cl_to'      => $cname,
+                                               'cl_sortkey' => $sortkey ) );
+                               }
+                               $dbw->insert( 'categorylinks', $arr, $fname, array( 'IGNORE' ) );
+                       }
+               }
+               
                $this->fixBrokenLinks();
 
-               if( $wgDBtransactions ) {
-                       $sql = "COMMIT";
-                       wfQuery( $sql, DB_WRITE, $fname );
-               }
                wfProfileOut( $fname );
        }
 
-       function doDumbUpdate()
-       {
-               # Old update function. This can probably be removed eventually, if the new one
-               # proves to be stable
-               global $wgLinkCache, $wgDBtransactions;
-               $fname = "LinksUpdate::doDumbUpdate";
+       /**
+         * Link update which clears the previous entries and inserts new ones
+         * May be slower or faster depending on level of lock contention and write speed of DB
+         * Also useful where link table corruption needs to be repaired, e.g. in refreshLinks.php
+        */
+       function doDumbUpdate() {
+               global $wgLinkCache, $wgDBtransactions, $wgUseCategoryMagic;
+               $fname = 'LinksUpdate::doDumbUpdate';
                wfProfileIn( $fname );
-
-               if( $wgDBtransactions ) {
-                       $sql = "BEGIN";
-                       wfQuery( $sql, DB_WRITE, $fname );
-               }
                
-               $sql = "DELETE FROM links WHERE l_from='{$this->mTitleEnc}'";
-               wfQuery( $sql, DB_WRITE, $fname );
+               
+               $dbw =& wfGetDB( DB_MASTER );
+               $links = $dbw->tableName( 'links' );
+               $brokenlinks = $dbw->tableName( 'brokenlinks' );
+               $imagelinks = $dbw->tableName( 'imagelinks' );
+               $categorylinks = $dbw->tableName( 'categorylinks' );
+               
+               $sql = "DELETE FROM $links WHERE l_from={$this->mId}";
+               $dbw->query( $sql, $fname );
 
                $a = $wgLinkCache->getGoodLinks();
-               $sql = "";
                if ( 0 != count( $a ) ) {
-                       $sql = "INSERT INTO links (l_from,l_to) VALUES ";
-                       $first = true;
+                       $arr = array();
                        foreach( $a as $lt => $lid ) {
-                               if ( ! $first ) { $sql .= ","; }
-                               $first = false;
-
-                               $sql .= "('{$this->mTitleEnc}',{$lid})";
+                               array_push( $arr, array(
+                                       'l_from' => $this->mId,
+                                       'l_to'   => $lid ) );
                        }
+                       $dbw->insert( 'links', $arr, $fname, array( 'IGNORE' ) );
                }
-               if ( "" != $sql ) { wfQuery( $sql, DB_WRITE, $fname ); }
 
-               $sql = "DELETE FROM brokenlinks WHERE bl_from={$this->mId}";
-               wfQuery( $sql, DB_WRITE, $fname );
+               $sql = "DELETE FROM $brokenlinks WHERE bl_from={$this->mId}";
+               $dbw->query( $sql, $fname );
 
                $a = $wgLinkCache->getBadLinks();
-               $sql = "";
                if ( 0 != count ( $a ) ) {
-                       $sql = "INSERT INTO brokenlinks (bl_from,bl_to) VALUES ";
-                       $first = true;
+                       $arr = array();
                        foreach( $a as $blt ) {
-                               $blt = wfStrencode( $blt );
-                               if ( ! $first ) { $sql .= ","; }
-                               $first = false;
-
-                               $sql .= "({$this->mId},'{$blt}')";
+                               array_push($arr,array(
+                                       'bl_from' => $this->mId,
+                                       'bl_to'   => $blt));
                        }
+                       $dbw->insert( 'brokenlinks', $arr, $fname, array( 'IGNORE' ) );
                }
-               if ( "" != $sql ) { wfQuery( $sql, DB_WRITE, $fname ); }
                
-               $sql = "DELETE FROM imagelinks WHERE il_from='{$this->mTitleEnc}'";
-               wfQuery( $sql, DB_WRITE, $fname );
+               $sql = "DELETE FROM $imagelinks WHERE il_from={$this->mId}";
+               $dbw->query( $sql, $fname );
 
                $a = $wgLinkCache->getImageLinks();
-               $sql = "";
+               $sql = '';
                if ( 0 != count ( $a ) ) {
-                       $sql = "INSERT INTO imagelinks (il_from,il_to) VALUES ";
-                       $first = true;
-                       foreach( $a as $iname => $val ) {
-                               $iname = wfStrencode( $iname );
-                               if ( ! $first ) { $sql .= ","; }
-                               $first = false;
-
-                               $sql .= "('{$this->mTitleEnc}','{$iname}')";
-                       }
+                       $arr = array();
+                       foreach( $a as $iname => $val )
+                               array_push( $arr, array(
+                                       'il_from' => $this->mId,
+                                       'il_to'   => $iname ) );
+                       $dbw->insert( 'imagelinks', $arr, $fname, array( 'IGNORE' ) );
                }
-               if ( "" != $sql ) { wfQuery( $sql, DB_WRITE, $fname ); }
 
-               $this->fixBrokenLinks();
-
-               if( $wgDBtransactions ) {
-                       $sql = "COMMIT";
-                       wfQuery( $sql, DB_WRITE, $fname );
+               if( $wgUseCategoryMagic ) {
+                       $sql = "DELETE FROM $categorylinks WHERE cl_from='{$this->mId}'";
+                       $dbw->query( $sql, $fname );
+                       
+                       # Get addition list
+                       $add = $wgLinkCache->getCategoryLinks();
+                       
+                       # Do the insertion
+                       $sql = '';
+                       if ( 0 != count ( $add ) ) {
+                               $arr = array();
+                               foreach( $add as $cname => $sortkey ) {
+                                       # FIXME: Change all this to avoid unnecessary duplication
+                                       $nt = Title::makeTitle( NS_CATEGORY, $cname );
+                                        if( !$nt ) continue;
+                                        $nt->invalidateCache();
+                                       array_push( $arr, array(
+                                               'cl_from'    => $this->mId,
+                                               'cl_to'      => $cname,
+                                               'cl_sortkey' => $sortkey ) );
+                               }
+                               $dbw->insert( 'categorylinks', $arr, $fname, array( 'IGNORE' ) );
+                       }
                }
+               $this->fixBrokenLinks();
                wfProfileOut( $fname );
        }
-       
+
+       /**
+        * Update any brokenlinks *to* this page
+        * Call for a newly created page, or just to make sure state is consistent
+        */
        function fixBrokenLinks() {
-               /* Update any brokenlinks *to* this page */
-               /* Call for a newly created page, or just to make sure state is consistent */
+               $fname = 'LinksUpdate::fixBrokenLinks';
+
+               $dbw =& wfGetDB( DB_MASTER );
+               $page = $dbw->tableName( 'page' );
+               $links = $dbw->tableName( 'links' );
                
-               $sql = "SELECT bl_from FROM brokenlinks WHERE bl_to='{$this->mTitleEnc}'";
-               $res = wfQuery( $sql, DB_READ, $fname );
-               if ( 0 == wfNumRows( $res ) ) { return; }
+               $res = $dbw->select( 'brokenlinks', array( 'bl_from' ), array( 'bl_to' => $this->mTitle ), 
+                 $fname, 'FOR UPDATE' );
+               if ( 0 == $dbw->numRows( $res ) ) { return; }
 
-               $sql = "INSERT INTO links (l_from,l_to) VALUES ";
-               $now = wfTimestampNow();
-               $sql2 = "UPDATE cur SET cur_touched='{$now}' WHERE cur_id IN (";
+               $arr=array();
+               $now = $dbw->timestamp();
+               $sql2 = "UPDATE $page SET page_touched='{$now}' WHERE page_id IN (";
                $first = true;
-               while ( $row = wfFetchObject( $res ) ) {
-                       if ( ! $first ) { $sql .= ","; $sql2 .= ","; }
+               while ( $row = $dbw->fetchObject( $res ) ) {
+                       if ( ! $first ) { $sql2 .= ","; }
                        $first = false;
-                       $nl = wfStrencode( Title::nameOf( $row->bl_from ) );
-
-                       $sql .= "('{$nl}',{$this->mId})";
+                       array_push( $arr, array(
+                               'l_from' => $row->bl_from,
+                               'l_to'   => $this->mId ) );
                        $sql2 .= $row->bl_from;
                }
-               $sql2 .= ")";
-               wfQuery( $sql, DB_WRITE, $fname );
-               wfQuery( $sql2, DB_WRITE, $fname );
-
-               $sql = "DELETE FROM brokenlinks WHERE bl_to='{$this->mTitleEnc}'";
-               wfQuery( $sql, DB_WRITE, $fname );
+               $sql2 .= ')';
+               
+               # Ignore errors. If a link existed in both the brokenlinks table and the links 
+               # table, that's an error which can be fixed at this stage by simply ignoring collisions
+               $dbw->insert( 'links', $arr, $fname, array( 'IGNORE' ) );
+               $dbw->query( $sql2, $fname );
+               $dbw->delete( 'brokenlinks', array( 'bl_to' => $this->mTitle ), $fname );
        }
-       
 }
-
-?>
+?>
\ No newline at end of file