Fix performance regression when running clang tools.

Brings tool start time for a large synthetic test case down from (on my
machine) 4 seconds to 0.5 seconds.

llvm-svn: 247018
This commit is contained in:
Manuel Klimek
2015-09-08 15:14:06 +00:00
parent 2da1437d62
commit 3ecd8c0aab
3 changed files with 42 additions and 28 deletions

View File

@@ -214,15 +214,26 @@ JSONCompilationDatabase::getAllCompileCommands() const {
return Commands;
}
static std::vector<std::string>
nodeToCommandLine(const std::vector<llvm::yaml::ScalarNode *> &Nodes) {
SmallString<1024> Storage;
if (Nodes.size() == 1) {
return unescapeCommandLine(Nodes[0]->getValue(Storage));
}
std::vector<std::string> Arguments;
for (auto *Node : Nodes) {
Arguments.push_back(Node->getValue(Storage));
}
return Arguments;
}
void JSONCompilationDatabase::getCommands(
ArrayRef<CompileCommandRef> CommandsRef,
std::vector<CompileCommand> &Commands) const {
ArrayRef<CompileCommandRef> CommandsRef,
std::vector<CompileCommand> &Commands) const {
for (int I = 0, E = CommandsRef.size(); I != E; ++I) {
SmallString<8> DirectoryStorage;
SmallString<1024> CommandStorage;
Commands.emplace_back(
CommandsRef[I].first->getValue(DirectoryStorage),
CommandsRef[I].second);
Commands.emplace_back(CommandsRef[I].first->getValue(DirectoryStorage),
nodeToCommandLine(CommandsRef[I].second));
}
}
@@ -249,11 +260,8 @@ bool JSONCompilationDatabase::parse(std::string &ErrorMessage) {
return false;
}
llvm::yaml::ScalarNode *Directory = nullptr;
std::vector<std::string> Arguments;
std::vector<std::string> Command;
llvm::Optional<std::vector<llvm::yaml::ScalarNode *>> Command;
llvm::yaml::ScalarNode *File = nullptr;
bool ArgumentsFound = false;
bool CommandFound = false;
for (auto& NextKeyValue : *Object) {
llvm::yaml::ScalarNode *KeyString =
dyn_cast<llvm::yaml::ScalarNode>(NextKeyValue.getKey());
@@ -282,18 +290,18 @@ bool JSONCompilationDatabase::parse(std::string &ErrorMessage) {
if (KeyValue == "directory") {
Directory = ValueString;
} else if (KeyValue == "arguments") {
for (auto& NextArgument : *SequenceString) {
SmallString<128> CommandStorage;
auto ValueString = dyn_cast<llvm::yaml::ScalarNode>(&NextArgument);
Arguments.push_back(ValueString->getValue(CommandStorage));
Command = std::vector<llvm::yaml::ScalarNode *>();
for (auto &Argument : *SequenceString) {
auto Scalar = dyn_cast<llvm::yaml::ScalarNode>(&Argument);
if (!Scalar) {
ErrorMessage = "Only strings are allowed in 'arguments'.";
return false;
}
Command->push_back(Scalar);
}
ArgumentsFound = true;
} else if (KeyValue == "command") {
SmallString<1024> CommandStorage;
// FIXME: Escape correctly:
Command = unescapeCommandLine(ValueString->getValue(CommandStorage));
CommandFound = true;
if (!Command)
Command = std::vector<llvm::yaml::ScalarNode *>(1, ValueString);
} else if (KeyValue == "file") {
File = ValueString;
} else {
@@ -306,7 +314,7 @@ bool JSONCompilationDatabase::parse(std::string &ErrorMessage) {
ErrorMessage = "Missing key: \"file\".";
return false;
}
if (!ArgumentsFound && !CommandFound) {
if (!Command) {
ErrorMessage = "Missing key: \"command\" or \"arguments\".";
return false;
}
@@ -327,7 +335,7 @@ bool JSONCompilationDatabase::parse(std::string &ErrorMessage) {
llvm::sys::path::native(FileName, NativeFilePath);
}
IndexByFile[NativeFilePath].push_back(
CompileCommandRef(Directory, ArgumentsFound ? Arguments : Command));
CompileCommandRef(Directory, *Command));
MatchTrie.insert(NativeFilePath);
}
return true;