forked from OSchip/llvm-project
34 lines
923 B
C++
34 lines
923 B
C++
//===--- 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
|