File indexing completed on 2024-09-08 12:12:31

 /*
     This file is part of the KDE Baloo project.
     SPDX-FileCopyrightText: 2015 Vishesh Handa <vhanda@kde.org>
 
     SPDX-License-Identifier: LGPL-2.1-or-later
 */
 
 #ifndef BALOO_DBSTATE_H
 #define BALOO_DBSTATE_H
 
 #include "transaction.h"
 #include "postingdb.h"
 #include "documentdb.h"
 #include "documenturldb.h"
 #include "documentiddb.h"
 #include "documentdatadb.h"
 #include "positiondb.h"
 #include "documenttimedb.h"
 
 namespace Baloo {
 
 class DBState {
 public:
     QMap<QByteArray, PostingList> postingDb;
     QMap<QByteArray, QVector<PositionInfo>> positionDb;
 
     QMap<quint64, QVector<QByteArray>> docTermsDb;
     QMap<quint64, QVector<QByteArray>> docFileNameTermsDb;
     QMap<quint64, QVector<QByteArray>> docXAttrTermsDb;
 
     QMap<quint64, DocumentTimeDB::TimeInfo> docTimeDb;
     QMap<quint32, quint64> mtimeDb;
 
     QMap<quint64, QByteArray> docDataDb;
     QMap<quint64, QByteArray> docUrlDb;
     QVector<quint64> contentIndexingDb;
     QVector<quint64> failedIdDb;
 
     bool operator== (const DBState& st) const {
         return postingDb == st.postingDb && positionDb == st.positionDb && docTermsDb == st.docTermsDb
                && docFileNameTermsDb == st.docFileNameTermsDb && docXAttrTermsDb == st.docXAttrTermsDb
                && docTimeDb == st.docTimeDb && mtimeDb == st.mtimeDb && docDataDb == st.docDataDb
                && docUrlDb == st.docUrlDb && contentIndexingDb == st.contentIndexingDb
                && failedIdDb == st.failedIdDb;
     }
 
     static DBState fromTransaction(Transaction* tr);
     static bool debugCompare(const DBState& st1, const DBState& st2);
 private:
 };
 
 DBState DBState::fromTransaction(Baloo::Transaction* tr)
 {
     auto dbis = tr->m_dbis;
     MDB_txn* txn = tr->m_txn;
 
     PostingDB postingDB(dbis.postingDbi, txn);
     PositionDB positionDB(dbis.positionDBi, txn);
     DocumentDB documentTermsDB(dbis.docTermsDbi, txn);
     DocumentDB documentXattrTermsDB(dbis.docXattrTermsDbi, txn);
     DocumentDB documentFileNameTermsDB(dbis.docFilenameTermsDbi, txn);
     DocumentTimeDB docTimeDB(dbis.docTimeDbi, txn);
     DocumentDataDB docDataDB(dbis.docDataDbi, txn);
     DocumentIdDB contentIndexingDB(dbis.contentIndexingDbi, txn);
     DocumentIdDB failedIdDb(dbis.failedIdDbi, txn);
     MTimeDB mtimeDB(dbis.mtimeDbi, txn);
     DocumentUrlDB docUrlDB(dbis.idTreeDbi, dbis.idFilenameDbi, txn);
 
     DBState state;
     state.postingDb = postingDB.toTestMap();
     state.positionDb = positionDB.toTestMap();
     state.docTermsDb = documentTermsDB.toTestMap();
     state.docXAttrTermsDb = documentXattrTermsDB.toTestMap();
     state.docFileNameTermsDb = documentFileNameTermsDB.toTestMap();
     state.docTimeDb = docTimeDB.toTestMap();
     state.docDataDb = docDataDB.toTestMap();
     state.mtimeDb = mtimeDB.toTestMap();
     state.contentIndexingDb = contentIndexingDB.toTestVector();
     state.failedIdDb = failedIdDb.toTestVector();
 
     // FIXME: What about DocumentUrlDB?
     // state.docUrlDb = docUrlDB.toTestMap();
 
     return state;
 }
 
 bool DBState::debugCompare(const DBState& st1, const DBState& st2)
 {
     if (st1.postingDb != st2.postingDb) {
         qDebug() << "Posting DB different";
         qDebug() << st1.postingDb;
         qDebug() << st2.postingDb;
         return false;
     }
 
     if (st1.positionDb != st2.positionDb) {
         qDebug() << "Position DB different";
         qDebug() << st1.positionDb;
         qDebug() << st2.positionDb;
         return false;
     }
 
     if (st1.docTermsDb != st2.docTermsDb) {
         qDebug() << "DocTerms DB different";
         qDebug() << st1.docTermsDb;
         qDebug() << st2.docTermsDb;
         return false;
     }
 
     if (st1.docFileNameTermsDb != st2.docFileNameTermsDb) {
         qDebug() << "Doc FileName Terms DB different";
         qDebug() << st1.docFileNameTermsDb;
         qDebug() << st2.docFileNameTermsDb;
         return false;
     }
 
     if (st1.docXAttrTermsDb != st2.docXAttrTermsDb) {
         qDebug() << "Doc XAttr Terms DB different";
         qDebug() << st1.docXAttrTermsDb;
         qDebug() << st2.docXAttrTermsDb;
         return false;
     }
 
     if (st1.docTimeDb != st2.docTimeDb) {
         qDebug() << "Doc Time DB different";
         qDebug() << st1.docTimeDb;
         qDebug() << st2.docTimeDb;
         return false;
     }
 
     if (st1.mtimeDb != st2.mtimeDb) {
         qDebug() << "MTime DB different";
         qDebug() << st1.mtimeDb;
         qDebug() << st2.mtimeDb;
         return false;
     }
 
     return st1 == st2;
 }
 } // namespace
 
 #endif