X-Git-Url: https://git.heureux-cyclage.org/?a=blobdiff_plain;f=includes%2FLinkCache.php;h=79727615e92538340321d8764374b85750ed6fcb;hb=3857496fe2bda374229d0652a9cf51817cab165b;hp=35e486bf92316ca8ac8143f0c3e25cf006a81072;hpb=e0ce5a322e41cf39f110fd4fa3c2c5c849964252;p=lhc%2Fweb%2Fwiklou.git diff --git a/includes/LinkCache.php b/includes/LinkCache.php index 35e486bf92..79727615e9 100644 --- a/includes/LinkCache.php +++ b/includes/LinkCache.php @@ -1,60 +1,45 @@ mActive = true; - $this->mPreFilled = false; + + function __construct() { $this->mForUpdate = false; $this->mPageLinks = array(); $this->mGoodLinks = array(); + $this->mGoodLinkFields = array(); $this->mBadLinks = array(); - $this->mImageLinks = array(); - $this->mCategoryLinks = array(); - $this->mOldGoodLinks = array(); - $this->mOldBadLinks = array(); } /** * General accessor to get/set whether SELECT FOR UPDATE should be used */ - function forUpdate( $update = NULL ) { + public function forUpdate( $update = NULL ) { return wfSetVar( $this->mForUpdate, $update ); } - - function getGoodLinkID( $title ) { + + public function getGoodLinkID( $title ) { if ( array_key_exists( $title, $this->mGoodLinks ) ) { return $this->mGoodLinks[$title]; } else { @@ -62,477 +47,143 @@ class LinkCache { } } - function isBadLink( $title ) { - return array_key_exists( $title, $this->mBadLinks ); + /** + * 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 addGoodLinkObj( $id, $title ) { - if ( $this->mActive ) { - $dbkey = $title->getPrefixedDbKey(); - $this->mGoodLinks[$dbkey] = $id; - $this->mPageLinks[$dbkey] = $title; - } + public function isBadLink( $title ) { + return array_key_exists( $title, $this->mBadLinks ); } - function addBadLinkObj( $title ) { + /** + * 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(); - if ( $this->mActive && ( ! $this->isBadLink( $dbkey ) ) ) { + $this->mGoodLinks[$dbkey] = $id; + $this->mGoodLinkFields[$dbkey] = array( 'length' => $len, 'redirect' => $redir ); + $this->mPageLinks[$dbkey] = $title; + } + + public function addBadLinkObj( $title ) { + $dbkey = $title->getPrefixedDbKey(); + if ( ! $this->isBadLink( $dbkey ) ) { $this->mBadLinks[$dbkey] = 1; $this->mPageLinks[$dbkey] = $title; } } - function addImageLink( $title ) { - if ( $this->mActive ) { $this->mImageLinks[$title] = 1; } - } - - 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 ); - } - - 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 getPageLinks() { return $this->mPageLinks; } - 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 getPageLinks() { return $this->mPageLinks; } + 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'; + 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 ) ) { - if ( $this->mForUpdate ) { - $db =& wfGetDB( DB_MASTER ); + + # Some fields heavily used for linking... + if ( $this->mForUpdate ) { + $db = wfGetDB( DB_MASTER ); + if ( !( $wgAntiLockFlags & ALF_NO_LINK_LOCK ) ) { $options = array( 'FOR UPDATE' ); } else { - $db =& wfGetDB( DB_SLAVE ); $options = array(); } - - $id = $db->selectField( 'page', 'page_id', array( 'page_namespace' => $ns, 'page_title' => $t ), $fname, $options ); - if ( !$id ) { - $id = 0; - } - if( $wgLinkCacheMemcached ) - $wgMemc->add( $key, $id, 3600*24 ); - } - - if( 0 == $id ) { - $this->addBadLinkObj( $nt ); } else { - $this->addGoodLinkObj( $id, $nt ); + $db = wfGetDB( DB_SLAVE ); + $options = array(); } - wfProfileOut( $fname ); - return $id; - } - /** - * Bulk-check the pagelinks and page arrays for existence info. - * @param Title $fromtitle - */ - function preFill( &$fromtitle ) { - global $wgEnablePersistentLC; + $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; - $fname = 'LinkCache::preFill'; - wfProfileIn( $fname ); - - $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; - } - } - if ( $this->mForUpdate ) { - $db =& wfGetDB( DB_MASTER ); - $options = 'FOR UPDATE'; + if( 0 == $id ) { + $this->addBadLinkObj( $nt ); } else { - $db =& wfGetDB( DB_SLAVE ); - $options = ''; - } - - $page = $db->tableName( 'page' ); - $pagelinks = $db->tableName( 'pagelinks' ); - - $sql = "SELECT page_id,pl_namespace,pl_title - FROM $pagelinks - LEFT JOIN $page - ON pl_namespace=page_namespace AND pl_title=page_title - WHERE pl_from=$id $options"; - $res = $db->query( $sql, $fname ); - while( $s = $db->fetchObject( $res ) ) { - $title = Title::makeTitle( $s->pl_namespace, $s->pl_title ); - if( $s->page_id ) { - $this->addGoodLinkObj( $s->page_id, $title ); - } else { - $this->addBadLinkObj( $title ); - } - } - $this->mOldPageLinks = $this->mPageLinks; - $this->mOldBadLinks = $this->mBadLinks; - $this->mOldGoodLinks = $this->mGoodLinks; - $this->mPreFilled = true; - - if ( $wgEnablePersistentLC ) { - $this->saveToLinkscc( $id ); + $this->addGoodLinkObj( $id, $nt, $len, $redirect ); } wfProfileOut( $fname ); - } - - 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 ); - } - - function getPageAdditions() { - $set = array_diff( array_keys( $this->mPageLinks ), array_keys( $this->mOldPageLinks ) ); - $out = array(); - foreach( $set as $key ) { - $out[$key] = $this->mPageLinks[$key]; - } - return $out; - } - - function getPageDeletions() { - $set = array_diff( array_keys( $this->mOldPageLinks ), array_keys( $this->mPageLinks ) ); - $out = array(); - foreach( $set as $key ) { - $out[$key] = $this->mOldPageLinks[$key]; - } - return $out; - } - - /** - * Parameters: - * @param $which is one of the LINKCACHE_xxx constants - * @param $del,$add are the incremental update arrays which will be filled. - * - * @return 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; - case LINKCACHE_PAGE: - $old =& $this->mOldPageLinks; - $cur =& $this->mPageLinks; - $del = $this->getPageDeletions(); - $add = $this->getPageAdditions(); - break; - default: # LINKCACHE_IMAGE - return false; - } - - return true; + return $id; } /** - * Clears cache but leaves old preFill copies alone + * Clears cache */ - function clear() { + public function clear() { $this->mPageLinks = array(); $this->mGoodLinks = array(); + $this->mGoodLinkFields = array(); $this->mBadLinks = array(); - $this->mImageLinks = array(); - } - - /** - * @access private - */ - function fillFromLinkscc( $id ){ - $fname = 'LinkCache::fillFromLinkscc'; - - $id = IntVal( $id ); - if ( $this->mForUpdate ) { - $db =& wfGetDB( DB_MASTER ); - $options = 'FOR UPDATE'; - } else { - $db =& wfGetDB( DB_SLAVE ); - $options = ''; - } - $raw = $db->selectField( 'linkscc', 'lcc_cacheobj', array( 'lcc_pageid' => $id ), $fname, $options ); - if ( $raw === false ) { - return false; - } - - $cacheobj = false; - if( function_exists( 'gzuncompress' ) ) - $cacheobj = @gzuncompress( $raw ); - - if($cacheobj == FALSE){ - $cacheobj = $raw; - } - $cc = @unserialize( $cacheobj ); - if( isset( $cc->mClassVer ) and ($cc->mClassVer == $this->mClassVer ) ){ - $this->mOldPageLinks = $this->mPageLinks = $cc->mPageLinks; - $this->mOldGoodLinks = $this->mGoodLinks = $cc->mGoodLinks; - $this->mOldBadLinks = $this->mBadLinks = $cc->mBadLinks; - $this->mPreFilled = true; - return TRUE; - } else { - return FALSE; - } - - } - - /** - * @access private - */ - function saveToLinkscc( $pid ){ - global $wgCompressedPersistentLC; - if( $wgCompressedPersistentLC and function_exists( 'gzcompress' ) ) { - $ser = gzcompress( serialize( $this ), 3 ); - } else { - $ser = serialize( $this ); - } - $db =& wfGetDB( DB_MASTER ); - $db->replace( 'linkscc', array( 'lcc_pageid' ), array( 'lcc_pageid' => $pid, 'lcc_cacheobj' => $ser ) ); - } - - /** - * Delete linkscc rows which link to here - * @param $title The page linked to - * @static - */ - function linksccClearLinksTo( $title ){ - global $wgEnablePersistentLC; - if ( $wgEnablePersistentLC ) { - $fname = 'LinkCache::linksccClearLinksTo'; - $pid = intval( $pid ); - $dbw =& wfGetDB( DB_MASTER ); - # Delete linkscc rows which link to here - $dbw->deleteJoin( 'linkscc', 'pagelinks', 'lcc_pageid', 'pl_from', - array( - 'pl_namespace' => $title->getNamespace(), - 'pl-title' => $title->getDbKey() ), - $fname ); - # Delete linkscc row representing this page - $dbw->delete( 'linkscc', array( 'lcc_pageid' => $pid ), $fname); - } - - } - - /** - * @param $pid is a page id - * @static - */ - function linksccClearPage( $pid ){ - global $wgEnablePersistentLC; - if ( $wgEnablePersistentLC ) { - $pid = intval( $pid ); - $dbw =& wfGetDB( DB_MASTER ); - $dbw->delete( 'linkscc', array( 'lcc_pageid' => $pid ) ); - } } } - -/** - * Class representing a list of titles - * The execute() method checks them all for existence and adds them to a LinkCache object - + - * @package MediaWikki - * @subpackage Cache - */ -class LinkBatch { - /** - * 2-d array, first index namespace, second index dbkey, value arbitrary - */ - var $data = array(); - - function LinkBatch( $arr = array() ) { - foreach( $arr as $item ) { - $this->addObj( $item ); - } - } - - function addObj( $title ) { - $this->add( $title->getNamespace(), $title->getDBkey() ); - } - - function add( $ns, $dbkey ) { - if ( $ns < 0 ) { - return; - } - if ( !array_key_exists( $ns, $this->data ) ) { - $this->data[$ns] = array(); - } - - $this->data[$ns][$dbkey] = 1; - } - - function execute( &$cache ) { - $fname = 'LinkBatch::execute'; - $namespaces = array(); - - if ( !count( $this->data ) ) { - return; - } - - wfProfileIn( $fname ); - - // Construct query - // This is very similar to Parser::replaceLinkHolders - $dbr = wfGetDB( DB_SLAVE ); - $page = $dbr->tableName( 'page' ); - $sql = "SELECT page_id, page_namespace, page_title FROM $page WHERE " - . $this->constructSet( 'page', $dbr ); - - // Do query - $res = $dbr->query( $sql, $fname ); - - // Process results - // For each returned entry, add it to the list of good links, and remove it from $remaining - - $remaining = $this->data; - while ( $row = $dbr->fetchObject( $res ) ) { - $title = Title::makeTitle( $row->page_namespace, $row->page_title ); - $cache->addGoodLinkObj( $row->page_id, $title ); - unset( $remaining[$row->page_namespace][$row->page_title] ); - } - $dbr->freeResult( $res ); - - // The remaining links in $data are bad links, register them as such - foreach ( $remaining as $ns => $dbkeys ) { - foreach ( $dbkeys as $dbkey => $nothing ) { - $title = Title::makeTitle( $ns, $dbkey ); - $cache->addBadLinkObj( $title ); - } - } - - wfProfileOut( $fname ); - } - - /** - * Construct a WHERE clause which will match all the given titles. - * Give the appropriate table's field name prefix ('page', 'pl', etc). - * - * @param string $prefix - * @return string - * @access public - */ - function constructSet( $prefix, $db ) { - $first = true; - $sql = ''; - foreach ( $this->data as $ns => $dbkeys ) { - if ( !count( $dbkeys ) ) { - continue; - } - - if ( $first ) { - $first = false; - } else { - $sql .= ' OR '; - } - $sql .= "({$prefix}_namespace=$ns AND {$prefix}_title IN ("; - - $firstTitle = true; - foreach( $dbkeys as $dbkey => $nothing ) { - if ( $firstTitle ) { - $firstTitle = false; - } else { - $sql .= ','; - } - $sql .= $db->addQuotes( $dbkey ); - } - - $sql .= '))'; - } - return $sql; - } -} - -?>