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 <unordered_map>
15 
16 namespace llvm {
17 namespace xray {
18 
19 using HandlerType = std::function<Error()>;
20 
getCommands()21 static std::unordered_map<cl::SubCommand *, HandlerType> &getCommands() {
22   static std::unordered_map<cl::SubCommand *, HandlerType> Commands;
23   return Commands;
24 }
25 
CommandRegistration(cl::SubCommand * SC,HandlerType Command)26 CommandRegistration::CommandRegistration(cl::SubCommand *SC,
27                                          HandlerType Command) {
28   assert(getCommands().count(SC) == 0 &&
29          "Attempting to overwrite a command handler");
30   assert(Command && "Attempting to register an empty std::function<Error()>");
31   getCommands()[SC] = Command;
32 }
33 
dispatch(cl::SubCommand * SC)34 HandlerType dispatch(cl::SubCommand *SC) {
35   auto It = getCommands().find(SC);
36   assert(It != getCommands().end() &&
37          "Attempting to dispatch on un-registered SubCommand.");
38   return It->second;
39 }
40 
41 } // namespace xray
42 } // namespace llvm
43