OLD | NEW |
---|---|
1 // Copyright (c) 2012 The Chromium Authors. All rights reserved. | 1 // Copyright (c) 2012 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 #ifndef BASE_SCOPED_OBSERVER_H_ | 5 #ifndef BASE_SCOPED_OBSERVER_H_ |
6 #define BASE_SCOPED_OBSERVER_H_ | 6 #define BASE_SCOPED_OBSERVER_H_ |
7 | 7 |
8 #include <algorithm> | 8 #include <algorithm> |
9 #include <vector> | 9 #include <vector> |
10 | 10 |
11 #include "base/basictypes.h" | 11 #include "base/basictypes.h" |
12 | 12 |
13 // ScopedObserver is used to keep track of the set of sources an object has | 13 // ScopedObserver is used to keep track of the set of sources an object has |
14 // attached itself to as an observer. When ScopedObserver is destroyed it | 14 // attached itself to as an observer. When ScopedObserver is destroyed it |
15 // removes the object as an observer from all sources it has been added to. | 15 // removes the object as an observer from all sources it has been added to. |
16 template <class Source, class Observer> | 16 template <class Source, class Observer> |
17 class ScopedObserver { | 17 class ScopedObserver { |
18 public: | 18 public: |
19 explicit ScopedObserver(Observer* observer) : observer_(observer) {} | 19 explicit ScopedObserver(Observer* observer) : observer_(observer) {} |
20 | 20 |
21 ~ScopedObserver() { | 21 ~ScopedObserver() { |
22 for (size_t i = 0; i < sources_.size(); ++i) | 22 RemoveAll(); |
23 sources_[i]->RemoveObserver(observer_); | |
24 } | 23 } |
25 | 24 |
26 // Adds the object passed to the constructor as an observer on |source|. | 25 // Adds the object passed to the constructor as an observer on |source|. |
27 void Add(Source* source) { | 26 void Add(Source* source) { |
28 sources_.push_back(source); | 27 sources_.push_back(source); |
29 source->AddObserver(observer_); | 28 source->AddObserver(observer_); |
30 } | 29 } |
31 | 30 |
32 // Remove the object passed to the constructor as an observer from |source|. | 31 // Remove the object passed to the constructor as an observer from |source|. |
33 void Remove(Source* source) { | 32 void Remove(Source* source) { |
34 sources_.erase(std::find(sources_.begin(), sources_.end(), source)); | 33 sources_.erase(std::find(sources_.begin(), sources_.end(), source)); |
35 source->RemoveObserver(observer_); | 34 source->RemoveObserver(observer_); |
36 } | 35 } |
37 | 36 |
37 void RemoveAll() { | |
38 for (size_t i = 0; i < sources_.size(); ++i) | |
39 sources_[i]->RemoveObserver(observer_); | |
Nico
2014/03/28 03:34:12
and you don't think it's better to do a sources_.c
Lei Zhang
2014/03/28 03:40:55
Good catch. Thanks.
| |
40 } | |
41 | |
38 bool IsObserving(Source* source) const { | 42 bool IsObserving(Source* source) const { |
39 for (size_t i = 0; i < sources_.size(); ++i) { | 43 return std::find(sources_.begin(), sources_.end(), source) != |
40 if (sources_[i] == source) | 44 sources_.end(); |
41 return true; | |
42 } | |
43 return false; | |
44 } | 45 } |
45 | 46 |
46 private: | 47 private: |
47 Observer* observer_; | 48 Observer* observer_; |
48 | 49 |
49 std::vector<Source*> sources_; | 50 std::vector<Source*> sources_; |
50 | 51 |
51 DISALLOW_COPY_AND_ASSIGN(ScopedObserver); | 52 DISALLOW_COPY_AND_ASSIGN(ScopedObserver); |
52 }; | 53 }; |
53 | 54 |
54 #endif // BASE_SCOPED_OBSERVER_H_ | 55 #endif // BASE_SCOPED_OBSERVER_H_ |
OLD | NEW |