1 //===- xray-registry.cpp: Implement a command registry. -------------------===//
2 //
3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4 // See https://llvm.org/LICENSE.txt for license information.
5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6 //
7 //===----------------------------------------------------------------------===//
8 //
9 // Implement a simple subcommand registry.
10 //
11 //===----------------------------------------------------------------------===//
12 #include "xray-registry.h"
13 
14 #include "llvm/Support/ManagedStatic.h"
15 #include <unordered_map>
16 
17 namespace llvm {
18 namespace xray {
19 
20 using HandlerType = std::function<Error()>;
21 
22 ManagedStatic<std::unordered_map<cl::SubCommand *, HandlerType>> Commands;
23 
24 CommandRegistration::CommandRegistration(cl::SubCommand *SC,
25                                          HandlerType Command) {
26   assert(Commands->count(SC) == 0 &&
27          "Attempting to overwrite a command handler");
28   assert(Command && "Attempting to register an empty std::function<Error()>");
29   (*Commands)[SC] = Command;
30 }
31 
32 HandlerType dispatch(cl::SubCommand *SC) {
33   auto It = Commands->find(SC);
34   assert(It != Commands->end() &&
35          "Attempting to dispatch on un-registered SubCommand.");
36   return It->second;
37 }
38 
39 } // namespace xray
40 } // namespace llvm
41