Chromium Code Reviews| Index: chrome/browser/process_singleton_lock.cc |
| diff --git a/chrome/browser/process_singleton_lock.cc b/chrome/browser/process_singleton_lock.cc |
| index 69636b76f8b3cbf1457b43c25a0bdbeed09605a1..9c5e837680b103831565c983e6771429300edf34 100644 |
| --- a/chrome/browser/process_singleton_lock.cc |
| +++ b/chrome/browser/process_singleton_lock.cc |
| @@ -1,54 +1,85 @@ |
| -// Copyright (c) 2013 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/process_singleton_lock.h" |
| - |
| -#include "base/bind.h" |
| -#include "base/logging.h" |
| - |
| -ProcessSingletonLock::ProcessSingletonLock( |
| - const ProcessSingleton::NotificationCallback& original_callback) |
| - : locked_(true), |
| - original_callback_(original_callback) {} |
| - |
| -ProcessSingletonLock::~ProcessSingletonLock() {} |
| - |
| -ProcessSingleton::NotificationCallback |
| -ProcessSingletonLock::AsNotificationCallback() { |
| - return base::Bind(&ProcessSingletonLock::NotificationCallbackImpl, |
| - base::Unretained(this)); |
| -} |
| - |
| -void ProcessSingletonLock::Unlock() { |
| - DCHECK(CalledOnValidThread()); |
| - locked_ = false; |
| - |
| - // Replay the command lines of the messages which were received while the |
| - // ProcessSingleton was locked. Only replay each message once. |
| - std::set<DelayedStartupMessage> replayed_messages; |
| - for (std::vector<DelayedStartupMessage>::const_iterator it = |
| - saved_startup_messages_.begin(); |
| - it != saved_startup_messages_.end(); ++it) { |
| - if (replayed_messages.find(*it) != |
| - replayed_messages.end()) |
| - continue; |
| - original_callback_.Run(CommandLine(it->first), it->second); |
| - replayed_messages.insert(*it); |
| - } |
| - saved_startup_messages_.clear(); |
| -} |
| - |
| -bool ProcessSingletonLock::NotificationCallbackImpl( |
| - const CommandLine& command_line, |
| - const base::FilePath& current_directory) { |
| - if (locked_) { |
| - // If locked, it means we are not ready to process this message because |
| - // we are probably in a first run critical phase. |
| - saved_startup_messages_.push_back( |
| - std::make_pair(command_line.argv(), current_directory)); |
| - return true; |
| - } else { |
| - return original_callback_.Run(command_line, current_directory); |
| - } |
| -} |
| +// Copyright (c) 2013 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/process_singleton_lock.h" |
| + |
| +#include "base/bind.h" |
| +#include "base/logging.h" |
| +#include "chrome/browser/operation_output.h" |
| + |
| +ProcessSingletonLock::ProcessSingletonLock( |
| + const OperationCallback& operation_callback) |
| + : locked_(true), |
| + operation_callback_(operation_callback) {} |
| + |
| +ProcessSingletonLock::~ProcessSingletonLock() { |
| + for (std::vector<DelayedStartupMessage>::const_iterator it = |
| + saved_startup_messages_.begin(); |
| + it != saved_startup_messages_.end(); ++it) { |
| + if (it->operation_output_) |
| + delete it->operation_output_; |
| + } |
| +} |
| + |
| +ProcessSingleton::NotificationCallback |
| +ProcessSingletonLock::AsNotificationCallback() { |
| + return base::Bind(&ProcessSingletonLock::NotificationCallbackImpl, |
| + base::Unretained(this)); |
| +} |
| + |
| +void ProcessSingletonLock::Unlock() { |
| + DCHECK(CalledOnValidThread()); |
| + locked_ = false; |
| + |
| + // Replay the command lines of the messages which were received while the |
| + // ProcessSingleton was locked. Only replay each message once. |
| + std::set<std::pair<CommandLine::StringVector, |
| + base::FilePath> > replayed_messages; |
| + for (std::vector<DelayedStartupMessage>::const_iterator it = |
| + saved_startup_messages_.begin(); |
| + it != saved_startup_messages_.end(); ++it) { |
| + scoped_ptr<OperationOutput> operation_output(it->operation_output_); |
| + std::pair<CommandLine::StringVector, base::FilePath> pair( |
| + it->argv_, it->current_directory_); |
| + if (replayed_messages.find(pair) != replayed_messages.end()) |
| + continue; |
| + operation_callback_.Run(CommandLine(it->argv_), |
| + it->current_directory_, |
| + operation_output.Pass()); |
| + replayed_messages.insert(pair); |
| + } |
| + saved_startup_messages_.clear(); |
| +} |
| + |
| +bool ProcessSingletonLock::NotificationCallbackImpl( |
| + const CommandLine& command_line, |
| + const base::FilePath& current_directory) { |
| + scoped_ptr<OperationOutput> operation_output = |
| + OperationOutput::Create(command_line); |
| + |
| + if (locked_) { |
| + // If locked, it means we are not ready to process this message because |
| + // we are probably in a first run critical phase. |
| + saved_startup_messages_.push_back( |
|
robertshield
2013/03/28 14:31:48
minor nit: since these are only going to be played
gab
2013/03/28 14:52:41
I thought about this too, the theory being that in
|
| + DelayedStartupMessage(command_line.argv(), |
| + current_directory, |
| + operation_output.release())); |
| + return true; |
| + } else { |
| + return operation_callback_.Run(command_line, |
| + current_directory, |
| + operation_output.Pass()); |
| + } |
| +} |
| + |
| +ProcessSingletonLock::DelayedStartupMessage::DelayedStartupMessage( |
| + const CommandLine::StringVector& argv, |
| + const base::FilePath& current_directory, |
| + OperationOutput* operation_output) |
| + : argv_(argv), |
| + current_directory_(current_directory), |
| + operation_output_(operation_output) { |
| +} |
| + |
| +ProcessSingletonLock::DelayedStartupMessage::~DelayedStartupMessage() { } |