* deprecated the third version number as storage update number * version can now have either 3 or 4 numbers * whether a .coatidb file is outdated is now solely determined by the SqliteStorage::STORAGE_VERSION
1432 lines
35 KiB
C++
1432 lines
35 KiB
C++
#include "data/PersistentStorage.h"
|
|
|
|
#include <sstream>
|
|
#include <queue>
|
|
|
|
#include "utility/Cache.h"
|
|
#include "utility/file/FileSystem.h"
|
|
#include "utility/logging/logging.h"
|
|
#include "utility/messaging/type/MessageClearErrorCount.h"
|
|
#include "utility/messaging/type/MessageShowErrors.h"
|
|
#include "utility/messaging/type/MessageStatus.h"
|
|
#include "utility/text/TextAccess.h"
|
|
#include "utility/TimePoint.h"
|
|
#include "utility/tracing.h"
|
|
#include "utility/utility.h"
|
|
#include "utility/utilityString.h"
|
|
#include "utility/Version.h"
|
|
#include "utility/utilityString.h"
|
|
|
|
#include "data/graph/token_component/TokenComponentAggregation.h"
|
|
#include "data/graph/token_component/TokenComponentSignature.h"
|
|
#include "data/graph/Graph.h"
|
|
#include "data/location/TokenLocation.h"
|
|
#include "data/location/TokenLocationFile.h"
|
|
#include "data/location/TokenLocationLine.h"
|
|
#include "data/parser/ParseLocation.h"
|
|
#include "data/type/DataType.h"
|
|
#include "settings/ApplicationSettings.h"
|
|
|
|
PersistentStorage::PersistentStorage(const FilePath& dbPath)
|
|
: m_sqliteStorage(dbPath)
|
|
{
|
|
m_commandIndex.addNode(0, NameHierarchy(SearchMatch::getCommandName(SearchMatch::COMMAND_ALL)));
|
|
m_commandIndex.addNode(0, NameHierarchy(SearchMatch::getCommandName(SearchMatch::COMMAND_ERROR)));
|
|
// m_commandIndex.addNode(0, NameHierarchy(SearchMatch::getCommandName(SearchMatch::COMMAND_COLOR_SCHEME_TEST)));
|
|
m_commandIndex.finishSetup();
|
|
}
|
|
|
|
PersistentStorage::~PersistentStorage()
|
|
{
|
|
}
|
|
|
|
Id PersistentStorage::addFile(const std::string& name, const std::string& filePath, const std::string& modificationTime)
|
|
{
|
|
Id fileId = m_sqliteStorage.getFileByPath(filePath).id;
|
|
if (fileId == 0)
|
|
{
|
|
NameHierarchy nameHierarchy;
|
|
nameHierarchy.push(std::make_shared<NameElement>(name));
|
|
|
|
fileId = m_sqliteStorage.addFile(
|
|
NameHierarchy::serialize(nameHierarchy),
|
|
filePath,
|
|
modificationTime
|
|
);
|
|
}
|
|
return fileId;
|
|
}
|
|
|
|
Id PersistentStorage::addNode(int type, const std::string& serializedName, int definitionType)
|
|
{
|
|
const StorageNode storedNode = m_sqliteStorage.getNodeBySerializedName(serializedName);
|
|
|
|
Id nodeId = storedNode.id;
|
|
|
|
if (nodeId == 0)
|
|
{
|
|
nodeId = m_sqliteStorage.addNode(type, serializedName, definitionType);
|
|
}
|
|
else
|
|
{
|
|
if (storedNode.definitionType == 0)
|
|
{
|
|
if (definitionType > 0)
|
|
{
|
|
m_sqliteStorage.setNodeDefinitionType(definitionType, nodeId);
|
|
}
|
|
|
|
if (storedNode.type < type)
|
|
{
|
|
m_sqliteStorage.setNodeType(type, nodeId);
|
|
}
|
|
}
|
|
}
|
|
return nodeId;
|
|
}
|
|
|
|
Id PersistentStorage::addEdge(int type, Id sourceId, Id targetId)
|
|
{
|
|
Id edgeId = m_sqliteStorage.getEdgeBySourceTargetType(sourceId, targetId, type).id;
|
|
if (edgeId == 0)
|
|
{
|
|
edgeId = m_sqliteStorage.addEdge(type, sourceId, targetId);
|
|
}
|
|
return edgeId;
|
|
}
|
|
|
|
Id PersistentStorage::addLocalSymbol(const std::string& name)
|
|
{
|
|
Id localSymbolId = m_sqliteStorage.getLocalSymbolByName(name).id;
|
|
if (localSymbolId == 0)
|
|
{
|
|
localSymbolId = m_sqliteStorage.addLocalSymbol(name);
|
|
}
|
|
return localSymbolId;
|
|
}
|
|
|
|
void PersistentStorage::addSourceLocation(
|
|
Id elementId, Id fileNodeId, uint startLine, uint startCol, uint endLine, uint endCol, int type)
|
|
{
|
|
m_sqliteStorage.addSourceLocation(
|
|
elementId,
|
|
fileNodeId,
|
|
startLine,
|
|
startCol,
|
|
endLine,
|
|
endCol,
|
|
type
|
|
);
|
|
}
|
|
|
|
void PersistentStorage::addComponentAccess(Id edgeId , int type)
|
|
{
|
|
m_sqliteStorage.addComponentAccess(edgeId, type);
|
|
}
|
|
|
|
void PersistentStorage::addCommentLocation(Id fileNodeId, uint startLine, uint startCol, uint endLine, uint endCol)
|
|
{
|
|
m_sqliteStorage.addCommentLocation(
|
|
fileNodeId,
|
|
startLine,
|
|
startCol,
|
|
endLine,
|
|
endCol
|
|
);
|
|
}
|
|
|
|
void PersistentStorage::addError(
|
|
const std::string& message, bool fatal, bool indexed, const std::string& filePath, uint startLine, uint startCol)
|
|
{
|
|
m_sqliteStorage.addError(
|
|
message,
|
|
fatal,
|
|
indexed,
|
|
filePath,
|
|
startLine,
|
|
startCol
|
|
);
|
|
}
|
|
|
|
void PersistentStorage::forEachFile(std::function<void(const Id /*id*/, const StorageFile& /*data*/)> callback) const
|
|
{
|
|
for (StorageFile& file: m_sqliteStorage.getAllFiles())
|
|
{
|
|
callback(file.id, file);
|
|
}
|
|
}
|
|
|
|
void PersistentStorage::forEachNode(std::function<void(const Id /*id*/, const StorageNode& /*data*/)> callback) const
|
|
{
|
|
for (StorageNode& node: m_sqliteStorage.getAllNodes())
|
|
{
|
|
callback(node.id, node);
|
|
}
|
|
}
|
|
|
|
void PersistentStorage::forEachEdge(std::function<void(const Id /*id*/, const StorageEdge& /*data*/)> callback) const
|
|
{
|
|
for (StorageEdge& edge: m_sqliteStorage.getAllEdges())
|
|
{
|
|
callback(edge.id, edge);
|
|
}
|
|
}
|
|
|
|
void PersistentStorage::forEachLocalSymbol(std::function<void(
|
|
const Id /*id*/, const StorageLocalSymbol& /*data*/)> callback) const
|
|
{
|
|
for (StorageLocalSymbol& localSymbol: m_sqliteStorage.getAllLocalSymbols())
|
|
{
|
|
callback(localSymbol.id, localSymbol);
|
|
}
|
|
}
|
|
|
|
void PersistentStorage::forEachSourceLocation(std::function<void(const StorageSourceLocation& /*data*/)> callback) const
|
|
{
|
|
for (StorageSourceLocation& sourceLocation: m_sqliteStorage.getAllSourceLocations())
|
|
{
|
|
callback(sourceLocation);
|
|
}
|
|
}
|
|
|
|
void PersistentStorage::forEachComponentAccess(std::function<void(const StorageComponentAccess& /*data*/)> callback) const
|
|
{
|
|
for (StorageComponentAccess& componentAccess: m_sqliteStorage.getAllComponentAccesses())
|
|
{
|
|
callback(componentAccess);
|
|
}
|
|
}
|
|
|
|
void PersistentStorage::forEachCommentLocation(std::function<void(const StorageCommentLocation& /*data*/)> callback) const
|
|
{
|
|
for (StorageCommentLocation& commentLocation: m_sqliteStorage.getAllCommentLocations())
|
|
{
|
|
callback(commentLocation);
|
|
}
|
|
}
|
|
|
|
void PersistentStorage::forEachError(std::function<void(const StorageError& /*data*/)> callback) const
|
|
{
|
|
for (StorageError& error: m_sqliteStorage.getAllErrors())
|
|
{
|
|
callback(error);
|
|
}
|
|
}
|
|
|
|
void PersistentStorage::startInjection()
|
|
{
|
|
m_preInjectionErrorCount = getErrorCount().total;
|
|
|
|
m_sqliteStorage.beginTransaction();
|
|
}
|
|
|
|
void PersistentStorage::finishInjection()
|
|
{
|
|
m_sqliteStorage.commitTransaction();
|
|
|
|
ErrorCountInfo errorCount = getErrorCount();
|
|
if (m_preInjectionErrorCount != errorCount.total)
|
|
{
|
|
MessageShowErrors msg(errorCount);
|
|
msg.setSendAsTask(false);
|
|
msg.dispatchImmediately();
|
|
}
|
|
}
|
|
|
|
FilePath PersistentStorage::getDbFilePath() const
|
|
{
|
|
return m_sqliteStorage.getDbFilePath();
|
|
}
|
|
|
|
bool PersistentStorage::isEmpty() const
|
|
{
|
|
return m_sqliteStorage.isEmpty();
|
|
}
|
|
|
|
bool PersistentStorage::isIncompatible() const
|
|
{
|
|
return m_sqliteStorage.isIncompatible();
|
|
}
|
|
|
|
void PersistentStorage::setup()
|
|
{
|
|
m_sqliteStorage.setup();
|
|
}
|
|
|
|
void PersistentStorage::clear()
|
|
{
|
|
m_sqliteStorage.clear();
|
|
|
|
clearCaches();
|
|
}
|
|
|
|
void PersistentStorage::clearCaches()
|
|
{
|
|
m_elementIndex.clear();
|
|
m_fileNodeIds.clear();
|
|
m_fileNodePaths.clear();
|
|
m_hierarchyCache.clear();
|
|
m_fullTextSearchIndex.clear();
|
|
}
|
|
|
|
std::set<FilePath> PersistentStorage::getDependingFilePaths(const std::set<FilePath>& filePaths)
|
|
{
|
|
TRACE();
|
|
|
|
std::set<FilePath> dependingFilePaths;
|
|
for (const FilePath& filePath: filePaths)
|
|
{
|
|
std::set<FilePath> dependingFilePathsSubset = getDependingFilePaths(filePath);
|
|
dependingFilePaths.insert(dependingFilePathsSubset.begin(), dependingFilePathsSubset.end());
|
|
}
|
|
return dependingFilePaths;
|
|
}
|
|
|
|
std::set<FilePath> PersistentStorage::getDependingFilePaths(const FilePath& filePath)
|
|
{
|
|
std::set<FilePath> dependingFilePaths;
|
|
|
|
std::vector<StorageEdge> incomingEdges = m_sqliteStorage.getEdgesByTargetType(
|
|
getFileNodeId(filePath), Edge::typeToInt(Edge::EDGE_INCLUDE)
|
|
);
|
|
for (const StorageEdge& incomingEdge: incomingEdges)
|
|
{
|
|
FilePath dependingFilePath = getFileNodePath(incomingEdge.sourceNodeId);
|
|
dependingFilePaths.insert(dependingFilePath);
|
|
|
|
std::set<FilePath> dependingFilePathsSubset = getDependingFilePaths(dependingFilePath);
|
|
dependingFilePaths.insert(dependingFilePathsSubset.begin(), dependingFilePathsSubset.end());
|
|
}
|
|
|
|
return dependingFilePaths;
|
|
}
|
|
|
|
void PersistentStorage::clearFileElements(const std::vector<FilePath>& filePaths)
|
|
{
|
|
TRACE();
|
|
|
|
std::vector<Id> fileNodeIds;
|
|
|
|
for (const FilePath& path : filePaths)
|
|
{
|
|
fileNodeIds.push_back(getFileNodeId(path));
|
|
}
|
|
|
|
if (fileNodeIds.size())
|
|
{
|
|
m_sqliteStorage.removeElementsWithLocationInFiles(fileNodeIds);
|
|
m_sqliteStorage.removeElements(fileNodeIds);
|
|
|
|
m_sqliteStorage.removeErrorsInFiles(filePaths);
|
|
}
|
|
}
|
|
|
|
std::vector<FileInfo> PersistentStorage::getInfoOnAllFiles() const
|
|
{
|
|
TRACE();
|
|
|
|
std::vector<FileInfo> fileInfos;
|
|
|
|
std::vector<StorageFile> storageFiles = m_sqliteStorage.getAllFiles();
|
|
for (size_t i = 0; i < storageFiles.size(); i++)
|
|
{
|
|
boost::posix_time::ptime modificationTime = boost::posix_time::not_a_date_time;
|
|
if (storageFiles[i].modificationTime != "not-a-date-time")
|
|
{
|
|
modificationTime = boost::posix_time::time_from_string(storageFiles[i].modificationTime);
|
|
}
|
|
fileInfos.push_back(FileInfo(
|
|
FilePath(storageFiles[i].filePath),
|
|
modificationTime
|
|
));
|
|
}
|
|
|
|
return fileInfos;
|
|
}
|
|
|
|
void PersistentStorage::logStats() const
|
|
{
|
|
TRACE();
|
|
|
|
std::stringstream ss;
|
|
StorageStats stats = getStorageStats();
|
|
|
|
ss << "\nGraph:\n";
|
|
ss << "\t" << stats.nodeCount << " Nodes\n";
|
|
ss << "\t" << stats.edgeCount << " Edges\n";
|
|
|
|
ss << "\nCode:\n";
|
|
ss << "\t" << stats.fileCount << " Files\n";
|
|
ss << "\t" << stats.fileLOCCount << " Lines of Code\n";
|
|
|
|
ss << "\nErrors:\n";
|
|
ss << "\t" << stats.errorCount.total << " Errors\n";
|
|
ss << "\t" << stats.errorCount.fatal << " Fatal Errors\n";
|
|
|
|
LOG_INFO(ss.str());
|
|
}
|
|
|
|
void PersistentStorage::startParsing()
|
|
{
|
|
MessageClearErrorCount().dispatch();
|
|
|
|
m_sqliteStorage.setVersion();
|
|
}
|
|
|
|
void PersistentStorage::finishParsing()
|
|
{
|
|
TRACE();
|
|
|
|
buildSearchIndex();
|
|
buildFilePathMaps();
|
|
buildHierarchyCache();
|
|
}
|
|
|
|
void PersistentStorage::optimizeMemory()
|
|
{
|
|
TRACE();
|
|
|
|
m_sqliteStorage.optimizeMemory();
|
|
}
|
|
|
|
Id PersistentStorage::getIdForNodeWithNameHierarchy(const NameHierarchy& nameHierarchy) const
|
|
{
|
|
return m_sqliteStorage.getNodeBySerializedName(NameHierarchy::serialize(nameHierarchy)).id;
|
|
}
|
|
|
|
Id PersistentStorage::getIdForEdge(
|
|
Edge::EdgeType type, const NameHierarchy& fromNameHierarchy, const NameHierarchy& toNameHierarchy
|
|
) const
|
|
{
|
|
Id sourceId = getIdForNodeWithNameHierarchy(fromNameHierarchy);
|
|
Id targetId = getIdForNodeWithNameHierarchy(toNameHierarchy);
|
|
return m_sqliteStorage.getEdgeBySourceTargetType(sourceId, targetId, type).id;
|
|
}
|
|
|
|
NameHierarchy PersistentStorage::getNameHierarchyForNodeWithId(Id nodeId) const
|
|
{
|
|
return NameHierarchy::deserialize(m_sqliteStorage.getNodeById(nodeId).serializedName);
|
|
}
|
|
|
|
Node::NodeType PersistentStorage::getNodeTypeForNodeWithId(Id nodeId) const
|
|
{
|
|
return Node::intToType(m_sqliteStorage.getNodeById(nodeId).type);
|
|
}
|
|
|
|
std::shared_ptr<TokenLocationCollection> PersistentStorage::getFullTextSearchLocations(
|
|
const std::string& searchTerm, bool caseSensitive
|
|
) const
|
|
{
|
|
TRACE();
|
|
|
|
if (m_fullTextSearchIndex.fileCount() == 0)
|
|
{
|
|
MessageStatus("Building fulltext search index", false, true).dispatch();
|
|
buildFullTextSearchIndex();
|
|
}
|
|
|
|
MessageStatus(
|
|
std::string("Searching fulltext (case-") + (caseSensitive ? "sensitive" : "insensitive") + "): " + searchTerm,
|
|
false, true
|
|
).dispatch();
|
|
|
|
std::shared_ptr<TokenLocationCollection> collection = std::make_shared<TokenLocationCollection>();
|
|
|
|
std::vector<FullTextSearchResult> hits = m_fullTextSearchIndex.searchForTerm(searchTerm);
|
|
|
|
int termLength = searchTerm.length();
|
|
FilePath filepath;
|
|
std::shared_ptr<TextAccess> file;
|
|
ParseLocation location;
|
|
for (size_t i = 0; i < hits.size(); i++)
|
|
{
|
|
filepath = getFileNodePath(hits[i].fileId);
|
|
file = getFileContent(filepath);
|
|
|
|
int charsInPreviousLines = 0;
|
|
int lineNumber = 1;
|
|
std::string line;
|
|
line = file->getLine(lineNumber);
|
|
|
|
for (int pos : hits[i].positions)
|
|
{
|
|
bool addHit = true;
|
|
while( (charsInPreviousLines + (int)line.length()) < pos)
|
|
{
|
|
lineNumber++;
|
|
charsInPreviousLines += line.length();
|
|
line = file->getLine(lineNumber);
|
|
}
|
|
location.startLineNumber = lineNumber;
|
|
location.startColumnNumber = pos - charsInPreviousLines + 1;
|
|
|
|
if ( caseSensitive )
|
|
{
|
|
if( line.substr(location.startColumnNumber-1, termLength) != searchTerm )
|
|
{
|
|
addHit = false;
|
|
}
|
|
}
|
|
|
|
while( (charsInPreviousLines + (int)line.length()) < pos + termLength)
|
|
{
|
|
lineNumber++;
|
|
charsInPreviousLines += line.length();
|
|
line = file->getLine(lineNumber);
|
|
}
|
|
|
|
location.endLineNumber = lineNumber;
|
|
location.endColumnNumber = pos + termLength - charsInPreviousLines;
|
|
|
|
if ( addHit )
|
|
{
|
|
collection->addTokenLocation(
|
|
collection->getTokenLocationCount(),
|
|
0,
|
|
filepath,
|
|
location.startLineNumber,
|
|
location.startColumnNumber,
|
|
location.endLineNumber,
|
|
location.endColumnNumber
|
|
)->setType(LOCATION_FULLTEXT);
|
|
}
|
|
}
|
|
}
|
|
|
|
MessageStatus(
|
|
std::to_string(collection->getTokenLocationCount()) + " results in " +
|
|
std::to_string(collection->getTokenLocationFileCount()) + " files for fulltext search (case-" +
|
|
(caseSensitive ? "sensitive" : "insensitive") + "): " + searchTerm,
|
|
false, false
|
|
).dispatch();
|
|
|
|
return collection;
|
|
}
|
|
|
|
std::vector<SearchMatch> PersistentStorage::getAutocompletionMatches(const std::string& query) const
|
|
{
|
|
TRACE();
|
|
|
|
std::vector<SearchResult> commandResults = m_commandIndex.search(query, 0);
|
|
|
|
const size_t maxResultCount = 100;
|
|
std::vector<SearchResult> elementResults = m_elementIndex.search(query, maxResultCount);
|
|
|
|
std::vector<SearchResult> results;
|
|
utility::append(results, commandResults);
|
|
utility::append(results, elementResults);
|
|
|
|
std::sort(results.begin(), results.end(),
|
|
[](const SearchResult& a, const SearchResult& b)
|
|
{
|
|
// should a be ranked higher than b?
|
|
if (a.score > b.score)
|
|
{
|
|
return true;
|
|
}
|
|
else if (a.score == b.score)
|
|
{
|
|
if (a.text.size() < b.text.size())
|
|
{
|
|
return true;
|
|
}
|
|
else if (a.text.size() == b.text.size())
|
|
{
|
|
for (size_t i = 0; i < a.text.size(); i++)
|
|
{
|
|
if (tolower(a.text[i]) != tolower(b.text[i]))
|
|
{
|
|
return tolower(a.text[i]) < tolower(b.text[i]);
|
|
}
|
|
else
|
|
{
|
|
if (a.text[i] < b.text[i])
|
|
{
|
|
return true;
|
|
}
|
|
else if (a.text[i] > b.text[i])
|
|
{
|
|
return false;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
);
|
|
|
|
std::map<Id, StorageNode> storageNodesMap;
|
|
{
|
|
std::vector<Id> elementIds;
|
|
|
|
for (const SearchResult& result : results)
|
|
{
|
|
elementIds.insert(elementIds.end(), result.elementIds.begin(), result.elementIds.end());
|
|
}
|
|
|
|
std::vector<StorageNode> storageNodes = m_sqliteStorage.getNodesByIds(elementIds);
|
|
|
|
for (StorageNode& node : storageNodes)
|
|
{
|
|
if (node.id > 0)
|
|
{
|
|
storageNodesMap.emplace(node.id, node);
|
|
}
|
|
}
|
|
}
|
|
|
|
std::vector<SearchMatch> matches;
|
|
for (const SearchResult& result : results)
|
|
{
|
|
SearchMatch match;
|
|
|
|
const StorageNode* firstNode = nullptr;
|
|
for (const Id& elementId : result.elementIds)
|
|
{
|
|
if (elementId != 0)
|
|
{
|
|
const StorageNode& node = storageNodesMap[elementId];
|
|
match.nameHierarchies.push_back(NameHierarchy::deserialize(node.serializedName));
|
|
|
|
if (!firstNode)
|
|
{
|
|
firstNode = &node;
|
|
}
|
|
}
|
|
}
|
|
|
|
match.text = result.text;
|
|
match.indices = result.indices;
|
|
|
|
if (firstNode)
|
|
{
|
|
match.nodeType = Node::intToType(firstNode->type);
|
|
match.typeName = Node::getTypeString(match.nodeType);
|
|
|
|
if (intToDefinitionType(firstNode->definitionType) == DEFINITION_NONE
|
|
&& match.nodeType != Node::NODE_UNDEFINED)
|
|
{
|
|
match.typeName = "undefined " + match.typeName;
|
|
}
|
|
match.searchType = SearchMatch::SEARCH_TOKEN;
|
|
}
|
|
else
|
|
{
|
|
match.searchType = SearchMatch::SEARCH_COMMAND;
|
|
match.typeName = "command";
|
|
}
|
|
|
|
matches.push_back(match);
|
|
}
|
|
|
|
return matches;
|
|
}
|
|
|
|
std::vector<SearchMatch> PersistentStorage::getSearchMatchesForTokenIds(const std::vector<Id>& elementIds) const
|
|
{
|
|
// todo: what if all these elements share the same node in the searchindex?
|
|
// In that case there should be only one search match.
|
|
std::vector<SearchMatch> matches;
|
|
|
|
for (Id elementId : elementIds)
|
|
{
|
|
SearchMatch match;
|
|
|
|
if (m_sqliteStorage.isFile(elementId))
|
|
{
|
|
match.nodeType = Node::NODE_FILE;
|
|
}
|
|
else if (m_sqliteStorage.isNode(elementId))
|
|
{
|
|
StorageNode node = m_sqliteStorage.getNodeById(elementId);
|
|
match.nodeType = Node::intToType(node.type);
|
|
}
|
|
else
|
|
{
|
|
continue;
|
|
}
|
|
|
|
NameHierarchy nameHierarchy = NameHierarchy::deserialize(m_sqliteStorage.getNodeById(elementId).serializedName);
|
|
match.text = nameHierarchy.getQualifiedName();
|
|
match.nameHierarchies.push_back(nameHierarchy.getQualifiedName());
|
|
match.searchType = SearchMatch::SEARCH_TOKEN;
|
|
|
|
matches.push_back(match);
|
|
}
|
|
|
|
return matches;
|
|
}
|
|
|
|
std::shared_ptr<Graph> PersistentStorage::getGraphForAll() const
|
|
{
|
|
TRACE();
|
|
|
|
std::shared_ptr<Graph> graph = std::make_shared<Graph>();
|
|
|
|
std::vector<Id> tokenIds;
|
|
for (StorageNode node: m_sqliteStorage.getAllNodes())
|
|
{
|
|
if (intToDefinitionType(node.definitionType) == DEFINITION_EXPLICIT &&
|
|
(!m_hierarchyCache.isChildOfVisibleNodeOrInvisible(node.id) ||
|
|
Node::intToType(node.type) == Node::NODE_NAMESPACE))
|
|
{
|
|
tokenIds.push_back(node.id);
|
|
}
|
|
}
|
|
|
|
addNodesToGraph(tokenIds, graph.get());
|
|
|
|
return graph;
|
|
}
|
|
|
|
std::shared_ptr<Graph> PersistentStorage::getGraphForActiveTokenIds(const std::vector<Id>& tokenIds) const
|
|
{
|
|
TRACE();
|
|
|
|
std::shared_ptr<Graph> g = std::make_shared<Graph>();
|
|
Graph* graph = g.get();
|
|
|
|
std::vector<Id> ids(tokenIds);
|
|
bool isNamespace = false;
|
|
|
|
std::vector<Id> nodeIds;
|
|
std::vector<Id> edgeIds;
|
|
bool addAggregations = false;
|
|
|
|
if (tokenIds.size() == 1)
|
|
{
|
|
const Id elementId = tokenIds[0];
|
|
StorageNode node = m_sqliteStorage.getNodeById(elementId);
|
|
|
|
if (node.id > 0)
|
|
{
|
|
if (Node::intToType(node.type) == Node::NODE_NAMESPACE)
|
|
{
|
|
ids.clear();
|
|
m_hierarchyCache.addFirstChildIdsForNodeId(elementId, &ids);
|
|
|
|
isNamespace = true;
|
|
}
|
|
else
|
|
{
|
|
nodeIds.push_back(elementId);
|
|
|
|
std::vector<StorageEdge> edges = m_sqliteStorage.getEdgesBySourceOrTargetId(elementId);
|
|
for (const StorageEdge& edge : edges)
|
|
{
|
|
if (Edge::intToType(edge.type) != Edge::EDGE_MEMBER)
|
|
{
|
|
edgeIds.push_back(edge.id);
|
|
}
|
|
}
|
|
|
|
addAggregations = true;
|
|
}
|
|
}
|
|
else if (m_sqliteStorage.isEdge(elementId))
|
|
{
|
|
edgeIds.push_back(elementId);
|
|
}
|
|
}
|
|
|
|
if (ids.size() >= 1 || isNamespace)
|
|
{
|
|
std::vector<StorageNode> nodes = m_sqliteStorage.getNodesByIds(ids);
|
|
for (const StorageNode& node : nodes)
|
|
{
|
|
if (node.id > 0 && (!isNamespace || intToDefinitionType(node.definitionType) != DEFINITION_IMPLICIT))
|
|
{
|
|
nodeIds.push_back(node.id);
|
|
}
|
|
}
|
|
|
|
if (nodeIds.size() != ids.size())
|
|
{
|
|
std::vector<StorageEdge> edges = m_sqliteStorage.getEdgesByIds(ids);
|
|
for (const StorageEdge& edge : edges)
|
|
{
|
|
if (edge.id > 0)
|
|
{
|
|
edgeIds.push_back(edge.id);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
if (isNamespace)
|
|
{
|
|
addNodesToGraph(nodeIds, graph);
|
|
}
|
|
else
|
|
{
|
|
addNodesWithChildrenAndEdgesToGraph(nodeIds, edgeIds, graph);
|
|
}
|
|
|
|
if (addAggregations)
|
|
{
|
|
addAggregationEdgesToGraph(tokenIds[0], graph);
|
|
}
|
|
|
|
addComponentAccessToGraph(graph);
|
|
|
|
return g;
|
|
}
|
|
|
|
// TODO: rename: getActiveElementIdsForId; TODO: make separate function for declarationId
|
|
std::vector<Id> PersistentStorage::getActiveTokenIdsForId(Id tokenId, Id* declarationId) const
|
|
{
|
|
std::vector<Id> activeTokenIds;
|
|
|
|
if (!(m_sqliteStorage.isEdge(tokenId) || m_sqliteStorage.isNode(tokenId)))
|
|
{
|
|
return activeTokenIds;
|
|
}
|
|
|
|
activeTokenIds.push_back(tokenId);
|
|
|
|
if (m_sqliteStorage.isNode(tokenId))
|
|
{
|
|
*declarationId = tokenId;
|
|
|
|
std::vector<StorageEdge> incomingEdges = m_sqliteStorage.getEdgesByTargetId(tokenId);
|
|
for (size_t i = 0; i < incomingEdges.size(); i++)
|
|
{
|
|
activeTokenIds.push_back(incomingEdges[i].id);
|
|
}
|
|
}
|
|
|
|
return activeTokenIds;
|
|
}
|
|
|
|
std::vector<Id> PersistentStorage::getNodeIdsForLocationIds(const std::vector<Id>& locationIds) const
|
|
{
|
|
TRACE();
|
|
|
|
std::set<Id> edgeIds;
|
|
std::set<Id> nodeIds;
|
|
std::set<Id> implicitNodeIds;
|
|
|
|
for (Id locationId : locationIds)
|
|
{
|
|
Id elementId = m_sqliteStorage.getElementIdByLocationId(locationId);
|
|
|
|
StorageEdge edge = m_sqliteStorage.getEdgeById(elementId);
|
|
if (edge.id != 0) // here we test if location is an edge.
|
|
{
|
|
edgeIds.insert(edge.targetNodeId);
|
|
}
|
|
else if(m_sqliteStorage.isNode(elementId))
|
|
{
|
|
StorageNode node = m_sqliteStorage.getNodeById(elementId);
|
|
if (node.id != 0)
|
|
{
|
|
if (intToDefinitionType(node.definitionType) == DEFINITION_IMPLICIT)
|
|
{
|
|
implicitNodeIds.insert(elementId);
|
|
}
|
|
else
|
|
{
|
|
nodeIds.insert(elementId);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
if (nodeIds.size() == 0)
|
|
{
|
|
nodeIds = implicitNodeIds;
|
|
}
|
|
|
|
if (nodeIds.size())
|
|
{
|
|
return utility::toVector(nodeIds);
|
|
}
|
|
|
|
return utility::toVector(edgeIds);
|
|
}
|
|
|
|
std::vector<Id> PersistentStorage::getLocalSymbolIdsForLocationIds(const std::vector<Id>& locationIds) const
|
|
{
|
|
std::set<Id> localSymbolIds;
|
|
|
|
for (Id locationId : locationIds)
|
|
{
|
|
Id elementId = m_sqliteStorage.getElementIdByLocationId(locationId);
|
|
|
|
if (m_sqliteStorage.getNodeById(elementId).id == 0 && m_sqliteStorage.getEdgeById(elementId).id == 0)
|
|
{
|
|
localSymbolIds.insert(elementId);
|
|
}
|
|
}
|
|
|
|
return utility::toVector(localSymbolIds);
|
|
}
|
|
|
|
std::vector<Id> PersistentStorage::getTokenIdsForMatches(const std::vector<SearchMatch>& matches) const
|
|
{
|
|
std::set<Id> idSet;
|
|
for (const SearchMatch& match : matches)
|
|
{
|
|
for (size_t i = 0; i < match.nameHierarchies.size(); i++)
|
|
{
|
|
idSet.insert(
|
|
m_sqliteStorage.getNodeBySerializedName(NameHierarchy::serialize(match.nameHierarchies[i])).id
|
|
);
|
|
}
|
|
}
|
|
|
|
std::vector<Id> ids;
|
|
for (std::set<Id>::const_iterator it = idSet.begin(); it != idSet.end(); it++)
|
|
{
|
|
if (*it != 0)
|
|
{
|
|
ids.push_back(*it);
|
|
}
|
|
}
|
|
|
|
return ids;
|
|
}
|
|
|
|
Id PersistentStorage::getTokenIdForFileNode(const FilePath& filePath) const
|
|
{
|
|
return m_sqliteStorage.getFileByPath(filePath.str()).id;
|
|
}
|
|
|
|
std::shared_ptr<TokenLocationCollection> PersistentStorage::getTokenLocationsForTokenIds(
|
|
const std::vector<Id>& tokenIds) const
|
|
{
|
|
TRACE();
|
|
|
|
std::shared_ptr<TokenLocationCollection> collection = std::make_shared<TokenLocationCollection>();
|
|
|
|
std::vector<Id> fileIds;
|
|
std::vector<Id> nonFileIds;
|
|
|
|
for (const Id tokenId : tokenIds)
|
|
{
|
|
if (!getFileNodePath(tokenId).empty())
|
|
{
|
|
fileIds.push_back(tokenId);
|
|
}
|
|
else
|
|
{
|
|
nonFileIds.push_back(tokenId);
|
|
}
|
|
}
|
|
|
|
for (Id fileId: fileIds)
|
|
{
|
|
StorageFile storageFile = m_sqliteStorage.getFileById(fileId);
|
|
collection->addTokenLocationFile(m_sqliteStorage.getTokenLocationsForFile(storageFile.filePath));
|
|
}
|
|
|
|
std::vector<StorageSourceLocation> locations = m_sqliteStorage.getTokenLocationsForElementIds(nonFileIds);
|
|
for (size_t i = 0; i < locations.size(); i++)
|
|
{
|
|
const StorageSourceLocation& location = locations[i];
|
|
|
|
TokenLocation* loc = collection->addTokenLocation(
|
|
location.id,
|
|
location.elementId,
|
|
getFileNodePath(location.fileNodeId),
|
|
location.startLine,
|
|
location.startCol,
|
|
location.endLine,
|
|
location.endCol
|
|
);
|
|
|
|
if (loc)
|
|
{
|
|
loc->setType(intToLocationType(location.type));
|
|
}
|
|
}
|
|
|
|
return collection;
|
|
}
|
|
|
|
std::shared_ptr<TokenLocationCollection> PersistentStorage::getTokenLocationsForLocationIds(
|
|
const std::vector<Id>& locationIds
|
|
) const
|
|
{
|
|
TRACE();
|
|
|
|
std::shared_ptr<TokenLocationCollection> collection = std::make_shared<TokenLocationCollection>();
|
|
|
|
for (size_t i = 0; i < locationIds.size(); i++)
|
|
{
|
|
StorageSourceLocation location = m_sqliteStorage.getSourceLocationById(locationIds[i]);
|
|
collection->addTokenLocation(
|
|
location.id,
|
|
location.elementId,
|
|
m_sqliteStorage.getFileById(location.fileNodeId).filePath, // TODO: optimize: only once per file!
|
|
location.startLine,
|
|
location.startCol,
|
|
location.endLine,
|
|
location.endCol
|
|
)->setType(intToLocationType(location.type));
|
|
}
|
|
|
|
return collection;
|
|
}
|
|
|
|
std::shared_ptr<TokenLocationFile> PersistentStorage::getTokenLocationsForFile(const std::string& filePath) const
|
|
{
|
|
TRACE();
|
|
|
|
return m_sqliteStorage.getTokenLocationsForFile(filePath);
|
|
}
|
|
|
|
std::shared_ptr<TokenLocationFile> PersistentStorage::getTokenLocationsForLinesInFile(
|
|
const std::string& filePath, uint firstLineNumber, uint lastLineNumber
|
|
) const
|
|
{
|
|
TRACE();
|
|
|
|
return getTokenLocationsForFile(filePath)->getFilteredByLines(firstLineNumber, lastLineNumber);
|
|
}
|
|
|
|
std::shared_ptr<TokenLocationCollection> PersistentStorage::getErrorTokenLocations(std::vector<ErrorInfo>* errors) const
|
|
{
|
|
TRACE();
|
|
|
|
std::shared_ptr<TokenLocationCollection> errorCollection = std::make_shared<TokenLocationCollection>();
|
|
|
|
bool showExternalNonFatalErrors = ApplicationSettings::getInstance()->getShowExternalNonFatalErrors();
|
|
|
|
std::vector<StorageError> storageErrors = m_sqliteStorage.getAllErrors();
|
|
for (size_t i = 0; i < storageErrors.size(); i++)
|
|
{
|
|
const StorageError& error = storageErrors[i];
|
|
if (error.fatal || error.indexed || showExternalNonFatalErrors)
|
|
{
|
|
errorCollection->addTokenLocation(
|
|
i, i, error.filePath, error.lineNumber, error.columnNumber, error.lineNumber, error.columnNumber
|
|
)->setType(LOCATION_ERROR);
|
|
errors->push_back(ErrorInfo(error.message, error.filePath, i, error.fatal));
|
|
}
|
|
}
|
|
|
|
return errorCollection;
|
|
}
|
|
|
|
std::shared_ptr<TokenLocationFile> PersistentStorage::getCommentLocationsInFile(const FilePath& filePath) const
|
|
{
|
|
TRACE();
|
|
|
|
std::shared_ptr<TokenLocationFile> file = std::make_shared<TokenLocationFile>(filePath);
|
|
|
|
std::vector<StorageCommentLocation> storageLocations = m_sqliteStorage.getCommentLocationsInFile(filePath);
|
|
for (size_t i = 0; i < storageLocations.size(); i++)
|
|
{
|
|
file->addTokenLocation(
|
|
storageLocations[i].id,
|
|
0, // comment token location has no element.
|
|
storageLocations[i].startLine,
|
|
storageLocations[i].startCol,
|
|
storageLocations[i].endLine,
|
|
storageLocations[i].endCol
|
|
);
|
|
}
|
|
|
|
return file;
|
|
}
|
|
|
|
std::shared_ptr<TextAccess> PersistentStorage::getFileContent(const FilePath& filePath) const
|
|
{
|
|
return m_sqliteStorage.getFileContentByPath(filePath.str());
|
|
}
|
|
|
|
FileInfo PersistentStorage::getFileInfoForFilePath(const FilePath& filePath) const
|
|
{
|
|
return FileInfo(filePath, m_sqliteStorage.getFileByPath(filePath).modificationTime);
|
|
}
|
|
|
|
std::vector<FileInfo> PersistentStorage::getFileInfosForFilePaths(const std::vector<FilePath>& filePaths) const
|
|
{
|
|
std::vector<FileInfo> fileInfos;
|
|
|
|
std::vector<StorageFile> storageFiles = m_sqliteStorage.getFilesByPaths(filePaths);
|
|
for (const StorageFile& file : storageFiles)
|
|
{
|
|
fileInfos.push_back(FileInfo(FilePath(file.filePath), file.modificationTime));
|
|
}
|
|
|
|
return fileInfos;
|
|
}
|
|
|
|
ErrorCountInfo PersistentStorage::getErrorCount() const
|
|
{
|
|
bool showExternalNonFatalErrors = ApplicationSettings::getInstance()->getShowExternalNonFatalErrors();
|
|
|
|
ErrorCountInfo info;
|
|
|
|
std::vector<StorageError> storageErrors = m_sqliteStorage.getAllErrors();
|
|
for (const StorageError& error : storageErrors)
|
|
{
|
|
if (error.fatal || error.indexed || showExternalNonFatalErrors)
|
|
{
|
|
info.total++;
|
|
}
|
|
|
|
if (error.fatal)
|
|
{
|
|
info.fatal++;
|
|
}
|
|
}
|
|
|
|
return info;
|
|
}
|
|
|
|
StorageStats PersistentStorage::getStorageStats() const
|
|
{
|
|
TRACE();
|
|
|
|
StorageStats stats;
|
|
|
|
stats.nodeCount = m_sqliteStorage.getNodeCount();
|
|
stats.edgeCount = m_sqliteStorage.getEdgeCount();
|
|
|
|
stats.fileCount = m_sqliteStorage.getFileCount();
|
|
stats.fileLOCCount = m_sqliteStorage.getFileLOCCount();
|
|
|
|
stats.errorCount = getErrorCount();
|
|
|
|
return stats;
|
|
}
|
|
|
|
Id PersistentStorage::getFileNodeId(const FilePath& filePath) const
|
|
{
|
|
if (filePath.empty())
|
|
{
|
|
LOG_ERROR("No file path set");
|
|
return 0;
|
|
}
|
|
|
|
std::map<FilePath, Id>::const_iterator it = m_fileNodeIds.find(filePath);
|
|
|
|
if (it != m_fileNodeIds.end())
|
|
{
|
|
return it->second;
|
|
}
|
|
|
|
return 0;
|
|
}
|
|
|
|
FilePath PersistentStorage::getFileNodePath(Id fileId) const
|
|
{
|
|
if (fileId == 0)
|
|
{
|
|
LOG_ERROR("No file id set");
|
|
return FilePath();
|
|
}
|
|
|
|
std::map<Id, FilePath>::const_iterator it = m_fileNodePaths.find(fileId);
|
|
|
|
if (it != m_fileNodePaths.end())
|
|
{
|
|
return it->second;
|
|
}
|
|
|
|
return FilePath();
|
|
}
|
|
|
|
Id PersistentStorage::getLastVisibleParentNodeId(const Id nodeId) const
|
|
{
|
|
return m_hierarchyCache.getLastVisibleParentNodeId(nodeId);
|
|
}
|
|
|
|
std::vector<Id> PersistentStorage::getAllChildNodeIds(const Id nodeId) const
|
|
{
|
|
std::vector<Id> childNodeIds;
|
|
std::vector<Id> edgeIds;
|
|
|
|
m_hierarchyCache.addAllChildIdsForNodeId(nodeId, &childNodeIds, &edgeIds);
|
|
|
|
return childNodeIds;
|
|
}
|
|
|
|
void PersistentStorage::addNodesToGraph(const std::vector<Id>& nodeIds, Graph* graph) const
|
|
{
|
|
TRACE();
|
|
|
|
if (nodeIds.size() == 0)
|
|
{
|
|
return;
|
|
}
|
|
|
|
std::vector<StorageNode> storageNodes = m_sqliteStorage.getNodesByIds(nodeIds);
|
|
|
|
for (const StorageNode& storageNode : storageNodes)
|
|
{
|
|
NameHierarchy nameHierarchy = NameHierarchy::deserialize(storageNode.serializedName);
|
|
|
|
Node::NodeType type = Node::intToType(storageNode.type);
|
|
DefinitionType defType = intToDefinitionType(storageNode.definitionType);
|
|
Node* node = graph->createNode(
|
|
storageNode.id,
|
|
type,
|
|
nameHierarchy,
|
|
defType != DEFINITION_NONE
|
|
);
|
|
|
|
if (defType == DEFINITION_IMPLICIT)
|
|
{
|
|
node->setImplicit(true);
|
|
}
|
|
else if (defType == DEFINITION_EXPLICIT)
|
|
{
|
|
node->setExplicit(true);
|
|
}
|
|
|
|
if (type == Node::NODE_FUNCTION || type == Node::NODE_METHOD)
|
|
{
|
|
std::string signatureString = nameHierarchy.getRawNameWithSignature();
|
|
if (signatureString.size() > 0) // this should always be the case since functions and methods must have sigs.
|
|
{
|
|
node->addComponentSignature(
|
|
std::make_shared<TokenComponentSignature>(signatureString)
|
|
);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
void PersistentStorage::addEdgesToGraph(const std::vector<Id>& edgeIds, Graph* graph) const
|
|
{
|
|
TRACE();
|
|
|
|
if (edgeIds.size() == 0)
|
|
{
|
|
return;
|
|
}
|
|
|
|
std::vector<StorageEdge> storageEdges = m_sqliteStorage.getEdgesByIds(edgeIds);
|
|
for (const StorageEdge& storageEdge : storageEdges)
|
|
{
|
|
Node* sourceNode = graph->getNodeById(storageEdge.sourceNodeId);
|
|
Node* targetNode = graph->getNodeById(storageEdge.targetNodeId);
|
|
|
|
if (sourceNode && targetNode)
|
|
{
|
|
graph->createEdge(storageEdge.id, Edge::intToType(storageEdge.type), sourceNode, targetNode);
|
|
}
|
|
else
|
|
{
|
|
LOG_ERROR("Can't add edge because nodes are not present");
|
|
}
|
|
}
|
|
}
|
|
|
|
void PersistentStorage::addNodesWithChildrenAndEdgesToGraph(
|
|
const std::vector<Id>& nodeIds, const std::vector<Id>& edgeIds, Graph* graph
|
|
) const
|
|
{
|
|
TRACE();
|
|
|
|
std::set<Id> parentNodeIds;
|
|
|
|
for (Id nodeId : nodeIds)
|
|
{
|
|
parentNodeIds.insert(getLastVisibleParentNodeId(nodeId));
|
|
}
|
|
|
|
if (edgeIds.size() > 0)
|
|
{
|
|
std::vector<StorageEdge> storageEdges = m_sqliteStorage.getEdgesByIds(edgeIds);
|
|
for (const StorageEdge& storageEdge : storageEdges)
|
|
{
|
|
parentNodeIds.insert(getLastVisibleParentNodeId(storageEdge.sourceNodeId));
|
|
parentNodeIds.insert(getLastVisibleParentNodeId(storageEdge.targetNodeId));
|
|
}
|
|
}
|
|
|
|
std::vector<Id> allNodeIds;
|
|
std::vector<Id> allEdgeIds = edgeIds;
|
|
|
|
for (Id parentNodeId : parentNodeIds)
|
|
{
|
|
allNodeIds.push_back(parentNodeId);
|
|
m_hierarchyCache.addAllChildIdsForNodeId(parentNodeId, &allNodeIds, &allEdgeIds);
|
|
}
|
|
|
|
addNodesToGraph(allNodeIds, graph);
|
|
addEdgesToGraph(allEdgeIds, graph);
|
|
}
|
|
|
|
void PersistentStorage::addAggregationEdgesToGraph(const Id nodeId, Graph* graph) const
|
|
{
|
|
TRACE();
|
|
|
|
struct EdgeInfo
|
|
{
|
|
Id edgeId;
|
|
bool forward;
|
|
};
|
|
|
|
// build aggregation edges:
|
|
// get all children of the active node
|
|
std::vector<Id> childNodeIds = getAllChildNodeIds(nodeId);
|
|
if (childNodeIds.size() == 0)
|
|
{
|
|
return;
|
|
}
|
|
|
|
// get all edges of the children
|
|
std::map<Id, std::vector<EdgeInfo>> connectedNodeIds;
|
|
|
|
std::vector<StorageEdge> outgoingEdges = m_sqliteStorage.getEdgesBySourceIds(childNodeIds);
|
|
for (size_t j = 0; j < outgoingEdges.size(); j++)
|
|
{
|
|
EdgeInfo edgeInfo;
|
|
edgeInfo.edgeId = outgoingEdges[j].id;
|
|
edgeInfo.forward = true;
|
|
connectedNodeIds[outgoingEdges[j].targetNodeId].push_back(edgeInfo);
|
|
}
|
|
|
|
std::vector<StorageEdge> incomingEdges = m_sqliteStorage.getEdgesByTargetIds(childNodeIds);
|
|
for (size_t j = 0; j < incomingEdges.size(); j++)
|
|
{
|
|
EdgeInfo edgeInfo;
|
|
edgeInfo.edgeId = incomingEdges[j].id;
|
|
edgeInfo.forward = false;
|
|
connectedNodeIds[incomingEdges[j].sourceNodeId].push_back(edgeInfo);
|
|
}
|
|
|
|
// get all parent nodes of all connected nodes (up to last level except namespace/undefined)
|
|
Id nodeParentNodeId = getLastVisibleParentNodeId(nodeId);
|
|
|
|
std::map<Id, std::vector<EdgeInfo>> connectedParentNodeIds;
|
|
for (const std::pair<Id, std::vector<EdgeInfo>>& p : connectedNodeIds)
|
|
{
|
|
Id parentNodeId = getLastVisibleParentNodeId(p.first);
|
|
|
|
if (parentNodeId != nodeParentNodeId)
|
|
{
|
|
utility::append(connectedParentNodeIds[parentNodeId], p.second);
|
|
}
|
|
}
|
|
|
|
// add hierarchies of these parents
|
|
std::vector<Id> nodeIdsToAdd;
|
|
for (const std::pair<Id, std::vector<EdgeInfo>> p : connectedParentNodeIds)
|
|
{
|
|
const Id aggregationTargetNodeId = p.first;
|
|
if (!graph->getNodeById(aggregationTargetNodeId))
|
|
{
|
|
nodeIdsToAdd.push_back(aggregationTargetNodeId);
|
|
}
|
|
}
|
|
addNodesWithChildrenAndEdgesToGraph(nodeIdsToAdd, std::vector<Id>(), graph);
|
|
|
|
// create aggregation edges between parents and active node
|
|
Node* sourceNode = graph->getNodeById(nodeId);
|
|
for (const std::pair<Id, std::vector<EdgeInfo>> p : connectedParentNodeIds)
|
|
{
|
|
const Id aggregationTargetNodeId = p.first;
|
|
|
|
Node* targetNode = graph->getNodeById(aggregationTargetNodeId);
|
|
if (!targetNode)
|
|
{
|
|
LOG_ERROR("Aggregation target node not present.");
|
|
}
|
|
|
|
std::shared_ptr<TokenComponentAggregation> componentAggregation = std::make_shared<TokenComponentAggregation>();
|
|
for (const EdgeInfo& edgeInfo: p.second)
|
|
{
|
|
componentAggregation->addAggregationId(edgeInfo.edgeId, edgeInfo.forward);
|
|
}
|
|
|
|
Edge* edge = graph->createEdge(
|
|
*componentAggregation->getAggregationIds().begin(),
|
|
Edge::EDGE_AGGREGATION,
|
|
sourceNode,
|
|
targetNode
|
|
);
|
|
|
|
edge->addComponentAggregation(componentAggregation);
|
|
}
|
|
}
|
|
|
|
void PersistentStorage::addComponentAccessToGraph(Graph* graph) const
|
|
{
|
|
TRACE();
|
|
|
|
std::vector<Id> memberEdgeIds;
|
|
|
|
graph->forEachEdge(
|
|
[&memberEdgeIds](Edge* edge)
|
|
{
|
|
if (!edge->isType(Edge::EDGE_MEMBER))
|
|
{
|
|
return;
|
|
}
|
|
|
|
memberEdgeIds.push_back(edge->getId());
|
|
}
|
|
);
|
|
|
|
std::vector<StorageComponentAccess> accesses = m_sqliteStorage.getComponentAccessByMemberEdgeIds(memberEdgeIds);
|
|
for (const StorageComponentAccess& access : accesses)
|
|
{
|
|
if (access.memberEdgeId && access.type)
|
|
{
|
|
graph->getEdgeById(access.memberEdgeId)->addComponentAccess(
|
|
std::make_shared<TokenComponentAccess>(TokenComponentAccess::intToType(access.type)));
|
|
}
|
|
}
|
|
}
|
|
|
|
void PersistentStorage::buildSearchIndex()
|
|
{
|
|
TRACE();
|
|
|
|
for (StorageNode node : m_sqliteStorage.getAllNodes())
|
|
{
|
|
if (intToDefinitionType(node.definitionType) != DEFINITION_IMPLICIT)
|
|
{
|
|
m_elementIndex.addNode(node.id, NameHierarchy::deserialize(node.serializedName));
|
|
}
|
|
}
|
|
m_elementIndex.finishSetup();
|
|
}
|
|
|
|
void PersistentStorage::buildFilePathMaps()
|
|
{
|
|
TRACE();
|
|
|
|
for (StorageFile file: m_sqliteStorage.getAllFiles())
|
|
{
|
|
m_fileNodeIds.emplace(file.filePath, file.id);
|
|
m_fileNodePaths.emplace(file.id, file.filePath);
|
|
}
|
|
}
|
|
|
|
void PersistentStorage::buildFullTextSearchIndex() const
|
|
{
|
|
TRACE();
|
|
|
|
for (StorageFile file : m_sqliteStorage.getAllFiles())
|
|
{
|
|
m_fullTextSearchIndex.addFile(file.id, m_sqliteStorage.getFileContentById(file.id)->getText());
|
|
}
|
|
}
|
|
|
|
void PersistentStorage::buildHierarchyCache()
|
|
{
|
|
TRACE();
|
|
|
|
std::vector<StorageEdge> memberEdges = m_sqliteStorage.getEdgesByType(Edge::typeToInt(Edge::EDGE_MEMBER));
|
|
|
|
Cache<Id, Node::NodeType> nodeTypeCache([this](Id id){
|
|
return Node::intToType(m_sqliteStorage.getNodeById(id).type);
|
|
});
|
|
|
|
for (const StorageEdge& edge : memberEdges)
|
|
{
|
|
bool isVisible = !(nodeTypeCache.getValue(edge.sourceNodeId) & Node::NODE_NOT_VISIBLE);
|
|
m_hierarchyCache.createConnection(edge.id, edge.sourceNodeId, edge.targetNodeId, isVisible);
|
|
}
|
|
}
|
|
|