don't use OR on empty variables
[lhc/web/wiklou.git] / includes / LinkCache.php
index 9b0f077..512bc7d 100644 (file)
@@ -1,15 +1,18 @@
-<?
-# Cache for article titles and ids linked from one source
+<?php
+# Cache for article titles (prefixed DB keys) and ids linked from one source
 
 # These are used in incrementalSetup()
 define ('LINKCACHE_GOOD', 0);
 define ('LINKCACHE_BAD', 1);
 define ('LINKCACHE_IMAGE', 2);
 
-class LinkCache {
+class LinkCache {      
+       // Increment $mClassVer whenever old serialized versions of this class
+       // becomes incompatible with the new version.
+       /* private */ var $mClassVer = 2;
 
        /* private */ var $mGoodLinks, $mBadLinks, $mActive;
-       /* private */ var $mImageLinks
+       /* private */ var $mImageLinks, $mCategoryLinks;
        /* private */ var $mPreFilled, $mOldGoodLinks, $mOldBadLinks;
        
        /* private */ function getKey( $title ) {
@@ -24,6 +27,7 @@ class LinkCache {
                $this->mGoodLinks = array();
                $this->mBadLinks = array();
                $this->mImageLinks = array();
+               $this->mCategoryLinks = array();
                $this->mOldGoodLinks = array();
                $this->mOldBadLinks = array();
        }
@@ -39,7 +43,7 @@ class LinkCache {
 
        function isBadLink( $title )
        {
-               return in_array( $title, $this->mBadLinks );
+               return array_key_exists( $title, $this->mBadLinks ); 
        }
 
        function addGoodLink( $id, $title )
@@ -52,7 +56,7 @@ class LinkCache {
        function addBadLink( $title )
        {
                if ( $this->mActive && ( ! $this->isBadLink( $title ) ) ) {
-                       array_push( $this->mBadLinks, $title );
+                       $this->mBadLinks[$title] = 1;
                }
        }
 
@@ -65,13 +69,18 @@ class LinkCache {
        {
                if ( $this->mActive ) { $this->mImageLinks[$nt->getDBkey()] = 1; }
        }
+       
+       function addCategoryLink( $title, $sortkey ) {
+               if ( $this->mActive ) { $this->mCategoryLinks[$title] = $sortkey; }
+       }
+       
+       function addCategoryLinkObj( &$nt, $sortkey ) {
+               $this->addCategoryLink( $nt->getDBkey(), $sortkey );
+       }
 
        function clearBadLink( $title )
        {
-               $index = array_search( $title, $this->mBadLinks );
-               if ( isset( $index ) ) {
-                       unset( $this->mBadLinks[$index] );
-               }
+               unset( $this->mBadLinks[$title] );
                $this->clearLink( $title );
        }
        
@@ -85,8 +94,9 @@ class LinkCache {
        function suspend() { $this->mActive = false; }
        function resume() { $this->mActive = true; }
        function getGoodLinks() { return $this->mGoodLinks; }
-       function getBadLinks() { return $this->mBadLinks; }
+       function getBadLinks() { return array_keys( $this->mBadLinks ); }
        function getImageLinks() { return $this->mImageLinks; }
+       function getCategoryLinks() { return $this->mCategoryLinks; }
 
        function addLink( $title )
        {
@@ -100,12 +110,12 @@ class LinkCache {
        
        function addLinkObj( &$nt )
        {
+               global $wgMemc, $wgLinkCacheMemcached;
                $title = $nt->getPrefixedDBkey();
-               if ( $this->isBadLink( $title ) ) { return 0; }
+               if ( $this->isBadLink( $title ) ) { return 0; }         
                $id = $this->getGoodLinkID( $title );
                if ( 0 != $id ) { return $id; }
 
-               global $wgMemc;
                $fname = "LinkCache::addLinkObj";
                wfProfileIn( $fname );
 
@@ -117,11 +127,10 @@ class LinkCache {
                        return 0; 
                }
                
-               $id = FALSE;
+               $id = NULL;
                if( $wgLinkCacheMemcached )
                        $id = $wgMemc->get( $key = $this->getKey( $title ) );
-
-               if( $id === FALSE ) {
+               if( ! is_integer( $id ) ) {
                        $sql = "SELECT cur_id FROM cur WHERE cur_namespace=" .
                          "{$ns} AND cur_title='" . wfStrencode( $t ) . "'";
                        $res = wfQuery( $sql, DB_READ, "LinkCache::addLink" );
@@ -133,8 +142,9 @@ class LinkCache {
                                $id = $s->cur_id;
                        }
                        if( $wgLinkCacheMemcached )
-                               $wgMemc->add( $key, $id, time()+3600 );
+                               $wgMemc->add( $key, $id, 3600*24 );
                }
+               
                if ( 0 == $id ) { $this->addBadLink( $title ); }
                else { $this->addGoodLink( $id, $title ); }
                wfProfileOut( $fname );
@@ -148,27 +158,27 @@ class LinkCache {
                $fname = "LinkCache::preFill";
                wfProfileIn( $fname );
                # Note -- $fromtitle is a Title *object*
-               $dbkeyfrom = wfStrencode( $fromtitle->getPrefixedDBKey() );
 
+               $this->suspend();
+               $id = $fromtitle->getArticleID();
+               $this->resume();
+               
+               if( $id == 0 ) {
+                       wfDebug( "$fname - got id 0 for title '" . $fromtitle->getPrefixedDBkey() . "'\n" );
+                       wfProfileOut( $fname );
+                       return;
+               }
+               
                if ( $wgEnablePersistentLC ) {
-                       $res = wfQuery("SELECT lcc_cacheobj FROM linkscc WHERE lcc_title = '{$dbkeyfrom}'", 
-                               DB_READ);
-                       $row = wfFetchObject( $res );
-                       if( $row != FALSE){
-                               $cacheobj = gzuncompress( $row->lcc_cacheobj );
-                               $cc = unserialize( $cacheobj );
-                               $this->mGoodLinks = $cc->mGoodLinks;
-                               $this->mBadLinks = $cc->mBadLinks;
-                               $this->mPreFilled = true;
+                       if( $this->fillFromLinkscc( $id ) ){
                                wfProfileOut( $fname );
                                return;
-                       } 
+                       }
                }
 
-
                $sql = "SELECT cur_id,cur_namespace,cur_title
                        FROM cur,links
-                       WHERE cur_id=l_to AND l_from='{$dbkeyfrom}'";
+                       WHERE cur_id=l_to AND l_from=$id";
                $res = wfQuery( $sql, DB_READ, $fname );
                while( $s = wfFetchObject( $res ) ) {
                        $this->addGoodLink( $s->cur_id,
@@ -176,10 +186,6 @@ class LinkCache {
                                );
                }
                
-               $this->suspend();
-               $id = $fromtitle->getArticleID();
-               $this->resume();
-               
                $sql = "SELECT bl_to
                        FROM brokenlinks
                        WHERE bl_from='{$id}'";
@@ -193,12 +199,8 @@ class LinkCache {
                $this->mPreFilled = true;
 
                if ( $wgEnablePersistentLC ) {
-                       // put fetched link data into cache
-                       $serCachegz = wfStrencode( gzcompress( serialize( $this ), 3) );
-                       wfQuery("REPLACE INTO linkscc VALUES({$id}, '{$dbkeyfrom}', '{$serCachegz}')", 
-                               DB_WRITE);
+                       $this->saveToLinkscc( $id );
                }
-
                wfProfileOut( $fname );
        }
 
@@ -209,7 +211,9 @@ class LinkCache {
 
        function getBadAdditions() 
        {
-               return array_values( array_diff( $this->mBadLinks, $this->mOldBadLinks ) );
+               #wfDebug( "mOldBadLinks: " . implode( ', ', array_keys( $this->mOldBadLinks ) ) . "\n" );
+               #wfDebug( "mBadLinks: " . implode( ', ', array_keys( $this->mBadLinks ) ) . "\n" );
+               return array_values( array_diff( array_keys( $this->mBadLinks ), array_keys( $this->mOldBadLinks ) ) );
        }
 
        function getImageAdditions()
@@ -224,7 +228,7 @@ class LinkCache {
 
        function getBadDeletions()
        {
-               return array_values( array_diff( $this->mOldBadLinks, $this->mBadLinks ) );
+               return array_values( array_diff( array_keys( $this->mOldBadLinks ), array_keys( $this->mBadLinks ) ));
        }
 
        function getImageDeletions()
@@ -269,6 +273,90 @@ class LinkCache {
                $this->mBadLinks = array();
                $this->mImageLinks = array();
        }
-       
+
+       /* private */ function fillFromLinkscc( $id ){ 
+               $id = IntVal( $id );
+               $res = wfQuery("SELECT lcc_cacheobj FROM linkscc WHERE lcc_pageid = $id", 
+                       DB_READ);
+               $row = wfFetchObject( $res );
+               if( $row == FALSE)
+                       return false;   
+
+               $cacheobj = false;
+               if( function_exists( "gzuncompress" ) )
+                       $cacheobj = @gzuncompress( $row->lcc_cacheobj );
+
+               if($cacheobj == FALSE){
+                       $cacheobj = $row->lcc_cacheobj;
+               }
+               $cc = @unserialize( $cacheobj );
+               if( isset( $cc->mClassVer ) and ($cc->mClassVer == $this->mClassVer ) ){
+                       $this->mOldGoodLinks = $this->mGoodLinks = $cc->mGoodLinks;
+                       $this->mOldBadLinks = $this->mBadLinks = $cc->mBadLinks;
+                       $this->mPreFilled = true;
+                       return TRUE;
+               } else {
+                       return FALSE;
+               }
+
+       }
+
+       /* private */ function saveToLinkscc( $pid ){
+               global $wgCompressedPersistentLC, $wgIsMySQL;
+               if( $wgCompressedPersistentLC and function_exists( "gzcompress" ) ) {
+                       $ser = wfStrencode( gzcompress( serialize( $this ), 3 ));
+               } else {
+                       $ser = wfStrencode( serialize( $this ) );
+               }
+               if ($isMySQL) {
+                       wfQuery("REPLACE INTO linkscc(lcc_pageid,lcc_cacheobj) " .
+                               "VALUES({$pid}, '{$ser}')", DB_WRITE);
+               } else {
+                       wfQuery("DELETE FROM linkscc WHERE lcc_pageid={$pid}",DB_WRITE);
+                       wfQuery("INSERT INTO linkscc(lcc_pageid,lcc_cacheobj) " .
+                               "VALUES({$pid}, '{$ser}')", DB_WRITE);
+               }
+       }
+
+       # $pid is a page id
+       /* static */ function linksccClearLinksTo( $pid ){
+               global $wgEnablePersistentLC, $wgIsMySQL;
+               if ( $wgEnablePersistentLC ) {
+                       $pid = intval( $pid );
+                       if ($wgIsMySQL) {
+                               wfQuery("DELETE linkscc FROM linkscc,links ".
+                                       "WHERE lcc_pageid=links.l_from AND l_to={$pid}", DB_WRITE);
+                       } else {
+                               wfQuery("DELETE FROM linkscc WHERE lcc_pageid IN ".
+                                       "(SELECT l_from FROM links WHERE l_to={$pid}", DB_WRITE);
+                       }
+                       wfQuery("DELETE FROM linkscc WHERE lcc_pageid='{$pid}')", DB_WRITE);
+               }
+       }
+
+       # $title is a prefixed db title, for example like Title->getPrefixedDBkey() returns.
+       /* static */ function linksccClearBrokenLinksTo( $title ){
+               global $wgEnablePersistentLC,$wgIsMySQL;
+               if ( $wgEnablePersistentLC ) {
+                       $title = wfStrencode( $title );
+                       if ($wgIsMySQL) {
+                               wfQuery("DELETE linkscc FROM linkscc,brokenlinks ".
+                                       "WHERE lcc_pageid=bl_from AND bl_to='{$title}'", DB_WRITE);
+                       } else {
+                               wfQuery("DELETE FROM linkscc WHERE lcc_pageid IN ".
+                                       "(SELECT bl_from FROM brokenlinks ".
+                                       "WHERE bl_to='{$title}')",DB_WRITE);
+                       }
+               }
+       }
+
+       # $pid is a page id
+       /* static */ function linksccClearPage( $pid ){
+               global $wgEnablePersistentLC;
+               if ( $wgEnablePersistentLC ) {
+                       $pid = intval( $pid );
+                       wfQuery("DELETE FROM linkscc WHERE lcc_pageid='{$pid}'", DB_WRITE);
+               }
+       }
 }
 ?>