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
|
// Copyright 2018 The Chromium Authors
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "third_party/blink/renderer/platform/testing/scoped_scheduler_overrider.h"
#include "base/memory/raw_ptr.h"
#include "base/task/single_thread_task_runner.h"
#include "third_party/blink/renderer/platform/wtf/wtf.h"
namespace blink {
namespace {
class ThreadWithCustomScheduler : public MainThread {
public:
explicit ThreadWithCustomScheduler(
ThreadScheduler* scheduler,
scoped_refptr<base::SingleThreadTaskRunner> task_runner)
: scheduler_(scheduler), task_runner_(std::move(task_runner)) {}
~ThreadWithCustomScheduler() override {}
ThreadScheduler* Scheduler() override { return scheduler_; }
scoped_refptr<base::SingleThreadTaskRunner> GetTaskRunner(
MainThreadTaskRunnerRestricted) const override {
return task_runner_;
}
private:
raw_ptr<ThreadScheduler> scheduler_;
scoped_refptr<base::SingleThreadTaskRunner> task_runner_;
};
} // namespace
ScopedSchedulerOverrider::ScopedSchedulerOverrider(
ThreadScheduler* scheduler,
scoped_refptr<base::SingleThreadTaskRunner> task_runner)
: main_thread_overrider_(std::make_unique<ThreadWithCustomScheduler>(
scheduler,
std::move(task_runner))) {}
ScopedSchedulerOverrider::~ScopedSchedulerOverrider() {}
} // namespace blink
|
