From 3257718a0226e7d2dc6775a1aea0597deefe5342 Mon Sep 17 00:00:00 2001
From: Dries Buytaert <dries@buytaert.net>
Date: Wed, 8 Oct 2008 11:11:09 +0000
Subject: - Patch #309821 by sun et al: automatic site maintenance during
 update.php.

---
 update.php | 21 +++++++++++----------
 1 file changed, 11 insertions(+), 10 deletions(-)

(limited to 'update.php')

diff --git a/update.php b/update.php
index 36a214a11..09cd8f2bd 100644
--- a/update.php
+++ b/update.php
@@ -290,10 +290,10 @@ function update_script_selection_form() {
 function update_batch() {
   global $base_url;
 
-  // During the update, toggle site maintenance so that schema changes
-  // don't effect visiting users.
-  $site_offline = variable_get('site_offline', FALSE);
-  if ($site_offline == FALSE) {
+  // During the update, toggle site maintenance so that schema changes do not
+  // affect visiting users.
+  $_SESSION['site_offline'] = variable_get('site_offline', FALSE);
+  if ($_SESSION['site_offline'] == FALSE) {
     variable_set('site_offline', TRUE);
   }
 
@@ -320,12 +320,6 @@ function update_batch() {
   );
   batch_set($batch);
   batch_process($base_url . '/update.php?op=results', $base_url . '/update.php');
-
-  // Now that the update is done, we can disable site maintenance if
-  // it was previously turned off.
-  if ($site_offline == FALSE) {
-    variable_set('site_offline', FALSE);
-  }
 }
 
 function update_finished($success, $results, $operations) {
@@ -335,6 +329,13 @@ function update_finished($success, $results, $operations) {
   $_SESSION['update_results'] = $results;
   $_SESSION['update_success'] = $success;
   $_SESSION['updates_remaining'] = $operations;
+  
+  // Now that the update is done, we can disable site maintenance if it was
+  // previously turned off.
+  if (isset($_SESSION['site_offline']) && $_SESSION['site_offline'] == FALSE) {
+    variable_set('site_offline', FALSE);
+    unset($_SESSION['site_offline']);
+  }
 }
 
 function update_helpful_links() {
-- 
cgit v1.2.3