sqlite_checkpoint.h

Go to the documentation of this file.

 
#pragma once
 
#include <neograph/api.h>
#include <neograph/graph/checkpoint.h>
#include <memory>
#include <mutex>
#include <string>
 
// Forward-declare so this header doesn't drag the sqlite3 C header
// into every TU that includes it.
struct sqlite3;
 
namespace neograph::graph {
 
class NEOGRAPH_API SqliteCheckpointStore : public CheckpointStore {
public:
    explicit SqliteCheckpointStore(const std::string& db_path);
 
    ~SqliteCheckpointStore() override;
 
    // sqlite3* is a unique resource — no copying or moving.
    SqliteCheckpointStore(const SqliteCheckpointStore&) = delete;
    SqliteCheckpointStore& operator=(const SqliteCheckpointStore&) = delete;
 
    void save(const Checkpoint& cp) override;
    std::optional<Checkpoint> load_latest(const std::string& thread_id) override;
    std::optional<Checkpoint> load_by_id(const std::string& id) override;
    std::vector<Checkpoint> list(const std::string& thread_id,
                                  int limit = 100) override;
    void delete_thread(const std::string& thread_id) override;
 
    void put_writes(const std::string& thread_id,
                    const std::string& parent_checkpoint_id,
                    const PendingWrite& write) override;
    std::vector<PendingWrite> get_writes(
        const std::string& thread_id,
        const std::string& parent_checkpoint_id) override;
    void clear_writes(const std::string& thread_id,
                      const std::string& parent_checkpoint_id) override;
 
    void drop_schema();
 
    size_t blob_count();
 
private:
    void ensure_schema();
    void exec_ddl(const char* sql);
 
    sqlite3* db_ = nullptr;
    std::mutex db_mutex_;
};
 
} // namespace neograph::graph