|
1 // Copyright (c) 2011 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/lazy_instance.h" |
|
6 |
|
7 #include "base/at_exit.h" |
|
8 #include "base/atomicops.h" |
|
9 #include "base/basictypes.h" |
|
10 #include "base/threading/platform_thread.h" |
|
11 #include "base/third_party/dynamic_annotations/dynamic_annotations.h" |
|
12 |
|
13 namespace base { |
|
14 namespace internal { |
|
15 |
|
16 // TODO(joth): This function could be shared with Singleton, in place of its |
|
17 // WaitForInstance() call. |
|
18 bool NeedsLazyInstance(subtle::AtomicWord* state) { |
|
19 // Try to create the instance, if we're the first, will go from 0 to |
|
20 // kLazyInstanceStateCreating, otherwise we've already been beaten here. |
|
21 // The memory access has no memory ordering as state 0 and |
|
22 // kLazyInstanceStateCreating have no associated data (memory barriers are |
|
23 // all about ordering of memory accesses to *associated* data). |
|
24 if (subtle::NoBarrier_CompareAndSwap(state, 0, |
|
25 kLazyInstanceStateCreating) == 0) |
|
26 // Caller must create instance |
|
27 return true; |
|
28 |
|
29 // It's either in the process of being created, or already created. Spin. |
|
30 // The load has acquire memory ordering as a thread which sees |
|
31 // state_ == STATE_CREATED needs to acquire visibility over |
|
32 // the associated data (buf_). Pairing Release_Store is in |
|
33 // CompleteLazyInstance(). |
|
34 while (subtle::Acquire_Load(state) == kLazyInstanceStateCreating) { |
|
35 PlatformThread::YieldCurrentThread(); |
|
36 } |
|
37 // Someone else created the instance. |
|
38 return false; |
|
39 } |
|
40 |
|
41 void CompleteLazyInstance(subtle::AtomicWord* state, |
|
42 subtle::AtomicWord new_instance, |
|
43 void* lazy_instance, |
|
44 void (*dtor)(void*)) { |
|
45 // See the comment to the corresponding HAPPENS_AFTER in Pointer(). |
|
46 ANNOTATE_HAPPENS_BEFORE(state); |
|
47 |
|
48 // Instance is created, go from CREATING to CREATED. |
|
49 // Releases visibility over private_buf_ to readers. Pairing Acquire_Load's |
|
50 // are in NeedsInstance() and Pointer(). |
|
51 subtle::Release_Store(state, new_instance); |
|
52 |
|
53 // Make sure that the lazily instantiated object will get destroyed at exit. |
|
54 if (dtor) |
|
55 AtExitManager::RegisterCallback(dtor, lazy_instance); |
|
56 } |
|
57 |
|
58 } // namespace internal |
|
59 } // namespace base |