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
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
|
/*
* Copyright (C) 2021 Apple Inc. All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
* 1. Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
* 2. Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
*
* THIS SOFTWARE IS PROVIDED BY APPLE INC. AND ITS CONTRIBUTORS ``AS IS''
* AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO,
* THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
* PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL APPLE INC. OR ITS CONTRIBUTORS
* BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
* CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
* SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
* INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
* CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
* ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF
* THE POSSIBILITY OF SUCH DAMAGE.
*/
#pragma once
#include <wtf/Function.h>
#include <wtf/RefCountedAndCanMakeWeakPtr.h>
#include <wtf/UniqueRef.h>
#include <wtf/WeakPtr.h>
namespace WTF {
class CancellableTask;
class TaskCancellationGroupImpl final : public RefCountedAndCanMakeWeakPtr<TaskCancellationGroupImpl> {
public:
static Ref<TaskCancellationGroupImpl> create()
{
return adoptRef(*new TaskCancellationGroupImpl);
}
void cancel() { weakPtrFactory().revokeAll(); }
bool hasPendingTask() const { return weakPtrFactory().weakPtrCount(); }
private:
TaskCancellationGroupImpl() = default;
};
class TaskCancellationGroupHandle {
public:
bool isCancelled() const { return !m_impl; }
void clear() { m_impl = nullptr; }
private:
friend class TaskCancellationGroup;
explicit TaskCancellationGroupHandle(TaskCancellationGroupImpl& impl)
: m_impl(impl)
{
}
WeakPtr<TaskCancellationGroupImpl> m_impl;
};
class TaskCancellationGroup {
public:
TaskCancellationGroup()
: m_impl(TaskCancellationGroupImpl::create())
{
}
void cancel() { m_impl->cancel(); }
bool hasPendingTask() const { return m_impl->hasPendingTask(); }
private:
friend class CancellableTask;
TaskCancellationGroupHandle createHandle() { return TaskCancellationGroupHandle { m_impl }; }
Ref<TaskCancellationGroupImpl> m_impl;
};
class CancellableTask {
public:
CancellableTask(TaskCancellationGroup&, Function<void()>&&);
void operator()();
private:
TaskCancellationGroupHandle m_cancellationGroup;
Function<void()> m_task;
};
inline CancellableTask::CancellableTask(TaskCancellationGroup& cancellationGroup, Function<void()>&& task)
: m_cancellationGroup(cancellationGroup.createHandle())
, m_task(WTFMove(task))
{ }
inline void CancellableTask::operator()()
{
if (m_cancellationGroup.isCancelled())
return;
m_cancellationGroup.clear();
m_task();
}
} // namespace WTF
using WTF::CancellableTask;
using WTF::TaskCancellationGroup;
|
