Fix blocks for non-memcache
[lhc/web/wiklou.git] / includes / LinkCache.php
index 23f5182..0c50c7d 100644 (file)
@@ -1,12 +1,15 @@
-<?
-# 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 = 1; 
 
        /* private */ var $mGoodLinks, $mBadLinks, $mActive;
        /* private */ var $mImageLinks; 
@@ -39,7 +42,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 +55,7 @@ class LinkCache {
        function addBadLink( $title )
        {
                if ( $this->mActive && ( ! $this->isBadLink( $title ) ) ) {
-                       array_push( $this->mBadLinks, $title );
+                       $this->mBadLinks[$title] = 1;
                }
        }
 
@@ -68,22 +71,21 @@ class LinkCache {
 
        function clearBadLink( $title )
        {
-               $index = array_search( $title, $this->mBadLinks );
-               if ( isset( $index ) ) {
-                       unset( $this->mBadLinks[$index] );
-               }
+               unset( $this->mBadLinks[$title] );
                $this->clearLink( $title );
        }
        
-       function clearLink( $title ) {
-               global $wgMemc;
-               $wgMemc->delete( $this->getKey( $title ) );
+       function clearLink( $title ) 
+       {
+               global $wgMemc, $wgLinkCacheMemcached;
+               if( $wgLinkCacheMemcached )
+                       $wgMemc->delete( $this->getKey( $title ) );
        }
 
        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 addLink( $title )
@@ -98,12 +100,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 );
 
@@ -114,9 +116,11 @@ class LinkCache {
                        wfProfileOut( $fname );
                        return 0; 
                }
-
-               $id = $wgMemc->get( $key = $this->getKey( $title ) );
-               if( $id === FALSE ) {
+               
+               $id = NULL;
+               if( $wgLinkCacheMemcached )
+                       $id = $wgMemc->get( $key = $this->getKey( $title ) );
+               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" );
@@ -127,8 +131,10 @@ class LinkCache {
                                $s = wfFetchObject( $res );
                                $id = $s->cur_id;
                        }
-                       $wgMemc->add( $key, $id, time()+3600 );
+                       if( $wgLinkCacheMemcached )
+                               $wgMemc->add( $key, $id, 3600*24 );
                }
+               
                if ( 0 == $id ) { $this->addBadLink( $title ); }
                else { $this->addGoodLink( $id, $title ); }
                wfProfileOut( $fname );
@@ -137,29 +143,31 @@ class LinkCache {
 
        function preFill( &$fromtitle )
        {
+               global $wgEnablePersistentLC;
+
                $fname = "LinkCache::preFill";
                wfProfileIn( $fname );
                # Note -- $fromtitle is a Title *object*
-               $dbkeyfrom = wfStrencode( $fromtitle->getPrefixedDBKey() );
-
-
-               $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;
-                 wfProfileOut( $fname );
-                 return;
-               } 
 
+               $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 ) {
+                       if( $this->fillFromLinkscc( $id ) ){
+                               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,
@@ -167,10 +175,6 @@ class LinkCache {
                                );
                }
                
-               $this->suspend();
-               $id = $fromtitle->getArticleID();
-               $this->resume();
-               
                $sql = "SELECT bl_to
                        FROM brokenlinks
                        WHERE bl_from='{$id}'";
@@ -183,11 +187,9 @@ class LinkCache {
                $this->mOldGoodLinks = $this->mGoodLinks;
                $this->mPreFilled = true;
 
-               // put fetched link data into cache
-               $serCachegz = wfStrencode( gzcompress( serialize( $this ), 3) );
-               wfQuery("REPLACE INTO linkscc VALUES({$id}, '{$dbkeyfrom}', '{$serCachegz}')", 
-                       DB_WRITE);
-
+               if ( $wgEnablePersistentLC ) {
+                       $this->saveToLinkscc( $id );
+               }
                wfProfileOut( $fname );
        }
 
@@ -198,7 +200,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()
@@ -213,7 +217,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()
@@ -258,6 +262,64 @@ 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;
+               if( $wgCompressedPersistentLC and function_exists( "gzcompress" ) ) {
+                       $ser = wfStrencode( gzcompress( serialize( $this ), 3 ));
+               } else {
+                       $ser = wfStrencode( serialize( $this ) );
+               }
+               wfQuery("REPLACE INTO linkscc(lcc_pageid,lcc_cacheobj) " .
+                       "VALUES({$pid}, '{$ser}')", DB_WRITE);
+       }
+
+       # $pid is a page id
+       /* static */ function linksccClearLinksTo( $pid ){
+               $pid = intval( $pid );
+               wfQuery("DELETE linkscc FROM linkscc,links ".
+                       "WHERE lcc_pageid=links.l_from AND 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 ){
+               $title = wfStrencode( $title );
+               wfQuery("DELETE linkscc FROM linkscc,brokenlinks ".
+                       "WHERE lcc_pageid=bl_from AND bl_to='{$title}'", DB_WRITE);
+       }
+
+       # $pid is a page id
+       /* static */ function linksccClearPage( $pid ){
+               $pid = intval( $pid );
+               wfQuery("DELETE FROM linkscc WHERE lcc_pageid='{$pid}'", DB_WRITE);
+       }
 }
 ?>