OLD | NEW |
1 // Copyright 2016 The Chromium Authors. All rights reserved. | 1 // Copyright 2016 The Chromium Authors. All rights reserved. |
2 // Use of this source code is governed by a BSD-style license that can be | 2 // Use of this source code is governed by a BSD-style license that can be |
3 // found in the LICENSE file. | 3 // found in the LICENSE file. |
4 | 4 |
5 #include "base/sequence_token.h" | 5 #include "base/sequence_token.h" |
6 | 6 |
7 #include "base/atomic_sequence_num.h" | 7 #include "base/atomic_sequence_num.h" |
8 #include "base/lazy_instance.h" | 8 #include "base/lazy_instance.h" |
9 #include "base/logging.h" | 9 #include "base/logging.h" |
10 #include "base/threading/thread_local.h" | 10 #include "base/threading/thread_local.h" |
11 | 11 |
12 namespace base { | 12 namespace base { |
13 | 13 |
14 namespace { | 14 namespace { |
15 | 15 |
16 base::StaticAtomicSequenceNumber g_sequence_token_generator; | 16 base::StaticAtomicSequenceNumber g_sequence_token_generator; |
17 | 17 |
18 base::StaticAtomicSequenceNumber g_task_token_generator; | 18 base::StaticAtomicSequenceNumber g_task_token_generator; |
19 | 19 |
20 LazyInstance<ThreadLocalPointer<const SequenceToken>>::Leaky | 20 LazyInstance<ThreadLocalPointer<const SequenceToken>>::Leaky |
21 tls_current_sequence_token = LAZY_INSTANCE_INITIALIZER; | 21 tls_current_sequence_token = LAZY_INSTANCE_INITIALIZER; |
22 | 22 |
23 LazyInstance<ThreadLocalPointer<const TaskToken>>::Leaky | 23 LazyInstance<ThreadLocalPointer<const TaskToken>>::Leaky |
24 tls_current_task_token = LAZY_INSTANCE_INITIALIZER; | 24 tls_current_task_token = LAZY_INSTANCE_INITIALIZER; |
25 | 25 |
26 } // namespace | 26 } // namespace |
27 | 27 |
28 bool SequenceToken::operator==(const SequenceToken& other) const { | 28 bool SequenceToken::operator==(const SequenceToken& other) const { |
29 return token_ == other.token_ && token_ != kInvalidSequenceToken; | 29 return token_ == other.token_ && IsValid(); |
30 } | 30 } |
31 | 31 |
32 bool SequenceToken::operator!=(const SequenceToken& other) const { | 32 bool SequenceToken::operator!=(const SequenceToken& other) const { |
33 return !(*this == other); | 33 return !(*this == other); |
34 } | 34 } |
35 | 35 |
36 bool SequenceToken::IsValid() const { | 36 bool SequenceToken::IsValid() const { |
37 return token_ != kInvalidSequenceToken; | 37 return token_ != kInvalidSequenceToken; |
38 } | 38 } |
39 | 39 |
40 SequenceToken SequenceToken::Create() { | 40 SequenceToken SequenceToken::Create() { |
41 return SequenceToken(g_sequence_token_generator.GetNext()); | 41 return SequenceToken(g_sequence_token_generator.GetNext()); |
42 } | 42 } |
43 | 43 |
44 SequenceToken SequenceToken::GetForCurrentThread() { | 44 SequenceToken SequenceToken::GetForCurrentThread() { |
45 const SequenceToken* current_sequence_token = | 45 const SequenceToken* current_sequence_token = |
46 tls_current_sequence_token.Get().Get(); | 46 tls_current_sequence_token.Get().Get(); |
47 return current_sequence_token ? *current_sequence_token : SequenceToken(); | 47 return current_sequence_token ? *current_sequence_token : SequenceToken(); |
48 } | 48 } |
49 | 49 |
50 bool TaskToken::operator==(const TaskToken& other) const { | 50 bool TaskToken::operator==(const TaskToken& other) const { |
51 return token_ == other.token_ && token_ != kInvalidTaskToken; | 51 return token_ == other.token_ && IsValid(); |
52 } | 52 } |
53 | 53 |
54 bool TaskToken::operator!=(const TaskToken& other) const { | 54 bool TaskToken::operator!=(const TaskToken& other) const { |
55 return !(*this == other); | 55 return !(*this == other); |
56 } | 56 } |
57 | 57 |
58 bool TaskToken::IsValid() const { | 58 bool TaskToken::IsValid() const { |
59 return token_ != kInvalidTaskToken; | 59 return token_ != kInvalidTaskToken; |
60 } | 60 } |
61 | 61 |
(...skipping 17 matching lines...) Expand all Loading... |
79 | 79 |
80 ScopedSetSequenceTokenForCurrentThread:: | 80 ScopedSetSequenceTokenForCurrentThread:: |
81 ~ScopedSetSequenceTokenForCurrentThread() { | 81 ~ScopedSetSequenceTokenForCurrentThread() { |
82 DCHECK_EQ(tls_current_sequence_token.Get().Get(), &sequence_token_); | 82 DCHECK_EQ(tls_current_sequence_token.Get().Get(), &sequence_token_); |
83 DCHECK_EQ(tls_current_task_token.Get().Get(), &task_token_); | 83 DCHECK_EQ(tls_current_task_token.Get().Get(), &task_token_); |
84 tls_current_sequence_token.Get().Set(nullptr); | 84 tls_current_sequence_token.Get().Set(nullptr); |
85 tls_current_task_token.Get().Set(nullptr); | 85 tls_current_task_token.Get().Set(nullptr); |
86 } | 86 } |
87 | 87 |
88 } // namespace base | 88 } // namespace base |
OLD | NEW |