averello/swift-mirror
1
0
Fork 0
mirror of https://github.com/apple/swift.git synced 2025-12-14 20:36:38 +01:00
Code Issues Packages Projects Releases Wiki Activity

Serialize symbol graphs for extended modules separately

Browse Source 
When a module extends a type from another module, serialize those symbols into
separated files dedicated to those extended modules. This makes it easier to
ingest and categorize those symbols under the extended module if desired.

rdar://58941718
This commit is contained in:
Ashley Garland
2020-02-06 17:26:15 -08:00
parent be68f864e0
commit 7190073a85
35 changed files with 254 additions and 61 deletions
Download Patch File Download Diff File Expand all files Collapse all files

46
lib/SymbolGraphGen/SymbolGraphGen.cpp
View File

@@ -20,6 +20,35 @@
using namespace swift;
using namespace symbolgraphgen;
namespace {
int serializeSymbolGraph(SymbolGraph &SG,
const SymbolGraphOptions &Options) {
SmallString<256> FileName(SG.M.getNameStr());
if (SG.ExtendedModule.hasValue()) {
FileName.push_back('@');
FileName.append(SG.ExtendedModule.getValue()->getNameStr());
}
FileName.append(".symbols.json");
SmallString<1024> OutputPath(Options.OutputDir);
llvm::sys::path::append(OutputPath, FileName);
std::error_code Error;
llvm::raw_fd_ostream OS(OutputPath, Error, llvm::sys::fs::FA_Write);
if (Error) {
llvm::errs() << "Couldn't open output file '" << OutputPath
<< " for writing: "
<< Error.message() << "\n";
return EXIT_FAILURE;
}
llvm::json::OStream J(OS, Options.PrettyPrint ? 2 : 0);
SG.serialize(J);
return EXIT_SUCCESS;
}
} // end anonymous namespace
// MARK: - Main Entry Point
/// Emit a symbol graph JSON file for a `ModuleDecl`.
@@ -41,16 +70,13 @@ symbolgraphgen::emitSymbolGraphForModule(ModuleDecl *M,
<< "Found " << Walker.Graph.Nodes.size() << " symbols and "
<< Walker.Graph.Edges.size() << " relationships.\n";
std::error_code Error;
llvm::raw_fd_ostream OS(Options.OutputPath, Error, llvm::sys::fs::FA_Write);
if (Error) {
llvm::errs() << "Couldn't open output file for writing: "
<< Error.message() << "\n";
return EXIT_FAILURE;
int Success = EXIT_SUCCESS;
Success |= serializeSymbolGraph(Walker.Graph, Options);
for (auto Pair : Walker.ExtendedModuleGraphs) {
Success |= serializeSymbolGraph(*Pair.getSecond(), Options);
}
llvm::json::OStream J(OS, Options.PrettyPrint ? 2 : 0);
Walker.Graph.serialize(J);
return EXIT_SUCCESS;
return Success;
}