forked from OSchip/llvm-project
PragmaNamespace::Handlers: Use unique_ptr to simplify memory management
The API actually passes and returns ownership too, but the usage uis complicated enough that I'm not going to unique_ptr-ify those add/remove calls.
This commit is contained in:
parent
957c5dd78b
commit
4bd5fbec4b
|
@ -96,11 +96,10 @@ public:
|
|||
class PragmaNamespace : public PragmaHandler {
|
||||
/// Handlers - This is a map of the handlers in this namespace with their name
|
||||
/// as key.
|
||||
llvm::StringMap<PragmaHandler *> Handlers;
|
||||
llvm::StringMap<std::unique_ptr<PragmaHandler>> Handlers;
|
||||
|
||||
public:
|
||||
explicit PragmaNamespace(StringRef Name) : PragmaHandler(Name) {}
|
||||
~PragmaNamespace() override;
|
||||
|
||||
/// FindHandler - Check to see if there is already a handler for the
|
||||
/// specified name. If not, return the handler for the null name if it
|
||||
|
|
|
@ -71,31 +71,36 @@ void EmptyPragmaHandler::HandlePragma(Preprocessor &PP,
|
|||
// PragmaNamespace Implementation.
|
||||
//===----------------------------------------------------------------------===//
|
||||
|
||||
PragmaNamespace::~PragmaNamespace() {
|
||||
llvm::DeleteContainerSeconds(Handlers);
|
||||
}
|
||||
|
||||
/// FindHandler - Check to see if there is already a handler for the
|
||||
/// specified name. If not, return the handler for the null identifier if it
|
||||
/// exists, otherwise return null. If IgnoreNull is true (the default) then
|
||||
/// the null handler isn't returned on failure to match.
|
||||
PragmaHandler *PragmaNamespace::FindHandler(StringRef Name,
|
||||
bool IgnoreNull) const {
|
||||
if (PragmaHandler *Handler = Handlers.lookup(Name))
|
||||
return Handler;
|
||||
return IgnoreNull ? nullptr : Handlers.lookup(StringRef());
|
||||
auto I = Handlers.find(Name);
|
||||
if (I != Handlers.end())
|
||||
return I->getValue().get();
|
||||
if (IgnoreNull)
|
||||
return nullptr;
|
||||
I = Handlers.find(StringRef());
|
||||
if (I != Handlers.end())
|
||||
return I->getValue().get();
|
||||
return nullptr;
|
||||
}
|
||||
|
||||
void PragmaNamespace::AddPragma(PragmaHandler *Handler) {
|
||||
assert(!Handlers.lookup(Handler->getName()) &&
|
||||
assert(!Handlers.count(Handler->getName()) &&
|
||||
"A handler with this name is already registered in this namespace");
|
||||
Handlers[Handler->getName()] = Handler;
|
||||
Handlers[Handler->getName()].reset(Handler);
|
||||
}
|
||||
|
||||
void PragmaNamespace::RemovePragmaHandler(PragmaHandler *Handler) {
|
||||
assert(Handlers.lookup(Handler->getName()) &&
|
||||
auto I = Handlers.find(Handler->getName());
|
||||
assert(I != Handlers.end() &&
|
||||
"Handler not registered in this namespace");
|
||||
Handlers.erase(Handler->getName());
|
||||
// Release ownership back to the caller.
|
||||
I->getValue().release();
|
||||
Handlers.erase(I);
|
||||
}
|
||||
|
||||
void PragmaNamespace::HandlePragma(Preprocessor &PP,
|
||||
|
|
Loading…
Reference in New Issue