1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
|
// Copyright 2016 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include <memory>
#include <utility>
#include "base/run_loop.h"
#include "base/task/cancelable_task_tracker.h"
#include "chrome/browser/history/history_service_factory.h"
#include "components/history/core/browser/history_db_task.h"
#include "components/history/core/browser/history_service.h"
#include "content/public/test/test_utils.h"
namespace {
// Note: WaitableEvent is not used for synchronization between the main thread
// and history backend thread because the history subsystem posts tasks back
// to the main thread. Had we tried to Signal an event in such a task
// and Wait for it on the main thread, the task would not run at all because
// the main thread would be blocked on the Wait call, resulting in a deadlock.
// A task to be scheduled on the history backend thread.
// Notifies the main thread after all history backend thread tasks have run.
class WaitForHistoryTask : public history::HistoryDBTask {
public:
explicit WaitForHistoryTask(base::OnceClosure quit_closure)
: quit_closure_(std::move(quit_closure)) {}
WaitForHistoryTask(const WaitForHistoryTask&) = delete;
WaitForHistoryTask& operator=(const WaitForHistoryTask&) = delete;
bool RunOnDBThread(history::HistoryBackend* backend,
history::HistoryDatabase* db) override {
return true;
}
void DoneRunOnMainThread() override { std::move(quit_closure_).Run(); }
private:
base::OnceClosure quit_closure_;
~WaitForHistoryTask() override = default;
};
} // namespace
void WaitForHistoryBackendToRun(Profile* profile) {
base::RunLoop loop;
base::CancelableTaskTracker task_tracker;
std::unique_ptr<history::HistoryDBTask> task(
new WaitForHistoryTask(loop.QuitWhenIdleClosure()));
history::HistoryService* history = HistoryServiceFactory::GetForProfile(
profile, ServiceAccessType::EXPLICIT_ACCESS);
history->ScheduleDBTask(FROM_HERE, std::move(task), &task_tracker);
loop.Run();
}
|
