storage.h


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64

#pragma once

#include <memory>
#include <string>
#include <tuple>

#include <SQLiteCpp/SQLiteCpp.h>

#include "config.h"
#include "compiledsql.h"

class Storage
{
public:
 Storage(const Config& config);
 ~Storage();

 uint64_t getNumberOfDocuments();
 uint64_t dbsize_gross();
 uint64_t dbsize_net();
 bool exists(const std::string& id);

 std::string getDocument(const std::string& id);
 int getRevision(const std::string& id);
 int getCursorPos(const std::string& id);
 std::tuple<std::string, int, int> getRow(const std::string& id);
 
 void setDocument(const std::string& id, const std::string& document);
 void setRevision(const std::string& id, int rev);
 void setCursorPos(const std::string& id, int cursorPos);
 void setRow(const std::string& id, const std::string& document, int rev, int cursorPos);

 void touchDocument(const std::string& id);
 
 void cleanup();

 std::string generate_id();

private:
 SQLite::Database m_db;
 uint64_t m_maxage;

 // shared_ptr to work around initialization in constructor
 CompiledSQL m_stmt_create;
 CompiledSQL m_stmt_getNumberOfDocuments;
 CompiledSQL m_stmt_cleanup;
 CompiledSQL m_stmt_vacuum;
 CompiledSQL m_stmt_exists;
 CompiledSQL m_stmt_getDocument;
 CompiledSQL m_stmt_getRevision;
 CompiledSQL m_stmt_getCursorPos;
 CompiledSQL m_stmt_getRow;
 CompiledSQL m_stmt_setDocument;
 CompiledSQL m_stmt_setDocument_new;
 CompiledSQL m_stmt_setRevision;
 CompiledSQL m_stmt_setCursorPos;
 CompiledSQL m_stmt_setRow;
 CompiledSQL m_stmt_getDbSizeGross;
 CompiledSQL m_stmt_getDbSizeNet;
 CompiledSQL m_stmt_touchDocument;
};

uint32_t checksum32(const std::string& s);