license.bot | f003cfe | 2008-08-24 09:55:55 +0900 | [diff] [blame^] | 1 | // Copyright (c) 2006-2008 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 "base/at_exit.h"
|
| 6 | #include "base/logging.h"
|
| 7 |
|
| 8 | namespace base {
|
| 9 |
|
| 10 | // Keep a stack of registered AtExitManagers. We always operate on the most
|
| 11 | // recent, and we should never have more than one outside of testing, when we
|
| 12 | // use the shadow version of the constructor. We don't protect this for
|
| 13 | // thread-safe access, since it will only be modified in testing.
|
| 14 | static AtExitManager* g_top_manager = NULL;
|
| 15 |
|
| 16 | AtExitManager::AtExitManager() : next_manager_(NULL) {
|
| 17 | DCHECK(!g_top_manager);
|
| 18 | g_top_manager = this;
|
| 19 | }
|
| 20 |
|
| 21 | AtExitManager::AtExitManager(bool shadow) : next_manager_(g_top_manager) {
|
| 22 | DCHECK(shadow || !g_top_manager);
|
| 23 | g_top_manager = this;
|
| 24 | }
|
| 25 |
|
| 26 | AtExitManager::~AtExitManager() {
|
| 27 | if (!g_top_manager) {
|
| 28 | NOTREACHED() << "Tried to ~AtExitManager without an AtExitManager";
|
| 29 | return;
|
| 30 | }
|
| 31 | DCHECK(g_top_manager == this);
|
| 32 |
|
| 33 | ProcessCallbacksNow();
|
| 34 | g_top_manager = next_manager_;
|
| 35 | }
|
| 36 |
|
| 37 | // static
|
| 38 | void AtExitManager::RegisterCallback(AtExitCallbackType func) {
|
| 39 | if (!g_top_manager) {
|
| 40 | NOTREACHED() << "Tried to RegisterCallback without an AtExitManager";
|
| 41 | return;
|
| 42 | }
|
| 43 |
|
| 44 | AutoLock lock(g_top_manager->lock_);
|
| 45 | g_top_manager->stack_.push(func);
|
| 46 | }
|
| 47 |
|
| 48 | // static
|
| 49 | void AtExitManager::ProcessCallbacksNow() {
|
| 50 | if (!g_top_manager) {
|
| 51 | NOTREACHED() << "Tried to ProcessCallbacksNow without an AtExitManager";
|
| 52 | return;
|
| 53 | }
|
| 54 |
|
| 55 | AutoLock lock(g_top_manager->lock_);
|
| 56 |
|
| 57 | while (!g_top_manager->stack_.empty()) {
|
| 58 | AtExitCallbackType func = g_top_manager->stack_.top();
|
| 59 | g_top_manager->stack_.pop();
|
| 60 | if (func)
|
| 61 | func();
|
| 62 | }
|
| 63 | }
|
| 64 |
|
| 65 | } // namespace base
|
| 66 |
|