Defer shutting down Chrome after the CF automation connection goes away.

chrome/browser/automation/chrome_frame_automation_provider.cc

Index: chrome/browser/automation/chrome_frame_automation_provider.cc
diff --git a/chrome/browser/automation/chrome_frame_automation_provider.cc b/chrome/browser/automation/chrome_frame_automation_provider.cc
index ce2f6a42d095bb7cad8534ab2d900338073a8ffc..42e7560d514d21f8863f86e9e4e0fd7acd1e12aa 100644
--- a/chrome/browser/automation/chrome_frame_automation_provider.cc
+++ b/chrome/browser/automation/chrome_frame_automation_provider.cc
@@ -1,15 +1,21 @@
-// Copyright (c) 2011 The Chromium Authors. All rights reserved.
+// Copyright (c) 2012 The Chromium Authors. All rights reserved.
 // Use of this source code is governed by a BSD-style license that can be
 // found in the LICENSE file.
 
 #include "chrome/browser/automation/chrome_frame_automation_provider.h"
+
+#include "base/command_line.h"
+#include "base/string_number_conversions.h"
 #include "chrome/browser/browser_process.h"
 #include "chrome/browser/profiles/profile.h"
 #include "chrome/browser/profiles/profile_manager.h"
 #include "chrome/common/automation_messages.h"
+#include "chrome/common/chrome_switches.h"
 #include "ipc/ipc_message.h"
 #include "ipc/ipc_channel.h"
 
+const int kMaxChromeShutdownDelaySeconds = 60*60;
+
 ChromeFrameAutomationProvider::ChromeFrameAutomationProvider(Profile* profile)
     : AutomationProvider(profile) {
   DCHECK(g_browser_process);
@@ -19,8 +25,37 @@ ChromeFrameAutomationProvider::ChromeFrameAutomationProvider(Profile* profile)
 
 ChromeFrameAutomationProvider::~ChromeFrameAutomationProvider() {
   DCHECK(g_browser_process);
-  if (g_browser_process)
-    g_browser_process->ReleaseModule();
+  if (g_browser_process) {
+    CommandLine& cmd_line = *CommandLine::ForCurrentProcess();
+
+    CommandLine::StringType shutdown_delay(
+        cmd_line.GetSwitchValueNative(switches::kChromeFrameShutdownDelay));
+    if (!shutdown_delay.empty()) {
+      VLOG(1) << "ChromeFrameAutomationProvider: "
+                 "Scheduling ReleaseBrowserProcess.";
+
+      // Grab the specified shutdown delay.
+      int shutdown_delay_seconds = 0;
+      base::StringToInt(shutdown_delay, &shutdown_delay_seconds);
+
+      // Clamp to reasonable values.
+      if (shutdown_delay_seconds < 0)
+        shutdown_delay_seconds = 0;
+      if (shutdown_delay_seconds > kMaxChromeShutdownDelaySeconds)

[grt (UTC plus 2), 2012/03/02 20:43:23]

nit: else if

[robertshield, 2012/03/02 20:52:27]

Done.

+        shutdown_delay_seconds = kMaxChromeShutdownDelaySeconds;
+
+      // We have Chrome Frame defer Chrome shutdown for a time to improve
+      // intra-page load times.
+      MessageLoop::current()->PostDelayedTask(
+          FROM_HERE,
+          base::Bind(&ChromeFrameAutomationProvider::ReleaseBrowserProcess),
+          base::TimeDelta::FromSeconds(shutdown_delay_seconds));
+    } else {
+      VLOG(1) << "ChromeFrameAutomationProvider: "
+                 "Releasing browser module with no delay.";
+      g_browser_process->ReleaseModule();
+    }
+  }
 }
 
 bool ChromeFrameAutomationProvider::OnMessageReceived(
@@ -86,3 +121,12 @@ bool ChromeFrameAutomationProvider::IsValidMessage(uint32 type) {
 
   return is_valid_message;
 }
+
+// static
+void ChromeFrameAutomationProvider::ReleaseBrowserProcess() {
+  if (g_browser_process) {
+    VLOG(1) << "ChromeFrameAutomationProvider: "
+               "Releasing browser process.";
+    g_browser_process->ReleaseModule();
+  }
+}