40 lines · cpp
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-exception6//7//===----------------------------------------------------------------------===//8//9// Implement a simple subcommand registry.10//11//===----------------------------------------------------------------------===//12#include "xray-registry.h"13 14#include <unordered_map>15 16using namespace llvm;17using namespace xray;18 19using HandlerType = std::function<Error()>;20 21static std::unordered_map<cl::SubCommand *, HandlerType> &getCommands() {22 static std::unordered_map<cl::SubCommand *, HandlerType> Commands;23 return Commands;24}25 26CommandRegistration::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 34HandlerType xray::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