X-Git-Url: https://git.heureux-cyclage.org/?a=blobdiff_plain;f=includes%2FMessageCache.php;h=33b7e4251c8aada795c08dee90d56f7d86459db9;hb=92ec8a2bec1c34474e5450eec34203c19f253908;hp=bbea94abe0b3debe62e95d8df5cf31db9ac4c098;hpb=225c34df76823dde0b1323449181ba662ce1697c;p=lhc%2Fweb%2Fwiklou.git diff --git a/includes/MessageCache.php b/includes/MessageCache.php index bbea94abe0..33b7e4251c 100644 --- a/includes/MessageCache.php +++ b/includes/MessageCache.php @@ -1,7 +1,7 @@ mUseCache = !is_null( $memCached ); - $this->mMemc = &$memCached; + function __construct( $memCached, $useDB, $expiry ) { + if ( !$memCached ) { + $memCached = wfGetCache( CACHE_NONE ); + } + + $this->mMemc = $memCached; $this->mDisable = !$useDB; $this->mExpiry = $expiry; - $this->mDisableTransform = false; - $this->mMemcKey = $memcPrefix.':messages'; - $this->mKeys = false; # initialised on demand - $this->mInitialised = true; - $this->mParser = null; - - # When we first get asked for a message, - # then we'll fill up the cache. If we - # can return a cache hit, this saves - # some extra milliseconds - $this->mDeferred = true; - - wfProfileOut( __METHOD__ ); } + + /** + * ParserOptions is lazy initialised. + */ function getParserOptions() { if ( !$this->mParserOptions ) { $this->mParserOptions = new ParserOptions; @@ -55,22 +84,25 @@ class MessageCache { } /** - * Try to load the cache from a local file + * Try to load the cache from a local file. + * Actual format of the file depends on the $wgLocalMessageCacheSerialized + * setting. + * + * @param $hash String: the hash of contents, to check validity. + * @param $code Mixed: Optional language code, see documenation of load(). + * @return false on failure. */ - function loadFromLocal( $hash ) { - global $wgLocalMessageCache, $wgLocalMessageCacheSerialized; + function loadFromLocal( $hash, $code ) { + global $wgCacheDirectory, $wgLocalMessageCacheSerialized; - if ( $wgLocalMessageCache === false ) { - return; - } - - $filename = "$wgLocalMessageCache/messages-" . wfWikiID(); + $filename = "$wgCacheDirectory/messages-" . wfWikiID() . "-$code"; + # Check file existence wfSuppressWarnings(); $file = fopen( $filename, 'r' ); wfRestoreWarnings(); if ( !$file ) { - return; + return false; // No cache file } if ( $wgLocalMessageCacheSerialized ) { @@ -82,37 +114,38 @@ class MessageCache { while ( !feof( $file ) ) { $serialized .= fread( $file, 100000 ); } - $this->setCache( unserialize( $serialized ) ); + fclose( $file ); + return $this->setCache( unserialize( $serialized ), $code ); + } else { + fclose( $file ); + return false; // Wrong hash } - fclose( $file ); } else { $localHash=substr(fread($file,40),8); fclose($file); if ($hash!=$localHash) { - return; + return false; // Wrong hash } - require("$wgLocalMessageCache/messages-" . wfWikiID()); - $this->setCache( $this->mCache); + # Require overwrites the member variable or just shadows it? + require( $filename ); + return $this->setCache( $this->mCache, $code ); } } /** - * Save the cache to a local file + * Save the cache to a local file. */ - function saveToLocal( $serialized, $hash ) { - global $wgLocalMessageCache, $wgLocalMessageCacheSerialized; - - if ( $wgLocalMessageCache === false ) { - return; - } + function saveToLocal( $serialized, $hash, $code ) { + global $wgCacheDirectory; - $filename = "$wgLocalMessageCache/messages-" . wfWikiID(); - $oldUmask = umask( 0 ); - wfMkdirParents( $wgLocalMessageCache, 0777 ); - umask( $oldUmask ); + $filename = "$wgCacheDirectory/messages-" . wfWikiID() . "-$code"; + wfMkdirParents( $wgCacheDirectory ); // might fail + wfSuppressWarnings(); $file = fopen( $filename, 'w' ); + wfRestoreWarnings(); + if ( !$file ) { wfDebug( "Unable to open local cache file for writing\n" ); return; @@ -123,32 +156,33 @@ class MessageCache { @chmod( $filename, 0666 ); } - function loadFromScript( $hash ) { - wfDeprecated( __METHOD__ ); - $this->loadFromLocal( $hash ); - } + function saveToScript( $array, $hash, $code ) { + global $wgCacheDirectory; + + $filename = "$wgCacheDirectory/messages-" . wfWikiID() . "-$code"; + $tempFilename = $filename . '.tmp'; + wfMkdirParents( $wgCacheDirectory ); // might fail + + wfSuppressWarnings(); + $file = fopen( $tempFilename, 'w'); + wfRestoreWarnings(); - function saveToScript($array, $hash) { - global $wgLocalMessageCache; - if ( $wgLocalMessageCache === false ) { + if ( !$file ) { + wfDebug( "Unable to open local cache file for writing\n" ); return; } - $filename = "$wgLocalMessageCache/messages-" . wfWikiID(); - $oldUmask = umask( 0 ); - wfMkdirParents( $wgLocalMessageCache, 0777 ); - umask( $oldUmask ); - $file = fopen( $filename.'.tmp', 'w'); fwrite($file,"mCache = array("); foreach ($array as $key => $message) { - fwrite($file, "'". $this->escapeForScript($key). - "' => '" . $this->escapeForScript($message). - "',\n"); + $key = $this->escapeForScript($key); + $messages = $this->escapeForScript($message); + fwrite($file, "'$key' => '$message',\n"); } + fwrite($file,");\n?>"); fclose($file); - rename($filename.'.tmp',$filename); + rename($tempFilename, $filename); } function escapeForScript($string) { @@ -160,222 +194,316 @@ class MessageCache { /** * Set the cache to $cache, if it is valid. Otherwise set the cache to false. */ - function setCache( $cache ) { + function setCache( $cache, $code ) { if ( isset( $cache['VERSION'] ) && $cache['VERSION'] == MSG_CACHE_VERSION ) { - $this->mCache = $cache; + $this->mCache[$code] = $cache; + return true; } else { - $this->mCache = false; + return false; } } /** - * Loads messages either from memcached or the database, if not disabled - * On error, quietly switches to a fallback mode - * Returns false for a reportable error, true otherwise + * Loads messages from caches or from database in this order: + * (1) local message cache (if $wgUseLocalMessageCache is enabled) + * (2) memcached + * (3) from the database. + * + * When succesfully loading from (2) or (3), all higher level caches are + * updated for the newest version. + * + * Nothing is loaded if member variable mDisable is true, either manually + * set by calling code or if message loading fails (is this possible?). + * + * Returns true if cache is already populated or it was succesfully populated, + * or false if populating empty cache fails. Also returns true if MessageCache + * is disabled. + * + * @param $code String: language to which load messages */ - function load() { - global $wgLocalMessageCache, $wgLocalMessageCacheSerialized; + function load( $code = false ) { + global $wgUseLocalMessageCache; + if( !is_string( $code ) ) { + # This isn't really nice, so at least make a note about it and try to + # fall back + wfDebug( __METHOD__ . " called without providing a language code\n" ); + $code = 'en'; + } + + # Don't do double loading... + if ( isset($this->mLoadedLanguages[$code]) ) return true; + + # 8 lines of code just to say (once) that message cache is disabled if ( $this->mDisable ) { static $shownDisabled = false; if ( !$shownDisabled ) { - wfDebug( "MessageCache::load(): disabled\n" ); + wfDebug( __METHOD__ . ": disabled\n" ); $shownDisabled = true; } return true; } - if ( !$this->mUseCache ) { - $this->mDeferred = false; - return true; - } - $fname = 'MessageCache::load'; - wfProfileIn( $fname ); - $success = true; + # Loading code starts + wfProfileIn( __METHOD__ ); + $success = false; # Keep track of success + $where = array(); # Debug info, delayed to avoid spamming debug log too much + $cacheKey = wfMemcKey( 'messages', $code ); # Key in memc for messages + - $this->mCache = false; + # (1) local cache + # Hash of the contents is stored in memcache, to detect if local cache goes + # out of date (due to update in other thread?) + if ( $wgUseLocalMessageCache ) { + wfProfileIn( __METHOD__ . '-fromlocal' ); - # Try local cache - if ( $wgLocalMessageCache !== false ) { - wfProfileIn( $fname.'-fromlocal' ); - $hash = $this->mMemc->get( "{$this->mMemcKey}-hash" ); + $hash = $this->mMemc->get( wfMemcKey( 'messages', $code, 'hash' ) ); if ( $hash ) { - $this->loadFromLocal( $hash ); - if ( $this->mCache ) { - wfDebug( "MessageCache::load(): got from local cache\n" ); - } + $success = $this->loadFromLocal( $hash, $code ); + if ( $success ) $where[] = 'got from local cache'; } - wfProfileOut( $fname.'-fromlocal' ); - } - - # Try memcached - if ( !$this->mCache ) { - wfProfileIn( $fname.'-fromcache' ); - $this->setCache( $this->mMemc->get( $this->mMemcKey ) ); - if ( $this->mCache ) { - wfDebug( "MessageCache::load(): got from global cache\n" ); - # Save to local cache - if ( $wgLocalMessageCache !== false ) { - $serialized = serialize( $this->mCache ); - if ( !$hash ) { - $hash = md5( $serialized ); - $this->mMemc->set( "{$this->mMemcKey}-hash", $hash, $this->mExpiry ); - } - if ($wgLocalMessageCacheSerialized) { - $this->saveToLocal( $serialized,$hash ); - } else { - $this->saveToScript( $this->mCache, $hash ); - } - } + wfProfileOut( __METHOD__ . '-fromlocal' ); + } + + # (2) memcache + # Fails if nothing in cache, or in the wrong version. + if ( !$success ) { + wfProfileIn( __METHOD__ . '-fromcache' ); + $cache = $this->mMemc->get( $cacheKey ); + $success = $this->setCache( $cache, $code ); + if ( $success ) { + $where[] = 'got from global cache'; + $this->saveToCaches( $cache, false, $code ); } - wfProfileOut( $fname.'-fromcache' ); + wfProfileOut( __METHOD__ . '-fromcache' ); } - # If there's nothing in memcached, load all the messages from the database - if ( !$this->mCache ) { - wfDebug( "MessageCache::load(): cache is empty\n" ); - $this->lock(); - # Other threads don't need to load the messages if another thread is doing it. - $success = $this->mMemc->add( $this->mMemcKey.'-status', "loading", MSG_LOAD_TIMEOUT ); - if ( $success ) { - wfProfileIn( $fname.'-load' ); - wfDebug( "MessageCache::load(): loading all messages from DB\n" ); - $this->loadFromDB(); - wfProfileOut( $fname.'-load' ); - - # Save in memcached - # Keep trying if it fails, this is kind of important - wfProfileIn( $fname.'-save' ); - for ($i=0; $i<20 && - !$this->mMemc->set( $this->mMemcKey, $this->mCache, $this->mExpiry ); - $i++ ) { - usleep(mt_rand(500000,1500000)); - } + # (3) + # Nothing in caches... so we need create one and store it in caches + if ( !$success ) { + $where[] = 'cache is empty'; + $where[] = 'loading from database'; - # Save to local cache - if ( $wgLocalMessageCache !== false ) { - $serialized = serialize( $this->mCache ); - $hash = md5( $serialized ); - $this->mMemc->set( "{$this->mMemcKey}-hash", $hash, $this->mExpiry ); - if ($wgLocalMessageCacheSerialized) { - $this->saveToLocal( $serialized,$hash ); - } else { - $this->saveToScript( $this->mCache, $hash ); - } - } + $this->lock($cacheKey); - wfProfileOut( $fname.'-save' ); - if ( $i == 20 ) { - $this->mMemc->set( $this->mMemcKey.'-status', 'error', 60*5 ); - wfDebug( "MemCached set error in MessageCache: restart memcached server!\n" ); + # Limit the concurrency of loadFromDB to a single process + # This prevents the site from going down when the cache expires + $statusKey = wfMemcKey( 'messages', $code, 'status' ); + $success = $this->mMemc->add( $statusKey, 'loading', MSG_LOAD_TIMEOUT ); + if ( $success ) { + $cache = $this->loadFromDB( $code ); + $success = $this->setCache( $cache, $code ); + } + if ( $success ) { + $success = $this->saveToCaches( $cache, true, $code ); + if ( $success ) { + $this->mMemc->delete( $statusKey ); } else { - $this->mMemc->delete( $this->mMemcKey.'-status' ); + $this->mMemc->set( $statusKey, 'error', 60*5 ); + wfDebug( "MemCached set error in MessageCache: restart memcached server!\n" ); } } - $this->unlock(); + $this->unlock($cacheKey); } - if ( !is_array( $this->mCache ) ) { - wfDebug( "MessageCache::load(): unable to load cache, disabled\n" ); + if ( !$success ) { + # Bad luck... this should not happen + $where[] = 'loading FAILED - cache is disabled'; + $info = implode( ', ', $where ); + wfDebug( __METHOD__ . ": Loading $code... $info\n" ); $this->mDisable = true; $this->mCache = false; + } else { + # All good, just record the success + $info = implode( ', ', $where ); + wfDebug( __METHOD__ . ": Loading $code... $info\n" ); + $this->mLoadedLanguages[$code] = true; } - wfProfileOut( $fname ); - $this->mDeferred = false; + wfProfileOut( __METHOD__ ); return $success; } /** - * Loads all or main part of cacheable messages from the database + * Loads cacheable messages from the database. Messages bigger than + * $wgMaxMsgCacheEntrySize are assigned a special value, and are loaded + * on-demand from the database later. + * + * @param $code \string Language code. + * @return \array Loaded messages for storing in caches. */ - function loadFromDB() { - global $wgMaxMsgCacheEntrySize; - + function loadFromDB( $code ) { wfProfileIn( __METHOD__ ); + global $wgMaxMsgCacheEntrySize, $wgLanguageCode, $wgAdaptiveMessageCache; $dbr = wfGetDB( DB_SLAVE ); - $this->mCache = array(); + $cache = array(); + + # Common conditions + $conds = array( + 'page_is_redirect' => 0, + 'page_namespace' => NS_MEDIAWIKI, + ); + + $mostused = array(); + if ( $wgAdaptiveMessageCache ) { + $mostused = $this->getMostUsedMessages(); + if ( $code !== $wgLanguageCode ) { + foreach ( $mostused as $key => $value ) $mostused[$key] = "$value/$code"; + } + } + + if ( count( $mostused ) ) { + $conds['page_title'] = $mostused; + } elseif ( $code !== $wgLanguageCode ) { + $conds[] = 'page_title' . $dbr->buildLike( $dbr->anyString(), "/$code" ); + } else { + # Effectively disallows use of '/' character in NS_MEDIAWIKI for uses + # other than language code. + $conds[] = 'page_title NOT' . $dbr->buildLike( $dbr->anyString(), '/', $dbr->anyString() ); + } + + # Conditions to fetch oversized pages to ignore them + $bigConds = $conds; + $bigConds[] = 'page_len > ' . intval( $wgMaxMsgCacheEntrySize ); # Load titles for all oversized pages in the MediaWiki namespace - $res = $dbr->select( 'page', 'page_title', - array( - 'page_len > ' . intval( $wgMaxMsgCacheEntrySize ), - 'page_is_redirect' => 0, - 'page_namespace' => NS_MEDIAWIKI, - ), - __METHOD__ ); - while ( $row = $dbr->fetchObject( $res ) ) { - $this->mCache[$row->page_title] = '!TOO BIG'; + $res = $dbr->select( 'page', 'page_title', $bigConds, __METHOD__ . "($code)-big" ); + foreach ( $res as $row ) { + $cache[$row->page_title] = '!TOO BIG'; } - $dbr->freeResult( $res ); - # Load text for the remaining pages + # Conditions to load the remaining pages with their contents + $smallConds = $conds; + $smallConds[] = 'page_latest=rev_id'; + $smallConds[] = 'rev_text_id=old_id'; + $smallConds[] = 'page_len <= ' . intval( $wgMaxMsgCacheEntrySize ); + $res = $dbr->select( array( 'page', 'revision', 'text' ), array( 'page_title', 'old_text', 'old_flags' ), - array( - 'page_is_redirect' => 0, - 'page_namespace' => NS_MEDIAWIKI, - 'page_latest=rev_id', - 'rev_text_id=old_id', - 'page_len <= ' . intval( $wgMaxMsgCacheEntrySize ) ), - __METHOD__ ); - - for ( $row = $dbr->fetchObject( $res ); $row; $row = $dbr->fetchObject( $res ) ) { - $this->mCache[$row->page_title] = ' ' . Revision::getRevisionText( $row ); - } - $this->mCache['VERSION'] = MSG_CACHE_VERSION; - $dbr->freeResult( $res ); + $smallConds, __METHOD__ . "($code)-small" ); + + foreach ( $res as $row ) { + $cache[$row->page_title] = ' ' . Revision::getRevisionText( $row ); + } + + foreach ( $mostused as $key ) { + if ( !isset( $cache[$key] ) ) { + $cache[$key] = '!NONEXISTENT'; + } + } + + $cache['VERSION'] = MSG_CACHE_VERSION; wfProfileOut( __METHOD__ ); + return $cache; } - function replace( $title, $text ) { - global $wgLocalMessageCache, $wgLocalMessageCacheSerialized, $parserMemc; + /** + * Updates cache as necessary when message page is changed + * + * @param $title String: name of the page changed. + * @param $text Mixed: new contents of the page. + */ + public function replace( $title, $text ) { global $wgMaxMsgCacheEntrySize; + wfProfileIn( __METHOD__ ); + + if ( $this->mDisable ) { + return; + } + + list( $msg, $code ) = $this->figureMessage( $title ); + + $cacheKey = wfMemcKey( 'messages', $code ); + $this->load( $code ); + $this->lock( $cacheKey ); + + $titleKey = wfMemcKey( 'messages', 'individual', $title ); + + if ( $text === false ) { + # Article was deleted + $this->mCache[$code][$title] = '!NONEXISTENT'; + $this->mMemc->delete( $titleKey ); + + } elseif ( strlen( $text ) > $wgMaxMsgCacheEntrySize ) { + # Check for size + $this->mCache[$code][$title] = '!TOO BIG'; + $this->mMemc->set( $titleKey, ' ' . $text, $this->mExpiry ); + + } else { + $this->mCache[$code][$title] = ' ' . $text; + $this->mMemc->delete( $titleKey ); + } + # Update caches + $this->saveToCaches( $this->mCache[$code], true, $code ); + $this->unlock( $cacheKey ); + + // Also delete cached sidebar... just in case it is affected + $codes = array( $code ); + if ( $code === 'en' ) { + // Delete all sidebars, like for example on action=purge on the + // sidebar messages + $codes = array_keys( Language::getLanguageNames() ); + } + + global $parserMemc; + foreach ( $codes as $code ) { + $sidebarKey = wfMemcKey( 'sidebar', $code ); + $parserMemc->delete( $sidebarKey ); + } + + // Update the message in the message blob store + global $wgContLang; + MessageBlobStore::updateMessage( $wgContLang->lcfirst( $msg ) ); + + wfRunHooks( "MessageCacheReplace", array( $title, $text ) ); + + wfProfileOut( __METHOD__ ); + } + + /** + * Shortcut to update caches. + * + * @param $cache Array: cached messages with a version. + * @param $memc Bool: Wether to update or not memcache. + * @param $code String: Language code. + * @return False on somekind of error. + */ + protected function saveToCaches( $cache, $memc = true, $code = false ) { wfProfileIn( __METHOD__ ); - $this->lock(); - $this->load(); - if ( is_array( $this->mCache ) ) { - if ( $text === false ) { - # Article was deleted - unset( $this->mCache[$title] ); - $this->mMemc->delete( "$this->mMemcKey:{$title}" ); - } elseif ( strlen( $text ) > $wgMaxMsgCacheEntrySize ) { - $this->mCache[$title] = '!TOO BIG'; - $this->mMemc->set( "$this->mMemcKey:{$title}", ' '.$text, $this->mExpiry ); + global $wgUseLocalMessageCache, $wgLocalMessageCacheSerialized; + + $cacheKey = wfMemcKey( 'messages', $code ); + + if ( $memc ) { + $success = $this->mMemc->set( $cacheKey, $cache, $this->mExpiry ); + } else { + $success = true; + } + + # Save to local cache + if ( $wgUseLocalMessageCache ) { + $serialized = serialize( $cache ); + $hash = md5( $serialized ); + $this->mMemc->set( wfMemcKey( 'messages', $code, 'hash' ), $hash, $this->mExpiry ); + if ($wgLocalMessageCacheSerialized) { + $this->saveToLocal( $serialized, $hash, $code ); } else { - $this->mCache[$title] = ' ' . $text; - $this->mMemc->delete( "$this->mMemcKey:{$title}" ); - } - $this->mMemc->set( $this->mMemcKey, $this->mCache, $this->mExpiry ); - - # Save to local cache - if ( $wgLocalMessageCache !== false ) { - $serialized = serialize( $this->mCache ); - $hash = md5( $serialized ); - $this->mMemc->set( "{$this->mMemcKey}-hash", $hash, $this->mExpiry ); - if ($wgLocalMessageCacheSerialized) { - $this->saveToLocal( $serialized,$hash ); - } else { - $this->saveToScript( $this->mCache, $hash ); - } + $this->saveToScript( $cache, $hash, $code ); } } - $this->unlock(); - $parserMemc->delete(wfMemcKey('sidebar')); + wfProfileOut( __METHOD__ ); + return $success; } /** - * Returns success * Represents a write lock on the messages key + * + * @return Boolean: success */ - function lock() { - if ( !$this->mUseCache ) { - return true; - } - - $lockKey = $this->mMemcKey . 'lock'; + function lock($key) { + $lockKey = $key . ':lock'; for ($i=0; $i < MSG_WAIT_TIMEOUT && !$this->mMemc->add( $lockKey, 1, MSG_LOCK_TIMEOUT ); $i++ ) { sleep(1); } @@ -383,113 +511,91 @@ class MessageCache { return $i >= MSG_WAIT_TIMEOUT; } - function unlock() { - if ( !$this->mUseCache ) { - return; - } - - $lockKey = $this->mMemcKey . 'lock'; + function unlock($key) { + $lockKey = $key . ':lock'; $this->mMemc->delete( $lockKey ); } /** * Get a message from either the content language or the user language. * - * @param string $key The message cache key - * @param bool $useDB Get the message from the DB, false to use only the localisation - * @param string $langcode Code of the language to get the message for, if - * it is a valid code create a language for that - * language, if it is a string but not a valid code - * then make a basic language object, if it is a - * false boolean then use the current users - * language (as a fallback for the old parameter - * functionality), or if it is a true boolean then - * use the wikis content language (also as a - * fallback). - * @param bool $isFullKey Specifies whether $key is a two part key "lang/msg". + * @param $key String: the message cache key + * @param $useDB Boolean: get the message from the DB, false to use only + * the localisation + * @param $langcode String: code of the language to get the message for, if + * it is a valid code create a language for that language, + * if it is a string but not a valid code then make a basic + * language object, if it is a false boolean then use the + * current users language (as a fallback for the old + * parameter functionality), or if it is a true boolean + * then use the wikis content language (also as a + * fallback). + * @param $isFullKey Boolean: specifies whether $key is a two part key + * "msg/lang". */ function get( $key, $useDB = true, $langcode = true, $isFullKey = false ) { - global $wgContLanguageCode, $wgContLang, $wgLang; - - # Identify which language to get or create a language object for. - if( $langcode === $wgContLang->getCode() || $langcode === true ) { - # $langcode is the language code of the wikis content language object. - # or it is a boolean and value is true - $lang =& $wgContLang; - } elseif( $langcode === $wgLang->getCode() || $langcode === false ) { - # $langcode is the language code of user language object. - # or it was a boolean and value is false - $lang =& $wgLang; - } else { - $validCodes = array_keys( Language::getLanguageNames() ); - if( in_array( $langcode, $validCodes ) ) { - # $langcode corresponds to a valid language. - $lang = Language::factory( $langcode ); - } else { - # $langcode is a string, but not a valid language code; use content language. - $lang =& $wgContLang; - wfDebug( 'Invalid language code passed to MessageCache::get, falling back to content language.' ); - } - } + global $wgLanguageCode, $wgContLang; - $langcode = $lang->getCode(); + if ( !is_string( $key ) ) { + throw new MWException( "Non-string key given" ); + } - # If uninitialised, someone is trying to call this halfway through Setup.php - if( !$this->mInitialised ) { - return '<' . htmlspecialchars($key) . '>'; + if ( strval( $key ) === '' ) { + # Shortcut: the empty key is always missing + return false; } - # If cache initialization was deferred, start it now. - if( $this->mDeferred && !$this->mDisable && $useDB ) { - $this->load(); + + $lang = wfGetLangObj( $langcode ); + if ( !$lang ) { + throw new MWException( "Bad lang code $langcode given" ); } + $langcode = $lang->getCode(); + $message = false; - # Normalise title-case input - $lckey = $wgContLang->lcfirst( $key ); - $lckey = str_replace( ' ', '_', $lckey ); + # Normalise title-case input (with some inlining) + $lckey = str_replace( ' ', '_', $key ); + if ( ord( $key ) < 128 ) { + $lckey[0] = strtolower( $lckey[0] ); + $uckey = ucfirst( $lckey ); + } else { + $lckey = $wgContLang->lcfirst( $lckey ); + $uckey = $wgContLang->ucfirst( $lckey ); + } + + /* Record each message request, but only once per request. + * This information is not used unless $wgAdaptiveMessageCache + * is enabled. */ + $this->mRequestedMessages[$uckey] = true; # Try the MediaWiki namespace if( !$this->mDisable && $useDB ) { - $title = $wgContLang->ucfirst( $lckey ); - if(!$isFullKey && ($langcode != $wgContLanguageCode) ) { + $title = $uckey; + if(!$isFullKey && ( $langcode != $wgLanguageCode ) ) { $title .= '/' . $langcode; } - $message = $this->getMsgFromNamespace( $title ); - } - # Try the extension array - if( $message === false && isset( $this->mExtensionMessages[$langcode][$lckey] ) ) { - $message = $this->mExtensionMessages[$langcode][$lckey]; - } - if ( $message === false && isset( $this->mExtensionMessages['en'][$lckey] ) ) { - $message = $this->mExtensionMessages['en'][$lckey]; + $message = $this->getMsgFromNamespace( $title, $langcode ); } # Try the array in the language object - if( $message === false ) { - #wfDebug( "Trying language object for message $key\n" ); - wfSuppressWarnings(); + if ( $message === false ) { $message = $lang->getMessage( $lckey ); - wfRestoreWarnings(); if ( is_null( $message ) ) { $message = false; } } # Try the array of another language - $pos = strrpos( $lckey, '/' ); - if( $message === false && $pos !== false) { - $mkey = substr( $lckey, 0, $pos ); - $code = substr( $lckey, $pos+1 ); - if ( $code ) { - $validCodes = array_keys( Language::getLanguageNames() ); - if ( in_array( $code, $validCodes ) ) { - $message = Language::getMessageFor( $mkey, $code ); - if ( is_null( $message ) ) { - $message = false; - } - } else { - wfDebug( __METHOD__ . ": Invalid code $code for $mkey/$code, not trying messages array\n" ); + if( $message === false ) { + $parts = explode( '/', $lckey ); + # We may get calls for things that are http-urls from sidebar + # Let's not load nonexistent languages for those + # They usually have more than one slash. + if ( count( $parts ) == 2 && $parts[1] !== '' ) { + $message = Language::getMessageFor( $parts[0], $parts[1] ); + if ( is_null( $message ) ) { + $message = false; } } } @@ -497,14 +603,25 @@ class MessageCache { # Is this a custom message? Try the default language in the db... if( ($message === false || $message === '-' ) && !$this->mDisable && $useDB && - !$isFullKey && ($langcode != $wgContLanguageCode) ) { - $message = $this->getMsgFromNamespace( $wgContLang->ucfirst( $lckey ) ); + !$isFullKey && ($langcode != $wgLanguageCode) ) { + $message = $this->getMsgFromNamespace( $uckey, $wgLanguageCode ); } # Final fallback if( $message === false ) { - return '<' . htmlspecialchars($key) . '>'; + return false; } + + # Fix whitespace + $message = strtr( $message, + array( + # Fix for trailing whitespace, removed by textarea + ' ' => ' ', + # Fix for NBSP, converted to space by firefox + ' ' => "\xc2\xa0", + ' ' => "\xc2\xa0", + ) ); + return $message; } @@ -512,86 +629,99 @@ class MessageCache { * Get a message from the MediaWiki namespace, with caching. The key must * first be converted to two-part lang/msg form if necessary. * - * @param string $title Message cache key with initial uppercase letter + * @param $title String: Message cache key with initial uppercase letter. + * @param $code String: code denoting the language to try. */ - function getMsgFromNamespace( $title ) { - $message = false; - $type = false; - - # Try the cache - if( $this->mUseCache && isset( $this->mCache[$title] ) ) { - $entry = $this->mCache[$title]; - $type = substr( $entry, 0, 1 ); - if ( $type == ' ' ) { + function getMsgFromNamespace( $title, $code ) { + global $wgAdaptiveMessageCache; + $big = false; + + $this->load( $code ); + if ( isset( $this->mCache[$code][$title] ) ) { + $entry = $this->mCache[$code][$title]; + if ( substr( $entry, 0, 1 ) === ' ' ) { return substr( $entry, 1 ); - } - } - - # Call message hooks, in case they are defined - wfRunHooks('MessagesPreLoad', array( $title, &$message ) ); - if ( $message !== false ) { - return $message; - } - - # If there is no cache entry and no placeholder, it doesn't exist - if ( $type != '!' && $message === false ) { - return false; - } - - $memcKey = $this->mMemcKey . ':' . $title; + } elseif ( $entry === '!NONEXISTENT' ) { + return false; + } elseif( $entry === '!TOO BIG' ) { + // Fall through and try invididual message cache below - # Try the individual message cache - if ( $this->mUseCache ) { - $entry = $this->mMemc->get( $memcKey ); - if ( $entry ) { - $type = substr( $entry, 0, 1 ); - - if ( $type == ' ' ) { - $message = substr( $entry, 1 ); - $this->mCache[$title] = $entry; + } else { + // XXX: This is not cached in process cache, should it? + $message = false; + wfRunHooks('MessagesPreLoad', array( $title, &$message ) ); + if ( $message !== false ) { return $message; - } elseif ( $entry == '!NONEXISTENT' ) { + } + + /* If message cache is in normal mode, it is guaranteed + * (except bugs) that there is always entry (or placeholder) + * in the cache if message exists. Thus we can do minor + * performance improvement and return false early. + */ + if ( !$wgAdaptiveMessageCache ) { return false; - } else { - # Corrupt/obsolete entry, delete it - $this->mMemc->delete( $memcKey ); } + } + } + # Try the individual message cache + $titleKey = wfMemcKey( 'messages', 'individual', $title ); + $entry = $this->mMemc->get( $titleKey ); + if ( $entry ) { + if ( substr( $entry, 0, 1 ) === ' ' ) { + $this->mCache[$code][$title] = $entry; + return substr( $entry, 1 ); + } elseif ( $entry === '!NONEXISTENT' ) { + $this->mCache[$code][$title] = '!NONEXISTENT'; + return false; + } else { + # Corrupt/obsolete entry, delete it + $this->mMemc->delete( $titleKey ); } } - # Try loading it from the DB + # Try loading it from the database $revision = Revision::newFromTitle( Title::makeTitle( NS_MEDIAWIKI, $title ) ); - if( $revision ) { + if ( $revision ) { $message = $revision->getText(); - if ($this->mUseCache) { - $this->mCache[$title] = ' ' . $message; - $this->mMemc->set( $memcKey, $message, $this->mExpiry ); - } + $this->mCache[$code][$title] = ' ' . $message; + $this->mMemc->set( $titleKey, ' ' . $message, $this->mExpiry ); } else { - # Negative caching - # Use some special text instead of false, because false gets converted to '' somewhere - $this->mMemc->set( $memcKey, '!NONEXISTENT', $this->mExpiry ); - $this->mCache[$title] = false; + $message = false; + $this->mCache[$code][$title] = '!NONEXISTENT'; + $this->mMemc->set( $titleKey, '!NONEXISTENT', $this->mExpiry ); } return $message; } - function transform( $message, $interface = false ) { - global $wgParser; + function transform( $message, $interface = false, $language = null ) { + // Avoid creating parser if nothing to transform + if( strpos( $message, '{{' ) === false ) { + return $message; + } + + global $wgParser, $wgParserConf; if ( !$this->mParser && isset( $wgParser ) ) { # Do some initialisation so that we don't have to do it twice $wgParser->firstCallInit(); # Clone it and store it - $this->mParser = clone $wgParser; + $class = $wgParserConf['class']; + if ( $class == 'Parser_DiffTest' ) { + # Uncloneable + $this->mParser = new $class( $wgParserConf ); + } else { + $this->mParser = clone $wgParser; + } + #wfDebug( __METHOD__ . ": following contents triggered transform: $message\n" ); } if ( $this->mParser ) { - if( strpos( $message, '{{' ) !== false ) { - $popts = $this->getParserOptions(); - $popts->setInterfaceMessage( $interface ); - $message = $this->mParser->transformMsg( $message, $popts ); - } + $popts = $this->getParserOptions(); + $popts->setInterfaceMessage( $interface ); + $popts->setTargetLanguage( $language ); + $popts->setUserLang( $language ); + $message = $this->mParser->transformMsg( $message, $popts ); } return $message; } @@ -615,153 +745,145 @@ class MessageCache { } /** + * Clear all stored messages. Mainly used after a mass rebuild. + */ + function clear() { + $langs = Language::getLanguageNames( false ); + foreach ( array_keys($langs) as $code ) { + # Global cache + $this->mMemc->delete( wfMemcKey( 'messages', $code ) ); + # Invalidate all local caches + $this->mMemc->delete( wfMemcKey( 'messages', $code, 'hash' ) ); + } + $this->mLoadedLanguages = array(); + } + + /** * Add a message to the cache + * @deprecated Use $wgExtensionMessagesFiles * - * @param mixed $key - * @param mixed $value - * @param string $lang The messages language, English by default + * @param $key Mixed + * @param $value Mixed + * @param $lang String: the messages language, English by default */ function addMessage( $key, $value, $lang = 'en' ) { - $this->mExtensionMessages[$lang][$key] = $value; + wfDeprecated( __METHOD__ ); + $lc = Language::getLocalisationCache(); + $lc->addLegacyMessages( array( $lang => array( $key => $value ) ) ); } /** * Add an associative array of message to the cache + * @deprecated Use $wgExtensionMessagesFiles * - * @param array $messages An associative array of key => values to be added - * @param string $lang The messages language, English by default + * @param $messages Array: an associative array of key => values to be added + * @param $lang String: the messages language, English by default */ function addMessages( $messages, $lang = 'en' ) { - wfProfileIn( __METHOD__ ); - if ( !is_array( $messages ) ) { - throw new MWException( __METHOD__.': Invalid message array' ); - } - if ( isset( $this->mExtensionMessages[$lang] ) ) { - $this->mExtensionMessages[$lang] = $messages + $this->mExtensionMessages[$lang]; - } else { - $this->mExtensionMessages[$lang] = $messages; - } - wfProfileOut( __METHOD__ ); + wfDeprecated( __METHOD__ ); + $lc = Language::getLocalisationCache(); + $lc->addLegacyMessages( array( $lang => $messages ) ); } /** * Add a 2-D array of messages by lang. Useful for extensions. + * @deprecated Use $wgExtensionMessagesFiles * - * @param array $messages The array to be added + * @param $messages Array: the array to be added */ function addMessagesByLang( $messages ) { - wfProfileIn( __METHOD__ ); - foreach ( $messages as $key => $value ) { - $this->addMessages( $value, $key ); - } - wfProfileOut( __METHOD__ ); + wfDeprecated( __METHOD__ ); + $lc = Language::getLocalisationCache(); + $lc->addLegacyMessages( $messages ); } /** - * Get the extension messages for a specific language. Only English, interface - * and content language are guaranteed to be loaded. - * - * @param string $lang The messages language, English by default + * Set a hook for addMessagesByLang() */ - function getExtensionMessagesFor( $lang = 'en' ) { - wfProfileIn( __METHOD__ ); - $messages = array(); - if ( isset( $this->mExtensionMessages[$lang] ) ) { - $messages = $this->mExtensionMessages[$lang]; - } - if ( $lang != 'en' ) { - $messages = $messages + $this->mExtensionMessages['en']; - } - wfProfileOut( __METHOD__ ); - return $messages; + function setExtensionMessagesHook( $callback ) { + $this->mAddMessagesHook = $callback; } /** - * Clear all stored messages. Mainly used after a mass rebuild. + * @deprecated */ - function clear() { - global $wgLocalMessageCache; - if( $this->mUseCache ) { - # Global cache - $this->mMemc->delete( $this->mMemcKey ); - # Invalidate all local caches - $this->mMemc->delete( "{$this->mMemcKey}-hash" ); - } + function loadAllMessages( $lang = false ) { + } + + /** + * @deprecated + */ + function loadMessagesFile( $filename, $langcode = false ) { + } + + public function figureMessage( $key ) { + global $wgLanguageCode; + $pieces = explode( '/', $key ); + if( count( $pieces ) < 2 ) + return array( $key, $wgLanguageCode ); + + $lang = array_pop( $pieces ); + $validCodes = Language::getLanguageNames(); + if( !array_key_exists( $lang, $validCodes ) ) + return array( $key, $wgLanguageCode ); + + $message = implode( '/', $pieces ); + return array( $message, $lang ); } - function loadAllMessages() { - global $wgExtensionMessagesFiles; - if ( $this->mAllMessagesLoaded ) { + public static function logMessages() { + global $wgMessageCache, $wgAdaptiveMessageCache; + if ( !$wgAdaptiveMessageCache || !$wgMessageCache instanceof MessageCache ) { return; } - $this->mAllMessagesLoaded = true; - - # Some extensions will load their messages when you load their class file - wfLoadAllExtensions(); - # Others will respond to this hook - wfRunHooks( 'LoadAllMessages' ); - # Some register their messages in $wgExtensionMessagesFiles - foreach ( $wgExtensionMessagesFiles as $name => $file ) { - if ( $file ) { - $this->loadMessagesFile( $file ); - $wgExtensionMessagesFiles[$name] = false; - } + + $cachekey = wfMemckey( 'message-profiling' ); + $cache = wfGetCache( CACHE_DB ); + $data = $cache->get( $cachekey ); + + if ( !$data ) $data = array(); + + $age = self::$mAdaptiveDataAge; + $filterDate = substr( wfTimestamp( TS_MW, time()-$age ), 0, 8 ); + foreach ( array_keys( $data ) as $key ) { + if ( $key < $filterDate ) unset( $data[$key] ); + } + + $index = substr( wfTimestampNow(), 0, 8 ); + if ( !isset( $data[$index] ) ) $data[$index] = array(); + + foreach ( $wgMessageCache->mRequestedMessages as $message => $_ ) { + if ( !isset( $data[$index][$message] ) ) $data[$index][$message] = 0; + $data[$index][$message]++; } - # Still others will respond to neither, they are EVIL. We sometimes need to know! + + $cache->set( $cachekey, $data ); } - /** - * Load messages from a given file - * - * @param string $filename Filename of file to load. - * @param boolean $all Whether or not to load all languages. - */ - function loadMessagesFile( $filename, $all = false ) { - global $wgLang, $wgContLang; - $messages = $magicWords = false; - require( $filename ); - - if( $all ) { - # Load all messages, regardless of language. - $validCodes = array_keys( Language::getLanguageNames() ); - foreach( $messages as $code => $array ) { - if( in_array( $code, $validCodes ) ) { - $this->processMessagesArray( $messages, $code ); - } - } - } else { - # Load only languages that are usually used, and merge all - # fallbacks, except English. - $langs = array_unique( array( 'en', $wgContLang->getCode(), $wgLang->getCode() ) ); - foreach( $langs as $code ) { - $this->processMessagesArray( $messages, $code ); + public function getMostUsedMessages() { + $cachekey = wfMemckey( 'message-profiling' ); + $cache = wfGetCache( CACHE_DB ); + $data = $cache->get( $cachekey ); + if ( !$data ) return array(); + + $list = array(); + + foreach( $data as $date => $messages ) { + foreach( $messages as $message => $count ) { + $key = $message; + if ( !isset( $list[$key] ) ) $list[$key] = 0; + $list[$key] += $count; } } - if ( $magicWords !== false ) { - global $wgContLang; - $wgContLang->addMagicWordsByLang( $magicWords ); + $max = max( $list ); + foreach ( $list as $message => $count ) { + if ( $count < intval( $max * self::$mAdaptiveInclusionThreshold ) ) { + unset( $list[$message] ); + } } - } - /** - * Process an array of messages, loading it into the message cache. - * - * @param array $messages Messages array. - * @param string $language Language code to process. - */ - function processMessagesArray( $messages, $language ) { - $fallbackCode = $language; - $mergedMessages = array(); - do { - if ( isset($messages[$fallbackCode]) ) { - $mergedMessages += $messages[$fallbackCode]; - } - $fallbackCode = Language::getFallbackfor( $fallbackCode ); - } while( $fallbackCode && $fallbackCode !== 'en' ); - - if ( !empty($mergedMessages) ) - $this->addMessages( $mergedMessages, $language ); + return array_keys( $list ); } }