4 * Abstract special page class with scaffolding for caching the HTML output.
6 * To add HTML that should be cached, use addCachedHTML like this:
7 * $this->addCachedHTML( $callback );
9 * The callback function is only called when needed, so do all your expensive
10 * computations here. This function should returns the HTML to be cached.
11 * It should not add anything to the PageOutput object!
13 * Before the first addCachedHTML call, you should call $this->startCache();
14 * After adding the last HTML that should be cached, call $this->saveCache();
18 * @file SpecialCachedPage.php
19 * @ingroup SpecialPage
21 * @licence GNU GPL v2 or later
22 * @author Jeroen De Dauw < jeroendedauw@gmail.com >
24 abstract class SpecialCachedPage
extends SpecialPage
{
27 * The time to live for the cache, in seconds or a unix timestamp indicating the point of expiry.
32 protected $cacheExpiry = 3600;
35 * List of HTML chunks to be cached (if !hasCached) or that where cashed (of hasCached).
36 * If no cached already, then the newly computed chunks are added here,
37 * if it as cached already, chunks are removed from this list as they are needed.
42 protected $cachedChunks;
45 * Indicates if the to be cached content was already cached.
46 * Null if this information is not available yet.
51 protected $hasCached = null;
54 * If the cache is enabled or not.
59 protected $cacheEnabled = true;
62 * Sets if the cache should be enabled or not.
65 * @param boolean $cacheEnabled
67 public function setCacheEnabled( $cacheEnabled ) {
68 $this->cacheEnabled
= $cacheEnabled;
72 * Initializes the caching.
73 * Should be called before the first time anything is added via addCachedHTML.
77 * @param integer|null $cacheExpiry Sets the cache expirty, either ttl in seconds or unix timestamp.
78 * @param boolean|null $cacheEnabled Sets if the cache should be enabled or not.
80 public function startCache( $cacheExpiry = null, $cacheEnabled = null ) {
81 if ( is_null( $this->hasCached
) ) {
82 if ( !is_null( $cacheExpiry ) ) {
83 $this->cacheExpiry
= $cacheExpiry;
86 if ( !is_null( $cacheEnabled ) ) {
87 $this->setCacheEnabled( $cacheEnabled );
90 if ( $this->getRequest()->getText( 'action' ) === 'purge' ) {
91 $this->hasCached
= false;
99 * Returns a message that notifies the user he/she is looking at
100 * a cached version of the page, including a refresh link.
106 protected function getCachedNotice() {
107 $refreshArgs = $this->getRequest()->getQueryValues();
108 unset( $refreshArgs['title'] );
109 $refreshArgs['action'] = 'purge';
111 $subPage = $this->getTitle()->getFullText();
112 $subPage = explode( '/', $subPage, 2 );
113 $subPage = count( $subPage ) > 1 ?
$subPage[1] : false;
115 $refreshLink = Linker
::link(
116 $this->getTitle( $subPage ),
117 $this->msg( 'cachedspecial-refresh-now' )->escaped(),
122 if ( $this->cacheExpiry
< 86400 * 3650 ) {
123 $message = $this->msg(
124 'cachedspecial-viewing-cached-ttl',
125 $this->getLanguage()->formatDuration( $this->cacheExpiry
)
129 $message = $this->msg(
130 'cachedspecial-viewing-cached-ts'
134 return $message . ' ' . $refreshLink;
138 * Initializes the caching if not already done so.
139 * Should be called before any of the caching functionality is used.
143 protected function initCaching() {
144 if ( $this->cacheEnabled
&& is_null( $this->hasCached
) ) {
145 $cachedChunks = wfGetCache( CACHE_ANYTHING
)->get( $this->getCacheKeyString() );
147 $this->hasCached
= is_array( $cachedChunks );
148 $this->cachedChunks
= $this->hasCached ?
$cachedChunks : array();
150 $this->onCacheInitialized();
155 * Gets called after the cache got initialized.
159 protected function onCacheInitialized() {
160 if ( $this->hasCached
) {
161 $this->getOutput()->setSubtitle( $this->getCachedNotice() );
166 * Add some HTML to be cached.
167 * This is done by providing a callback function that should
168 * return the HTML to be added. It will only be called if the
169 * item is not in the cache yet or when the cache has been invalidated.
173 * @param {function} $computeFunction
175 * @param string|null $key
177 public function addCachedHTML( $computeFunction, $args = array(), $key = null ) {
178 $this->getOutput()->addHTML( $this->getCachedValue( $computeFunction, $args, $key ) );
182 * Get a cached value if available or compute it if not and then cache it if possible.
183 * The provided $computeFunction is only called when the computation needs to happen
184 * and should return a result value. $args are arguments that will be passed to the
185 * compute function when called.
189 * @param {function} $computeFunction
190 * @param array|mixed $args
191 * @param string|null $key
195 protected function getCachedValue( $computeFunction, $args = array(), $key = null ) {
196 $this->initCaching();
198 if ( $this->cacheEnabled
&& $this->hasCached
) {
201 if ( is_null( $key ) ) {
202 $itemKey = array_keys( array_slice( $this->cachedChunks
, 0, 1 ) );
203 $itemKey = array_shift( $itemKey );
205 if ( !is_integer( $itemKey ) ) {
206 wfWarn( "Attempted to get item with non-numeric key while the next item in the queue has a key ($itemKey) in " . __METHOD__
);
208 elseif ( is_null( $itemKey ) ) {
209 wfWarn( "Attempted to get an item while the queue is empty in " . __METHOD__
);
212 $value = array_shift( $this->cachedChunks
);
216 if ( array_key_exists( $key, $this->cachedChunks
) ) {
217 $value = $this->cachedChunks
[$key];
218 unset( $this->cachedChunks
[$key] );
221 wfWarn( "There is no item with key '$key' in this->cachedChunks in " . __METHOD__
);
226 if ( !is_array( $args ) ) {
227 $args = array( $args );
230 $value = call_user_func_array( $computeFunction, $args );
232 if ( $this->cacheEnabled
) {
233 if ( is_null( $key ) ) {
234 $this->cachedChunks
[] = $value;
237 $this->cachedChunks
[$key] = $value;
246 * Saves the HTML to the cache in case it got recomputed.
247 * Should be called after the last time anything is added via addCachedHTML.
251 public function saveCache() {
252 if ( $this->cacheEnabled
&& $this->hasCached
=== false && !empty( $this->cachedChunks
) ) {
253 wfGetCache( CACHE_ANYTHING
)->set( $this->getCacheKeyString(), $this->cachedChunks
, $this->cacheExpiry
);
258 * Sets the time to live for the cache, in seconds or a unix timestamp indicating the point of expiry..
262 * @param integer $cacheExpiry
264 protected function setExpirey( $cacheExpiry ) {
265 $this->cacheExpiry
= $cacheExpiry;
269 * Returns the cache key to use to cache this page's HTML output.
270 * Is constructed from the special page name and language code.
276 protected function getCacheKeyString() {
277 $keyArgs = $this->getCacheKey();
279 if ( array_key_exists( 'action', $keyArgs ) && $keyArgs['action'] === 'purge' ) {
280 unset( $keyArgs['action'] );
283 return call_user_func_array( 'wfMemcKey', $keyArgs );
287 * Returns the variables used to constructed the cache key in an array.
293 protected function getCacheKey() {
296 $this->getLanguage()->getCode()