Differential D18664 Diff 50706 src/engine/idfilenamedb.cpp

Changeset View

Standalone View

View Options

src/engine/idfilenamedb.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 "idfilenamedb.h" 21​#include "idfilenamedb.h"
22​#include "enginedebug.h"
22 23​
23using namespace Baloo; 24​using namespace Baloo;
24 25​
25IdFilenameDB::IdFilenameDB(MDB_dbi dbi, MDB_txn* txn) 26​IdFilenameDB::IdFilenameDB(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​
33IdFilenameDB::~IdFilenameDB() 34​IdFilenameDB::~IdFilenameDB()
34{ 35​{
35} 36​}
36 37​
37MDB_dbi IdFilenameDB::create(MDB_txn* txn) 38​MDB_dbi IdFilenameDB::create(MDB_txn* txn)
38{ 39​{
39 MDB_dbi dbi; 40​ MDB_dbi dbi = 0;
40 int rc = mdb_dbi_open(txn, "idfilename", MDB_CREATE | MDB_INTEGERKEY, &dbi); 41​ int rc = mdb_dbi_open(txn, "idfilename", MDB_CREATE | MDB_INTEGERKEY, &dbi);
41 Q_ASSERT_X(rc == 0, "IdFilenameDB::create", mdb_strerror(rc)); 42​ if (rc) {
43​ qCDebug(ENGINE) << "IdFilenameDB::create" << mdb_strerror(rc);
44​ return 0;
45​ }
42 46​
43 return dbi; 47​ return dbi;
44} 48​}
45 49​
46MDB_dbi IdFilenameDB::open(MDB_txn* txn) 50​MDB_dbi IdFilenameDB::open(MDB_txn* txn)
47{ 51​{
48 MDB_dbi dbi; 52​ MDB_dbi dbi = 0;
49 int rc = mdb_dbi_open(txn, "idfilename", MDB_INTEGERKEY, &dbi); 53​ int rc = mdb_dbi_open(txn, "idfilename", MDB_INTEGERKEY, &dbi);
50 if (rc == MDB_NOTFOUND) { 54​ if (rc) {
55​ qCDebug(ENGINE) << "IdFilenameDB::open" << mdb_strerror(rc);
51 return 0; 56​ return 0;
52 } 57​ }
53 Q_ASSERT_X(rc == 0, "IdFilenameDB::open", mdb_strerror(rc)); ​
54 58​
55 return dbi; 59​ return dbi;
56} 60​}
57 61​
58void IdFilenameDB::put(quint64 docId, const FilePath& path) 62​void IdFilenameDB::put(quint64 docId, const FilePath& path)
59{ 63​{
60 Q_ASSERT(docId > 0); 64​ Q_ASSERT(docId > 0);
61 Q_ASSERT(!path.name.isEmpty()); 65​ Q_ASSERT(!path.name.isEmpty());
Show All 19 Lines84{
81 Q_ASSERT(docId > 0); 85​ Q_ASSERT(docId > 0);
82 86​
83 MDB_val key; 87​ MDB_val key;
84 key.mv_size = sizeof(quint64); 88​ key.mv_size = sizeof(quint64);
85 key.mv_data = static_cast<void*>(&docId); 89​ key.mv_data = static_cast<void*>(&docId);
86 90​
87 FilePath path; 91​ FilePath path;
88 92​
89 MDB_val val; 93​ MDB_val val{0, nullptr};
90 int rc = mdb_get(m_txn, m_dbi, &key, &val); 94​ int rc = mdb_get(m_txn, m_dbi, &key, &val);
91 if (rc == MDB_NOTFOUND) { 95​ if (rc) {
96​ qCDebug(ENGINE) << "IdfilenameDB::get" << docId << mdb_strerror(rc);
92 return path; 97​ return path;
93 } 98​ }
94 Q_ASSERT_X(rc == 0, "IdfilenameDB::get", mdb_strerror(rc)); ​
95 99​
96 path.parentId = static_cast<quint64*>(val.mv_data)[0]; 100​ path.parentId = static_cast<quint64*>(val.mv_data)[0];
97 path.name = QByteArray(static_cast<char*>(val.mv_data) + 8, val.mv_size - 8); 101​ path.name = QByteArray(static_cast<char*>(val.mv_data) + 8, val.mv_size - 8);
98 102​
99 return path; 103​ return path;
100} 104​}
101 105​
102bool IdFilenameDB::contains(quint64 docId) 106​bool IdFilenameDB::contains(quint64 docId)
103{ 107​{
104 Q_ASSERT(docId > 0); 108​ Q_ASSERT(docId > 0);
105 109​
106 MDB_val key; 110​ MDB_val key;
107 key.mv_size = sizeof(quint64); 111​ key.mv_size = sizeof(quint64);
108 key.mv_data = static_cast<void*>(&docId); 112​ key.mv_data = static_cast<void*>(&docId);
109 113​
110 MDB_val val; 114​ MDB_val val{0, nullptr};
111 int rc = mdb_get(m_txn, m_dbi, &key, &val); 115​ int rc = mdb_get(m_txn, m_dbi, &key, &val);
112 if (rc == MDB_NOTFOUND) { 116​ if (rc) {
117​ qCDebug(ENGINE) << "IdfilenameDB::contains" << docId << mdb_strerror(rc);
113 return false; 118​ return false;
114 } 119​ }
115 Q_ASSERT_X(rc == 0, "IdfilenameDB::contains", mdb_strerror(rc)); ​
116 return true; 120​ return true;
117} 121​}
118 122​
119void IdFilenameDB::del(quint64 docId) 123​void IdFilenameDB::del(quint64 docId)
120{ 124​{
121 Q_ASSERT(docId > 0); 125​ Q_ASSERT(docId > 0);
122 126​
123 MDB_val key; 127​ MDB_val key;
Show All 10 Lines136{
134 mdb_cursor_open(m_txn, m_dbi, &cursor); 138​ mdb_cursor_open(m_txn, m_dbi, &cursor);
135 139​
136 MDB_val key = {0, nullptr}; 140​ MDB_val key = {0, nullptr};
137 MDB_val val; 141​ MDB_val val;
138 142​
139 QMap<quint64, FilePath> map; 143​ QMap<quint64, FilePath> map;
140 while (1) { 144​ while (1) {
141 int rc = mdb_cursor_get(cursor, &key, &val, MDB_NEXT); 145​ int rc = mdb_cursor_get(cursor, &key, &val, MDB_NEXT);
142 if (rc == MDB_NOTFOUND) { 146​ if (rc) {
143 break; 147​ break;
144 } 148​ }
145 Q_ASSERT_X(rc == 0, "IdFilenameDB::toTestMap", mdb_strerror(rc)); 149​ Q_ASSERT_X(rc == 0, "IdFilenameDB::toTestMap", mdb_strerror(rc));
146 150​
147 const quint64 id = *(static_cast<quint64*>(key.mv_data)); 151​ const quint64 id = *(static_cast<quint64*>(key.mv_data));
148 152​
149 FilePath path; 153​ FilePath path;
150 path.parentId = static_cast<quint64*>(val.mv_data)[0]; 154​ path.parentId = static_cast<quint64*>(val.mv_data)[0];
151 path.name = QByteArray(static_cast<char*>(val.mv_data) + 8, val.mv_size - 8); 155​ path.name = QByteArray(static_cast<char*>(val.mv_data) + 8, val.mv_size - 8);
152 156​
153 map.insert(id, path); 157​ map.insert(id, path);
154 } 158​ }
155 159​
156 mdb_cursor_close(cursor); 160​ mdb_cursor_close(cursor);
157 return map; 161​ return map;
158} 162​}