summaryrefslogtreecommitdiffstats
path: root/clangd/Cancellation.cpp
blob: cc1c11cebb9b3fb05c4bc6df861f228c47b124b5 (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
//===--- Cancellation.cpp -----------------------------------------*-C++-*-===//
//
//                     The LLVM Compiler Infrastructure
//
// This file is distributed under the University of Illinois Open Source
// License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//

#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