Differential D18664 Diff 52216 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​ qCWarning(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​ qCWarning(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​ qCWarning(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​ if (rc != MDB_NOTFOUND) {
92​ qCDebug(ENGINE) << "DocumentTimeDB::get" << docId << mdb_strerror(rc);
93​ }
85 return TimeInfo(); 94​ return TimeInfo();
86 } 95​ }
87 Q_ASSERT_X(rc == 0, "DocumentTimeDB::get", mdb_strerror(rc)); ​
88 96​
89 return *(static_cast<TimeInfo*>(val.mv_data)); 97​ return *(static_cast<TimeInfo*>(val.mv_data));
90} 98​}
91 99​
92void DocumentTimeDB::del(quint64 docId) 100​void DocumentTimeDB::del(quint64 docId)
93{ 101​{
94 Q_ASSERT(docId > 0); 102​ Q_ASSERT(docId > 0);
95 103​
96 MDB_val key; 104​ MDB_val key;
97 key.mv_size = sizeof(quint64); 105​ key.mv_size = sizeof(quint64);
98 key.mv_data = static_cast<void*>(&docId); 106​ key.mv_data = static_cast<void*>(&docId);
99 107​
100 int rc = mdb_del(m_txn, m_dbi, &key, nullptr); 108​ int rc = mdb_del(m_txn, m_dbi, &key, nullptr);
101 if (rc == MDB_NOTFOUND) { 109​ if (rc != 0 && rc != MDB_NOTFOUND) {
102 return; 110​ qCDebug(ENGINE) << "DocumentTimeDB::del" << docId << mdb_strerror(rc);
103 } 111​ }
104 Q_ASSERT_X(rc == 0, "DocumentTimeDB::del", mdb_strerror(rc)); ​
105} 112​}
106 113​
107bool DocumentTimeDB::contains(quint64 docId) 114​bool DocumentTimeDB::contains(quint64 docId)
108{ 115​{
109 Q_ASSERT(docId > 0); 116​ Q_ASSERT(docId > 0);
110 117​
111 MDB_val key; 118​ MDB_val key;
112 key.mv_size = sizeof(quint64); 119​ key.mv_size = sizeof(quint64);
113 key.mv_data = static_cast<void*>(&docId); 120​ key.mv_data = static_cast<void*>(&docId);
114 121​
115 MDB_val val; 122​ MDB_val val;
116 int rc = mdb_get(m_txn, m_dbi, &key, &val); 123​ int rc = mdb_get(m_txn, m_dbi, &key, &val);
117 if (rc == MDB_NOTFOUND) { 124​ if (rc) {
125​ if (rc != MDB_NOTFOUND) {
126​ qCDebug(ENGINE) << "DocumentTimeDB::contains" << docId << mdb_strerror(rc);
127​ }
118 return false; 128​ return false;
119 } 129​ }
120 Q_ASSERT_X(rc == 0, "DocumentTimeDB::contains", mdb_strerror(rc)); ​
121 130​
122 return true; 131​ return true;
123} 132​}
124 133​
125QMap<quint64, DocumentTimeDB::TimeInfo> DocumentTimeDB::toTestMap() const 134​QMap<quint64, DocumentTimeDB::TimeInfo> DocumentTimeDB::toTestMap() const
126{ 135​{
127 MDB_cursor* cursor; 136​ MDB_cursor* cursor;
128 mdb_cursor_open(m_txn, m_dbi, &cursor); 137​ mdb_cursor_open(m_txn, m_dbi, &cursor);
129 138​
130 MDB_val key = {0, nullptr}; 139​ MDB_val key = {0, nullptr};
131 MDB_val val; 140​ MDB_val val;
132 141​
133 QMap<quint64, TimeInfo> map; 142​ QMap<quint64, TimeInfo> map;
134 while (1) { 143​ while (1) {
135 int rc = mdb_cursor_get(cursor, &key, &val, MDB_NEXT); 144​ int rc = mdb_cursor_get(cursor, &key, &val, MDB_NEXT);
136 if (rc == MDB_NOTFOUND) { 145​ if (rc) {
146​ qCDebug(ENGINE) << "DocumentTimeDB::toTestMap" << mdb_strerror(rc);
137 break; 147​ break;
138 } 148​ }
139 Q_ASSERT_X(rc == 0, "DocumentTimeDB::toTestMap", mdb_strerror(rc)); ​
140 149​
141 const quint64 id = *(static_cast<quint64*>(key.mv_data)); 150​ const quint64 id = *(static_cast<quint64*>(key.mv_data));
142 const TimeInfo ti = *(static_cast<TimeInfo*>(val.mv_data)); 151​ const TimeInfo ti = *(static_cast<TimeInfo*>(val.mv_data));
143 map.insert(id, ti); 152​ map.insert(id, ti);
144 } 153​ }
145 154​
146 mdb_cursor_close(cursor); 155​ mdb_cursor_close(cursor);
147 return map; 156​ return map;
148} 157​}