X-Git-Url: https://git.heureux-cyclage.org/?a=blobdiff_plain;f=includes%2FLinkCache.php;h=5666193759dcf9baa5e142d302ed43ee98c9d84a;hb=e2f1a14d9e27ff6da71be14619e878a78e9385e1;hp=4f7b03a227facc9d55c4966989461f059257f69e;hpb=797e7d34ec7aef5110a898effcc6d1473c28284f;p=lhc%2Fweb%2Fwiklou.git diff --git a/includes/LinkCache.php b/includes/LinkCache.php index 4f7b03a227..5666193759 100644 --- a/includes/LinkCache.php +++ b/includes/LinkCache.php @@ -1,39 +1,43 @@ mActive = true; - $this->mPreFilled = false; + + function __construct() { + $this->mForUpdate = false; $this->mGoodLinks = array(); + $this->mGoodLinkFields = array(); $this->mBadLinks = array(); - $this->mImageLinks = array(); - $this->mCategoryLinks = array(); - $this->mOldGoodLinks = array(); - $this->mOldBadLinks = array(); } - function getGoodLinkID( $title ) - { + /** + * General accessor to get/set whether SELECT FOR UPDATE should be used + */ + public function forUpdate( $update = NULL ) { + return wfSetVar( $this->mForUpdate, $update ); + } + + public function getGoodLinkID( $title ) { if ( array_key_exists( $title, $this->mGoodLinks ) ) { return $this->mGoodLinks[$title]; } else { @@ -41,322 +45,139 @@ class LinkCache { } } - function isBadLink( $title ) - { - return array_key_exists( $title, $this->mBadLinks ); - } - - function addGoodLink( $id, $title ) - { - if ( $this->mActive ) { - $this->mGoodLinks[$title] = $id; + /** + * Get a field of a title object from cache. + * If this link is not good, it will return NULL. + * @param Title $title + * @param string $field ('length','redirect') + * @return mixed + */ + public function getGoodLinkFieldObj( $title, $field ) { + $dbkey = $title->getPrefixedDbKey(); + if ( array_key_exists( $dbkey, $this->mGoodLinkFields ) ) { + return $this->mGoodLinkFields[$dbkey][$field]; + } else { + return NULL; } } - function addBadLink( $title ) - { - if ( $this->mActive && ( ! $this->isBadLink( $title ) ) ) { - $this->mBadLinks[$title] = 1; - } + public function isBadLink( $title ) { + return array_key_exists( $title, $this->mBadLinks ); } - function addImageLink( $title ) - { - if ( $this->mActive ) { $this->mImageLinks[$title] = 1; } + /** + * Add a link for the title to the link cache + * @param int $id + * @param Title $title + * @param int $len + * @param int $redir + */ + public function addGoodLinkObj( $id, $title, $len = -1, $redir = NULL ) { + $dbkey = $title->getPrefixedDbKey(); + $this->mGoodLinks[$dbkey] = $id; + $this->mGoodLinkFields[$dbkey] = array( 'length' => $len, 'redirect' => $redir ); } - function addImageLinkObj( $nt ) - { - 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 ); + public function addBadLinkObj( $title ) { + $dbkey = $title->getPrefixedDbKey(); + if ( ! $this->isBadLink( $dbkey ) ) { + $this->mBadLinks[$dbkey] = 1; + } } - function clearBadLink( $title ) - { + public function clearBadLink( $title ) { unset( $this->mBadLinks[$title] ); - $this->clearLink( $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 array_keys( $this->mBadLinks ); } - function getImageLinks() { return $this->mImageLinks; } - function getCategoryLinks() { return $this->mCategoryLinks; } + /* obsolete, for old $wgLinkCacheMemcached stuff */ + public function clearLink( $title ) {} + + public function getGoodLinks() { return $this->mGoodLinks; } + public function getBadLinks() { return array_keys( $this->mBadLinks ); } - function addLink( $title ) - { + /** + * Add a title to the link cache, return the page_id or zero if non-existent + * @param $title String: title to add + * @param $len int, page size + * @param $redir bool, is redirect? + * @return integer + */ + public function addLink( $title, $len = -1, $redir = NULL ) { $nt = Title::newFromDBkey( $title ); if( $nt ) { - return $this->addLinkObj( $nt ); + return $this->addLinkObj( $nt, $len, $redir ); } else { return 0; } } - - function addLinkObj( &$nt ) - { - global $wgMemc, $wgLinkCacheMemcached; + + /** + * Add a title to the link cache, return the page_id or zero if non-existent + * @param $nt Title to add. + * @param $len int, page size + * @param $redir bool, is redirect? + * @return integer + */ + public function addLinkObj( &$nt, $len = -1, $redirect = NULL ) { + global $wgAntiLockFlags, $wgProfiler; + $title = $nt->getPrefixedDBkey(); - if ( $this->isBadLink( $title ) ) { return 0; } + if ( $this->isBadLink( $title ) ) { return 0; } $id = $this->getGoodLinkID( $title ); if ( 0 != $id ) { return $id; } - $fname = "LinkCache::addLinkObj"; + $fname = 'LinkCache::addLinkObj'; + if ( isset( $wgProfiler ) ) { + $fname .= ' (' . $wgProfiler->getCurrentSection() . ')'; + } + wfProfileIn( $fname ); $ns = $nt->getNamespace(); $t = $nt->getDBkey(); - if ( "" == $title ) { + if ( '' == $title ) { wfProfileOut( $fname ); - return 0; + return 0; } - - $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" ); - if ( 0 == wfNumRows( $res ) ) { - $id = 0; + # Some fields heavily used for linking... + if ( $this->mForUpdate ) { + $db = wfGetDB( DB_MASTER ); + if ( !( $wgAntiLockFlags & ALF_NO_LINK_LOCK ) ) { + $options = array( 'FOR UPDATE' ); } else { - $s = wfFetchObject( $res ); - $id = $s->cur_id; - } - if( $wgLinkCacheMemcached ) - $wgMemc->add( $key, $id, 3600*24 ); - } - - if ( 0 == $id ) { $this->addBadLink( $title ); } - else { $this->addGoodLink( $id, $title ); } - wfProfileOut( $fname ); - return $id; - } - - function preFill( &$fromtitle ) - { - global $wgEnablePersistentLC; - - $fname = "LinkCache::preFill"; - wfProfileIn( $fname ); - # Note -- $fromtitle is a Title *object* - - $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 ) ){ - wfProfileOut( $fname ); - return; + $options = array(); } + } else { + $db = wfGetDB( DB_SLAVE ); + $options = array(); } - $sql = "SELECT cur_id,cur_namespace,cur_title - FROM cur,links - WHERE cur_id=l_to AND l_from=$id"; - $res = wfQuery( $sql, DB_READ, $fname ); - while( $s = wfFetchObject( $res ) ) { - $this->addGoodLink( $s->cur_id, - Title::makeName( $s->cur_namespace, $s->cur_title ) - ); - } - - $sql = "SELECT bl_to - FROM brokenlinks - WHERE bl_from='{$id}'"; - $res = wfQuery( $sql, DB_READ, "LinkCache::preFill" ); - while( $s = wfFetchObject( $res ) ) { - $this->addBadLink( $s->bl_to ); - } - - $this->mOldBadLinks = $this->mBadLinks; - $this->mOldGoodLinks = $this->mGoodLinks; - $this->mPreFilled = true; + $s = $db->selectRow( 'page', + array( 'page_id', 'page_len', 'page_is_redirect' ), + array( 'page_namespace' => $ns, 'page_title' => $t ), + $fname, $options ); + # Set fields... + $id = $s ? $s->page_id : 0; + $len = $s ? $s->page_len : -1; + $redirect = $s ? $s->page_is_redirect : 0; - if ( $wgEnablePersistentLC ) { - $this->saveToLinkscc( $id ); + if( 0 == $id ) { + $this->addBadLinkObj( $nt ); + } else { + $this->addGoodLinkObj( $id, $nt, $len, $redirect ); } wfProfileOut( $fname ); + return $id; } - function getGoodAdditions() - { - return array_diff( $this->mGoodLinks, $this->mOldGoodLinks ); - } - - function getBadAdditions() - { - #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() - { - return array_diff_assoc( $this->mImageLinks, $this->mOldImageLinks ); - } - - function getGoodDeletions() - { - return array_diff( $this->mOldGoodLinks, $this->mGoodLinks ); - } - - function getBadDeletions() - { - return array_values( array_diff( array_keys( $this->mOldBadLinks ), array_keys( $this->mBadLinks ) )); - } - - function getImageDeletions() - { - return array_diff_assoc( $this->mOldImageLinks, $this->mImageLinks ); - } - - # Parameters: $which is one of the LINKCACHE_xxx constants, $del and $add are - # the incremental update arrays which will be filled. Returns whether or not it's - # worth doing the incremental version. For example, if [[List of mathematical topics]] - # was blanked, it would take a long, long time to do incrementally. - function incrementalSetup( $which, &$del, &$add ) - { - if ( ! $this->mPreFilled ) { - return false; - } - - switch ( $which ) { - case LINKCACHE_GOOD: - $old =& $this->mOldGoodLinks; - $cur =& $this->mGoodLinks; - $del = $this->getGoodDeletions(); - $add = $this->getGoodAdditions(); - break; - case LINKCACHE_BAD: - $old =& $this->mOldBadLinks; - $cur =& $this->mBadLinks; - $del = $this->getBadDeletions(); - $add = $this->getBadAdditions(); - break; - default: # LINKCACHE_IMAGE - return false; - } - - return true; - } - - # Clears cache but leaves old preFill copies alone - function clear() - { + /** + * Clears cache + */ + public function clear() { $this->mGoodLinks = array(); + $this->mGoodLinkFields = array(); $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 ($wgIsMySQL) { - 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); - } } } -?>