blob: 1111e9bf4b72fd842444b5a939cd3115dc9c0aa0 (
plain)
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
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
|
// Copyright 2013 The Chromium Authors. All rights reserved.
// Use of this source code is governed by a BSD-style license that can be
// found in the LICENSE file.
#include "cc/trees/blocking_task_runner.h"
#include <utility>
#include "base/callback.h"
#include "base/logging.h"
#include "base/message_loop/message_loop_proxy.h"
namespace cc {
// static
scoped_ptr<BlockingTaskRunner> BlockingTaskRunner::Create(
scoped_refptr<base::SingleThreadTaskRunner> task_runner) {
return make_scoped_ptr(new BlockingTaskRunner(task_runner));
}
BlockingTaskRunner::BlockingTaskRunner(
scoped_refptr<base::SingleThreadTaskRunner> task_runner)
: thread_id_(base::PlatformThread::CurrentId()),
task_runner_(task_runner),
capture_(0) {
}
BlockingTaskRunner::~BlockingTaskRunner() {}
bool BlockingTaskRunner::BelongsToCurrentThread() {
return base::PlatformThread::CurrentId() == thread_id_;
}
bool BlockingTaskRunner::PostTask(const tracked_objects::Location& from_here,
const base::Closure& task) {
base::AutoLock lock(lock_);
DCHECK(task_runner_.get() || capture_);
if (!capture_)
return task_runner_->PostTask(from_here, task);
captured_tasks_.push_back(task);
return true;
}
void BlockingTaskRunner::SetCapture(bool capture) {
DCHECK(BelongsToCurrentThread());
std::vector<base::Closure> tasks;
{
base::AutoLock lock(lock_);
capture_ += capture ? 1 : -1;
DCHECK_GE(capture_, 0);
if (capture_)
return;
// We're done capturing, so grab all the captured tasks and run them.
tasks.swap(captured_tasks_);
}
for (size_t i = 0; i < tasks.size(); ++i)
tasks[i].Run();
}
BlockingTaskRunner::CapturePostTasks::CapturePostTasks(
BlockingTaskRunner* blocking_runner)
: blocking_runner_(blocking_runner) {
blocking_runner_->SetCapture(true);
}
BlockingTaskRunner::CapturePostTasks::~CapturePostTasks() {
blocking_runner_->SetCapture(false);
}
} // namespace cc
|
