12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485 |
- #ifndef BASE_SCOPED_OBSERVER_H_
- #define BASE_SCOPED_OBSERVER_H_
- #include <stddef.h>
- #include <algorithm>
- #include <vector>
- #include "base/check.h"
- #include "base/stl_util.h"
- template <class Source,
- class Observer,
- void (Source::*AddObsFn)(Observer*) = &Source::AddObserver,
- void (Source::*RemoveObsFn)(Observer*) = &Source::RemoveObserver>
- class ScopedObserver {
- public:
- explicit ScopedObserver(Observer* observer) : observer_(observer) {}
- ScopedObserver(const ScopedObserver&) = delete;
- ScopedObserver& operator=(const ScopedObserver&) = delete;
- ~ScopedObserver() {
- RemoveAll();
- }
-
- void Add(Source* source) {
- sources_.push_back(source);
- (source->*AddObsFn)(observer_);
- }
-
- void Remove(Source* source) {
- auto it = std::find(sources_.begin(), sources_.end(), source);
- DCHECK(it != sources_.end());
- sources_.erase(it);
- (source->*RemoveObsFn)(observer_);
- }
- void RemoveAll() {
- for (size_t i = 0; i < sources_.size(); ++i)
- (sources_[i]->*RemoveObsFn)(observer_);
- sources_.clear();
- }
- bool IsObserving(Source* source) const {
- return base::Contains(sources_, source);
- }
- bool IsObservingSources() const { return !sources_.empty(); }
- size_t GetSourcesCount() const { return sources_.size(); }
- private:
- Observer* observer_;
- std::vector<Source*> sources_;
- };
- #endif
|