From 25b1a5360b9b2ca3e7d89727266f429a794b260d Mon Sep 17 00:00:00 2001
From: jeroendedauw <jeroendedauw@gmail.com>
Date: Fri, 6 Jul 2012 13:27:23 +0200
Subject: [PATCH] Always save the cache if stuff was added to it - dont require
 deriving class to call save

Change-Id: Ifd57c4c3f3d10be97080c9aae5599f6057ea9baf
---
 includes/specials/SpecialCachedPage.php | 35 ++++++++++++++++++-------
 1 file changed, 25 insertions(+), 10 deletions(-)

diff --git a/includes/specials/SpecialCachedPage.php b/includes/specials/SpecialCachedPage.php
index 32eb2e751e..f5e1134d9a 100644
--- a/includes/specials/SpecialCachedPage.php
+++ b/includes/specials/SpecialCachedPage.php
@@ -56,6 +56,19 @@ abstract class SpecialCachedPage extends SpecialPage implements ICacheHelper {
 	 */
 	protected $cacheEnabled = true;
 
+	/**
+	 * Gets called after @see SpecialPage::execute.
+	 *
+	 * @since 1.20
+	 *
+	 * @param $subPage string|null
+	 */
+	protected function afterExecute( $subPage ) {
+		$this->saveCache();
+
+		parent::afterExecute( $subPage );
+	}
+
 	/**
 	 * Sets if the cache should be enabled or not.
 	 *
@@ -76,21 +89,23 @@ abstract class SpecialCachedPage extends SpecialPage implements ICacheHelper {
 	 * @param boolean|null $cacheEnabled Sets if the cache should be enabled or not.
 	 */
 	public function startCache( $cacheExpiry = null, $cacheEnabled = null ) {
-		$this->cacheHelper = new CacheHelper();
+		if ( !isset( $this->cacheHelper ) ) {
+			$this->cacheHelper = new CacheHelper();
 
-		$this->cacheHelper->setCacheEnabled( $this->cacheEnabled );
-		$this->cacheHelper->setOnInitializedHandler( array( $this, 'onCacheInitialized' ) );
+			$this->cacheHelper->setCacheEnabled( $this->cacheEnabled );
+			$this->cacheHelper->setOnInitializedHandler( array( $this, 'onCacheInitialized' ) );
 
-		$keyArgs = $this->getCacheKey();
+			$keyArgs = $this->getCacheKey();
 
-		if ( array_key_exists( 'action', $keyArgs ) && $keyArgs['action'] === 'purge' ) {
-			unset( $keyArgs['action'] );
-		}
+			if ( array_key_exists( 'action', $keyArgs ) && $keyArgs['action'] === 'purge' ) {
+				unset( $keyArgs['action'] );
+			}
 
-		$this->cacheHelper->setCacheKey( $keyArgs );
+			$this->cacheHelper->setCacheKey( $keyArgs );
 
-		if ( $this->getRequest()->getText( 'action' ) === 'purge' ) {
-			$this->cacheHelper->rebuildOnDemand();
+			if ( $this->getRequest()->getText( 'action' ) === 'purge' ) {
+				$this->cacheHelper->rebuildOnDemand();
+			}
 		}
 
 		$this->cacheHelper->startCache( $cacheExpiry, $cacheEnabled );
-- 
2.20.1