Chromium Code Reviews| OLD | NEW |
|---|---|
| 1 // Copyright 2014 The Chromium Authors. All rights reserved. | 1 // Copyright 2014 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/metrics/user_metrics.h" | 5 #include "base/metrics/user_metrics.h" |
| 6 | 6 |
| 7 #include <stddef.h> | 7 #include <stddef.h> |
| 8 | 8 |
| 9 #include <vector> | 9 #include <vector> |
| 10 | 10 |
| 11 #include "base/bind.h" | |
| 11 #include "base/lazy_instance.h" | 12 #include "base/lazy_instance.h" |
| 13 #include "base/location.h" | |
| 12 #include "base/macros.h" | 14 #include "base/macros.h" |
| 13 #include "base/threading/thread_checker.h" | 15 #include "base/threading/thread_checker.h" |
| 14 | 16 |
| 15 namespace base { | 17 namespace base { |
| 16 namespace { | 18 namespace { |
| 17 | 19 |
| 18 // A helper class for tracking callbacks and ensuring thread-safety. | 20 base::LazyInstance<std::vector<ActionCallback>> g_callbacks = |
|
Lei Zhang
2016/04/18 21:34:13
nit: you don't need base:: inside namespace base.
beaudoin
2016/04/19 16:57:27
Done.
| |
| 19 class Callbacks { | 21 LAZY_INSTANCE_INITIALIZER; |
| 20 public: | 22 base::LazyInstance<scoped_refptr<base::SingleThreadTaskRunner>> g_task_runner = |
| 21 Callbacks() {} | 23 LAZY_INSTANCE_INITIALIZER; |
| 22 | |
| 23 // Records the |action|. | |
| 24 void Record(const std::string& action) { | |
| 25 DCHECK(thread_checker_.CalledOnValidThread()); | |
| 26 for (size_t i = 0; i < callbacks_.size(); ++i) { | |
| 27 callbacks_[i].Run(action); | |
| 28 } | |
| 29 } | |
| 30 | |
| 31 // Adds |callback| to the list of |callbacks_|. | |
| 32 void AddCallback(const ActionCallback& callback) { | |
| 33 DCHECK(thread_checker_.CalledOnValidThread()); | |
| 34 callbacks_.push_back(callback); | |
| 35 } | |
| 36 | |
| 37 // Removes the first instance of |callback| from the list of |callbacks_|, if | |
| 38 // there is one. | |
| 39 void RemoveCallback(const ActionCallback& callback) { | |
| 40 DCHECK(thread_checker_.CalledOnValidThread()); | |
| 41 for (size_t i = 0; i < callbacks_.size(); ++i) { | |
| 42 if (callbacks_[i].Equals(callback)) { | |
| 43 callbacks_.erase(callbacks_.begin() + i); | |
| 44 return; | |
| 45 } | |
| 46 } | |
| 47 } | |
| 48 | |
| 49 private: | |
| 50 base::ThreadChecker thread_checker_; | |
| 51 std::vector<ActionCallback> callbacks_; | |
| 52 | |
| 53 DISALLOW_COPY_AND_ASSIGN(Callbacks); | |
| 54 }; | |
| 55 | |
| 56 base::LazyInstance<Callbacks> g_callbacks = LAZY_INSTANCE_INITIALIZER; | |
| 57 | 24 |
| 58 } // namespace | 25 } // namespace |
| 59 | 26 |
| 60 void RecordAction(const UserMetricsAction& action) { | 27 void RecordAction(const UserMetricsAction& action) { |
| 61 g_callbacks.Get().Record(action.str_); | 28 RecordComputedAction(action.str_); |
| 62 } | 29 } |
| 63 | 30 |
| 64 void RecordComputedAction(const std::string& action) { | 31 void RecordComputedAction(const std::string& action) { |
| 65 g_callbacks.Get().Record(action); | 32 if (!g_task_runner.Get()) { |
| 33 DCHECK(g_callbacks.Get().empty()); | |
| 34 return; | |
| 35 } | |
| 36 | |
| 37 if (!g_task_runner.Get()->BelongsToCurrentThread()) { | |
| 38 g_task_runner.Get()->PostTask(FROM_HERE, | |
| 39 base::Bind(&RecordComputedAction, action)); | |
| 40 return; | |
| 41 } | |
| 42 | |
| 43 for (const ActionCallback& callback : g_callbacks.Get()) { | |
| 44 callback.Run(action); | |
| 45 } | |
| 66 } | 46 } |
| 67 | 47 |
| 68 void AddActionCallback(const ActionCallback& callback) { | 48 void AddActionCallback(const ActionCallback& callback) { |
| 69 g_callbacks.Get().AddCallback(callback); | 49 // Only allow adding a callback if the task runner is set. |
| 50 DCHECK(g_task_runner.Get()); | |
| 51 DCHECK(g_task_runner.Get()->BelongsToCurrentThread()); | |
| 52 g_callbacks.Get().push_back(callback); | |
| 70 } | 53 } |
| 71 | 54 |
| 72 void RemoveActionCallback(const ActionCallback& callback) { | 55 void RemoveActionCallback(const ActionCallback& callback) { |
| 73 g_callbacks.Get().RemoveCallback(callback); | 56 DCHECK(g_task_runner.Get()); |
| 57 DCHECK(g_task_runner.Get()->BelongsToCurrentThread()); | |
| 58 std::vector<ActionCallback>* callbacks = &g_callbacks.Get(); | |
|
Lei Zhang
2016/04/18 21:34:13
Make |callbacks| a reference, or use g_callbacks.P
beaudoin
2016/04/19 16:57:27
I used a reference first, but asvitkine@ mentioned
Lei Zhang
2016/04/19 17:52:00
I'm ok with them. Sometimes they make code easier
beaudoin
2016/04/19 18:36:22
Acknowledged.
| |
| 59 for (size_t i = 0; i < callbacks->size(); ++i) { | |
| 60 if ((*callbacks)[i].Equals(callback)) { | |
| 61 callbacks->erase(callbacks->begin() + i); | |
| 62 return; | |
| 63 } | |
| 64 } | |
| 65 } | |
| 74 | 66 |
| 67 void SetRecordActionTaskRunner( | |
| 68 scoped_refptr<base::SingleThreadTaskRunner> task_runner) { | |
| 69 DCHECK(task_runner->BelongsToCurrentThread()); | |
| 70 DCHECK(!g_task_runner.Get() || g_task_runner.Get()->BelongsToCurrentThread()); | |
| 71 g_task_runner.Get() = task_runner; | |
| 75 } | 72 } |
| 76 | 73 |
| 77 } // namespace base | 74 } // namespace base |
| OLD | NEW |