chrome/browser/process_singleton_startup_lock.cc

   1 // Copyright (c) 2013 The Chromium Authors. All rights reserved.
   2 // Use of this source code is governed by a BSD-style license that can be
   3 // found in the LICENSE file.
   4
   5 #include "chrome/browser/process_singleton_startup_lock.h"
   6
   7 #include "base/bind.h"
   8 #include "base/logging.h"
   9
  10 ProcessSingletonStartupLock::ProcessSingletonStartupLock(
  11     const ProcessSingleton::NotificationCallback& original_callback)
  12     : locked_(true),
  13       original_callback_(original_callback) {}
  14
  15 ProcessSingletonStartupLock::~ProcessSingletonStartupLock() {}
  16
  17 ProcessSingleton::NotificationCallback
  18 ProcessSingletonStartupLock::AsNotificationCallback() {
  19   return base::Bind(&ProcessSingletonStartupLock::NotificationCallbackImpl,
  20                     base::Unretained(this));
  21 }
  22
  23 void ProcessSingletonStartupLock::Unlock() {
  24   DCHECK(CalledOnValidThread());
  25   locked_ = false;
  26
  27   // Replay the command lines of the messages which were received while the
  28   // ProcessSingleton was locked. Only replay each message once.
  29   std::set<DelayedStartupMessage> replayed_messages;
  30   for (std::vector<DelayedStartupMessage>::const_iterator it =
  31            saved_startup_messages_.begin();
  32        it != saved_startup_messages_.end(); ++it) {
  33     if (replayed_messages.find(*it) != replayed_messages.end())
  34       continue;
  35     original_callback_.Run(CommandLine(it->first), it->second);
  36     replayed_messages.insert(*it);
  37   }
  38   saved_startup_messages_.clear();
  39 }
  40
  41 bool ProcessSingletonStartupLock::NotificationCallbackImpl(
  42     const CommandLine& command_line,
  43     const base::FilePath& current_directory) {
  44   if (locked_) {
  45     // If locked, it means we are not ready to process this message because
  46     // we are probably in a first run critical phase.
  47     saved_startup_messages_.push_back(
  48         std::make_pair(command_line.argv(), current_directory));
  49     return true;
  50   } else {
  51     return original_callback_.Run(command_line, current_directory);
  52   }
  53 }