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
| //===--- Cancellation.cpp -----------------------------------------*-C++-*-===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//
#include "Cancellation.h"
#include <atomic>
namespace clang {
namespace clangd {
char CancelledError::ID = 0;
static Key<std::shared_ptr<std::atomic<bool>>> FlagKey;
std::pair<Context, Canceler> cancelableTask() {
auto Flag = std::make_shared<std::atomic<bool>>();
return {
Context::current().derive(FlagKey, Flag),
[Flag] { *Flag = true; },
};
}
bool isCancelled(const Context &Ctx) {
if (auto *Flag = Ctx.get(FlagKey))
return **Flag;
return false; // Not in scope of a task.
}
} // namespace clangd
} // namespace clang
|