malte_langkabel
3d94b78b8c
* database stores modification time of files. * FileManager updates internal state based on database info. * reimplemented FileManager's check if a file is in project. * implemented clearing data of updated or removed files before reparsing the updated and added files.
108 lines
3.6 KiB
C++
108 lines
3.6 KiB
C++
#ifndef SQLITE_STORAGE_H
|
|
#define SQLITE_STORAGE_H
|
|
|
|
#include <memory>
|
|
#include <string>
|
|
#include <vector>
|
|
|
|
#include "sqlite/CppSQLite3.h"
|
|
|
|
#include "utility/file/FilePath.h"
|
|
#include "utility/types.h"
|
|
#include "data/name/NameHierarchy.h"
|
|
#include "data/location/TokenLocationFile.h"
|
|
#include "data/location/TokenLocationCollection.h"
|
|
#include "data/StorageTypes.h"
|
|
|
|
class SqliteStorage
|
|
{
|
|
public:
|
|
SqliteStorage(const std::string& dbFilePath);
|
|
~SqliteStorage();
|
|
|
|
void clear();
|
|
|
|
void beginTransaction();
|
|
void commitTransaction();
|
|
void rollbackTransaction();
|
|
|
|
Id addEdge(int type, Id sourceNodeId, Id targetNodeId);
|
|
Id addNode(int type, Id nameId);
|
|
Id addFile(Id nameId, const std::string& filePath, const std::string& modificationTime);
|
|
int addSourceLocation(Id elementId, Id fileNodeId, uint startLine, uint startCol, uint endLine, uint endCol, bool isScope);
|
|
Id addNameHierarchyElement(const std::string& name);
|
|
Id addNameHierarchyElement(const std::string& name, Id parentId);
|
|
|
|
void removeElement(Id id);
|
|
void removeNameHierarchyElement(Id id);
|
|
void removeElementsWithLocationInFile(Id fileId);
|
|
void removeFile(Id id);
|
|
void removeUnusedNameHierarchyElements();
|
|
|
|
std::vector<StorageNode> getAllNodes() const;
|
|
|
|
bool isEdge(Id elementId) const;
|
|
bool isNode(Id elementId) const;
|
|
bool isFile(Id elementId) const;
|
|
|
|
StorageEdge getEdgeBySourceTargetType(Id sourceId, Id targetId, int type) const;
|
|
std::vector<StorageEdge> getEdgesBySourceId(Id sourceId) const;
|
|
std::vector<StorageEdge> getEdgesByTargetId(Id targetId) const;
|
|
std::vector<StorageEdge> getEdgesBySourceType(Id sourceId, int type) const;
|
|
std::vector<StorageEdge> getEdgesByTargetType(Id targetId, int type) const;
|
|
StorageEdge getEdgeById(Id edgeId) const;
|
|
|
|
StorageNode getNodeById(Id id) const;
|
|
StorageNode getNodeByNameId(Id nameId) const;
|
|
StorageNode getNodeByName(const std::string& nodeName) const; // hmm... we need to use name hierarchy here...??
|
|
|
|
StorageFile getFileById(const Id id) const;
|
|
StorageFile getFileByName(const std::string& fileName) const;
|
|
std::vector<StorageFile> getAllFiles() const;
|
|
|
|
void setNodeType(int type, Id nodeId);
|
|
|
|
Id getNameHierarchyElementIdByName(const std::string& name) const;
|
|
Id getNameHierarchyElementIdByName(const std::string& name, Id parentId) const;
|
|
|
|
Id getNameHierarchyElementIdByNodeId(const Id nodeId) const;
|
|
|
|
std::vector<StorageNameHierarchyElement> getAllNameHierarchyElements() const;
|
|
|
|
NameHierarchy getNameHierarchyById(const Id id) const;
|
|
|
|
StorageSourceLocation getSourceLocationById(const Id id) const;
|
|
std::vector<StorageSourceLocation> getAllSourceLocations() const;
|
|
std::shared_ptr<TokenLocationFile> getTokenLocationsForFile(const FilePath& filePath) const;
|
|
std::vector<StorageSourceLocation> getTokenLocationsForElementId(const Id elementId) const;
|
|
|
|
Id getElementIdByLocationId(Id locationId) const;
|
|
|
|
int getNodeCount() const;
|
|
int getEdgeCount() const;
|
|
int getNameHierarchyElementCount() const;
|
|
|
|
private:
|
|
void clearTables();
|
|
void setupTables();
|
|
|
|
StorageFile getFirstFile(const std::string& query) const;
|
|
std::vector<StorageFile> getAllFiles(const std::string& query) const;
|
|
StorageSourceLocation getFirstSourceLocation(const std::string& query) const;
|
|
std::vector<StorageSourceLocation> getAllSourceLocations(const std::string& query) const;
|
|
|
|
template <typename ResultType>
|
|
ResultType getFirstResult(const std::string& query) const;
|
|
|
|
mutable CppSQLite3DB m_database;
|
|
};
|
|
|
|
template <typename ResultType>
|
|
ResultType SqliteStorage::getFirstResult(const std::string& query) const
|
|
{
|
|
CppSQLite3Query q = m_database.execQuery(query.c_str());
|
|
return q.getIntField(0, 0);
|
|
}
|
|
|
|
#endif // SQLITE_STORAGE_H
|