Differential D18664 Diff 51252 src/engine/documenttimedb.cpp

Changeset View

Standalone View

View Options

src/engine/documenttimedb.cpp

Show All 13 Lines
14 * 14​ *
15 * You should have received a copy of the GNU Lesser General Public 15​ * You should have received a copy of the GNU Lesser General Public
16 * License along with this library; if not, write to the Free Software 16​ * License along with this library; if not, write to the Free Software
17 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA 17​ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
18 * 18​ *
19 */ 19​ */
20 20​
21#include "documenttimedb.h" 21​#include "documenttimedb.h"
22​#include "enginedebug.h"
22 23​
23using namespace Baloo; 24​using namespace Baloo;
24 25​
25DocumentTimeDB::DocumentTimeDB(MDB_dbi dbi, MDB_txn* txn) 26​DocumentTimeDB::DocumentTimeDB(MDB_dbi dbi, MDB_txn* txn)
26 : m_txn(txn) 27​ : m_txn(txn)
27 , m_dbi(dbi) 28​ , m_dbi(dbi)
28{ 29​{
29 Q_ASSERT(txn != nullptr); 30​ Q_ASSERT(txn != nullptr);
30 Q_ASSERT(dbi != 0); 31​ Q_ASSERT(dbi != 0);
31} 32​}
32 33​
33DocumentTimeDB::~DocumentTimeDB() 34​DocumentTimeDB::~DocumentTimeDB()
34{ 35​{
35} 36​}
36 37​
37MDB_dbi DocumentTimeDB::create(MDB_txn* txn) 38​MDB_dbi DocumentTimeDB::create(MDB_txn* txn)
38{ 39​{
39 MDB_dbi dbi; 40​ MDB_dbi dbi = 0;
40 int rc = mdb_dbi_open(txn, "documenttimedb", MDB_CREATE | MDB_INTEGERKEY, &dbi); 41​ int rc = mdb_dbi_open(txn, "documenttimedb", MDB_CREATE | MDB_INTEGERKEY, &dbi);
41 Q_ASSERT_X(rc == 0, "DocumentTimeDB::create", mdb_strerror(rc)); 42​ if (rc) {
43​ qCDebug(ENGINE) << "DocumentTimeDB::create" << mdb_strerror(rc);
44​ return 0;
45​ }
42 46​
43 return dbi; 47​ return dbi;
44} 48​}
45 49​
46MDB_dbi DocumentTimeDB::open(MDB_txn* txn) 50​MDB_dbi DocumentTimeDB::open(MDB_txn* txn)
47{ 51​{
48 MDB_dbi dbi; 52​ MDB_dbi dbi = 0;
49 int rc = mdb_dbi_open(txn, "documenttimedb", MDB_INTEGERKEY, &dbi); 53​ int rc = mdb_dbi_open(txn, "documenttimedb", MDB_INTEGERKEY, &dbi);
50 if (rc == MDB_NOTFOUND) { 54​ if (rc) {
55​ qCDebug(ENGINE) << "DocumentTimeDB::open" << mdb_strerror(rc);
51 return 0; 56​ return 0;
52 } 57​ }
53 Q_ASSERT_X(rc == 0, "DocumentTimeDB::create", mdb_strerror(rc)); ​
54 58​
55 return dbi; 59​ return dbi;
56} 60​}
57 61​
58void DocumentTimeDB::put(quint64 docId, const TimeInfo& info) 62​void DocumentTimeDB::put(quint64 docId, const TimeInfo& info)
59{ 63​{
60 Q_ASSERT(docId > 0); 64​ Q_ASSERT(docId > 0);
61 65​
62 MDB_val key; 66​ MDB_val key;
63 key.mv_size = sizeof(quint64); 67​ key.mv_size = sizeof(quint64);
64 key.mv_data = &docId; 68​ key.mv_data = &docId;
65 69​
66 MDB_val val; 70​ MDB_val val;
67 val.mv_size = sizeof(TimeInfo); 71​ val.mv_size = sizeof(TimeInfo);
68 val.mv_data = static_cast<void*>(const_cast<TimeInfo*>(&info)); 72​ val.mv_data = static_cast<void*>(const_cast<TimeInfo*>(&info));
69 73​
70 int rc = mdb_put(m_txn, m_dbi, &key, &val, 0); 74​ int rc = mdb_put(m_txn, m_dbi, &key, &val, 0);
71 Q_ASSERT_X(rc == 0, "DocumentTimeDB::put", mdb_strerror(rc)); 75​ if (rc) {
76​ qCDebug(ENGINE) << "DocumentTimeDB::put" << docId << mdb_strerror(rc);
77​ }
72} 78​}
73 79​
74DocumentTimeDB::TimeInfo DocumentTimeDB::get(quint64 docId) 80​DocumentTimeDB::TimeInfo DocumentTimeDB::get(quint64 docId)
75{ 81​{
76 Q_ASSERT(docId > 0); 82​ Q_ASSERT(docId > 0);
77 83​
78 MDB_val key; 84​ MDB_val key;
79 key.mv_size = sizeof(quint64); 85​ key.mv_size = sizeof(quint64);
80 key.mv_data = &docId; 86​ key.mv_data = &docId;
81 87​
82 MDB_val val; 88​ MDB_val val{0, nullptr};
83 int rc = mdb_get(m_txn, m_dbi, &key, &val); 89​ int rc = mdb_get(m_txn, m_dbi, &key, &val);
84 if (rc == MDB_NOTFOUND) { 90​ if (rc) {
91​ qCDebug(ENGINE) << "DocumentTimeDB::get" << docId << mdb_strerror(rc);
85 return TimeInfo(); 92​ return TimeInfo();
86 } 93​ }
87 Q_ASSERT_X(rc == 0, "DocumentTimeDB::get", mdb_strerror(rc)); ​
88 94​
89 return *(static_cast<TimeInfo*>(val.mv_data)); 95​ return *(static_cast<TimeInfo*>(val.mv_data));
90} 96​}
91 97​
92void DocumentTimeDB::del(quint64 docId) 98​void DocumentTimeDB::del(quint64 docId)
93{ 99​{
94 Q_ASSERT(docId > 0); 100​ Q_ASSERT(docId > 0);
95 101​
96 MDB_val key; 102​ MDB_val key;
97 key.mv_size = sizeof(quint64); 103​ key.mv_size = sizeof(quint64);
98 key.mv_data = static_cast<void*>(&docId); 104​ key.mv_data = static_cast<void*>(&docId);
99 105​
100 int rc = mdb_del(m_txn, m_dbi, &key, nullptr); 106​ int rc = mdb_del(m_txn, m_dbi, &key, nullptr);
101 if (rc == MDB_NOTFOUND) { 107​ if (rc) {
102 return; 108​ qCDebug(ENGINE) << "DocumentTimeDB::del" << docId << mdb_strerror(rc);
103 } 109​ }
104 Q_ASSERT_X(rc == 0, "DocumentTimeDB::del", mdb_strerror(rc)); ​
105} 110​}
106 111​
107bool DocumentTimeDB::contains(quint64 docId) 112​bool DocumentTimeDB::contains(quint64 docId)
108{ 113​{
109 Q_ASSERT(docId > 0); 114​ Q_ASSERT(docId > 0);
110 115​
111 MDB_val key; 116​ MDB_val key;
112 key.mv_size = sizeof(quint64); 117​ key.mv_size = sizeof(quint64);
113 key.mv_data = static_cast<void*>(&docId); 118​ key.mv_data = static_cast<void*>(&docId);
114 119​
115 MDB_val val; 120​ MDB_val val;
116 int rc = mdb_get(m_txn, m_dbi, &key, &val); 121​ int rc = mdb_get(m_txn, m_dbi, &key, &val);
117 if (rc == MDB_NOTFOUND) { 122​ if (rc) {
123​ qCDebug(ENGINE) << "DocumentTimeDB::contains" << docId << mdb_strerror(rc);
118 return false; 124​ return false;
119 } 125​ }
120 Q_ASSERT_X(rc == 0, "DocumentTimeDB::contains", mdb_strerror(rc)); ​
121 126​
122 return true; 127​ return true;
123} 128​}
124 129​
125QMap<quint64, DocumentTimeDB::TimeInfo> DocumentTimeDB::toTestMap() const 130​QMap<quint64, DocumentTimeDB::TimeInfo> DocumentTimeDB::toTestMap() const
126{ 131​{
127 MDB_cursor* cursor; 132​ MDB_cursor* cursor;
128 mdb_cursor_open(m_txn, m_dbi, &cursor); 133​ mdb_cursor_open(m_txn, m_dbi, &cursor);
129 134​
130 MDB_val key = {0, nullptr}; 135​ MDB_val key = {0, nullptr};
131 MDB_val val; 136​ MDB_val val;
132 137​
133 QMap<quint64, TimeInfo> map; 138​ QMap<quint64, TimeInfo> map;
134 while (1) { 139​ while (1) {
135 int rc = mdb_cursor_get(cursor, &key, &val, MDB_NEXT); 140​ int rc = mdb_cursor_get(cursor, &key, &val, MDB_NEXT);
136 if (rc == MDB_NOTFOUND) { 141​ if (rc) {
142​ qCDebug(ENGINE) << "DocumentTimeDB::toTestMap" << mdb_strerror(rc);
137 break; 143​ break;
138 } 144​ }
139 Q_ASSERT_X(rc == 0, "DocumentTimeDB::toTestMap", mdb_strerror(rc)); ​
140 145​
141 const quint64 id = *(static_cast<quint64*>(key.mv_data)); 146​ const quint64 id = *(static_cast<quint64*>(key.mv_data));
142 const TimeInfo ti = *(static_cast<TimeInfo*>(val.mv_data)); 147​ const TimeInfo ti = *(static_cast<TimeInfo*>(val.mv_data));
143 map.insert(id, ti); 148​ map.insert(id, ti);
144 } 149​ }
145 150​
146 mdb_cursor_close(cursor); 151​ mdb_cursor_close(cursor);
147 return map; 152​ return map;
148} 153​}