local_storage.h

Go to the documentation of this file.

#pragma once
 
#include <unordered_map>
 
#include "copy_constructors.h"
#include "local_table.h"
 
namespace kuzu {
namespace main {
class ClientContext;
} // namespace main
namespace storage {
 
class WAL;
// Data structures in LocalStorage are not thread-safe.
// For now, we only support single thread insertions and updates. Once we optimize them with
// multiple threads, LocalStorage and its related data structures should be reworked to be
// thread-safe.
class LocalStorage {
public:
 enum class NotExistAction { CREATE, RETURN_NULL };
 
 explicit LocalStorage(main::ClientContext& clientContext) : clientContext{clientContext} {}
 DELETE_COPY_AND_MOVE(LocalStorage);
 
 LocalTable* getLocalTable(common::table_id_t tableID,
 NotExistAction action = NotExistAction::RETURN_NULL);
 
 void commit();
 void rollback();
 
 uint64_t getEstimatedMemUsage() const;
 
private:
 main::ClientContext& clientContext;
 std::unordered_map<common::table_id_t, std::unique_ptr<LocalTable>> tables;
};
 
} // namespace storage
} // namespace kuzu