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