diff --git a/libsearch/index/search-manager.cpp b/libsearch/index/search-manager.cpp index 09edbf1..7ca066d 100644 --- a/libsearch/index/search-manager.cpp +++ b/libsearch/index/search-manager.cpp @@ -18,14 +18,13 @@ * */ #include "search-manager.h" - using namespace Zeeker; -size_t SearchManager::uniqueSymbol1 = 0; -size_t SearchManager::uniqueSymbol2 = 0; -size_t SearchManager::uniqueSymbol3 = 0; -QMutex SearchManager::m_mutex1; -QMutex SearchManager::m_mutex2; -QMutex SearchManager::m_mutex3; +size_t SearchManager::uniqueSymbolFile = 0; +size_t SearchManager::uniqueSymbolDir = 0; +size_t SearchManager::uniqueSymbolContent = 0; +QMutex SearchManager::m_mutexFile; +QMutex SearchManager::m_mutexDir; +QMutex SearchManager::m_mutexContent; SearchManager::SearchManager(QObject *parent) : QObject(parent) { m_pool.setMaxThreadCount(3); m_pool.setExpiryTimeout(1000); @@ -46,31 +45,31 @@ int SearchManager::getCurrentIndexCount() { void SearchManager::onKeywordSearch(QString keyword, QQueue *searchResultFile, QQueue *searchResultDir, QQueue> *searchResultContent) { - m_mutex1.lock(); - ++uniqueSymbol1; - m_mutex1.unlock(); - m_mutex2.lock(); - ++uniqueSymbol2; - m_mutex2.unlock(); - m_mutex3.lock(); - ++uniqueSymbol3; - m_mutex3.unlock(); + m_mutexFile.lock(); + ++uniqueSymbolFile; + m_mutexFile.unlock(); + m_mutexDir.lock(); + ++uniqueSymbolDir; + m_mutexDir.unlock(); + m_mutexContent.lock(); + ++uniqueSymbolContent; + m_mutexContent.unlock(); if(FileUtils::SearchMethod::DIRECTSEARCH == FileUtils::searchMethod) { DirectSearch *directSearch; - directSearch = new DirectSearch(keyword, searchResultFile, searchResultDir, uniqueSymbol1); + directSearch = new DirectSearch(keyword, searchResultFile, searchResultDir, uniqueSymbolFile); m_pool.start(directSearch); } else if(FileUtils::SearchMethod::INDEXSEARCH == FileUtils::searchMethod) { FileSearch *filesearch; - filesearch = new FileSearch(searchResultFile, uniqueSymbol1, keyword, "0", 1, 0, 5); + filesearch = new FileSearch(searchResultFile, uniqueSymbolFile, keyword, "0", 1, 0, 5); m_pool.start(filesearch); FileSearch *dirsearch; - dirsearch = new FileSearch(searchResultDir, uniqueSymbol2, keyword, "1", 1, 0, 5); + dirsearch = new FileSearch(searchResultDir, uniqueSymbolDir, keyword, "1", 1, 0, 5); m_pool.start(dirsearch); FileContentSearch *contentSearch; - contentSearch = new FileContentSearch(searchResultContent, uniqueSymbol3, keyword, 0, 5); + contentSearch = new FileContentSearch(searchResultContent, uniqueSymbolContent, keyword, 0, 5); m_pool.start(contentSearch); } else { qWarning() << "Unknown search method! FileUtils::searchMethod: " << static_cast(FileUtils::searchMethod); @@ -104,9 +103,20 @@ FileSearch::~FileSearch() { } void FileSearch::run() { - if(!m_search_result->isEmpty()) { - m_search_result->clear(); + if(m_value == "0") { + SearchManager::m_mutexFile.lock(); + if(!m_search_result->isEmpty()) { + m_search_result->clear(); + } + SearchManager::m_mutexFile.unlock(); + } else if(m_value == "1") { + SearchManager::m_mutexDir.lock(); + if(!m_search_result->isEmpty()) { + m_search_result->clear(); + } + SearchManager::m_mutexDir.unlock(); } + int resultCount = 0; int total = 0; while(total < 100) { @@ -187,23 +197,23 @@ int FileSearch::getResult(Xapian::MSet &result) { } else { switch(m_value.toInt()) { case 1: - SearchManager::m_mutex1.lock(); - if(m_uniqueSymbol == SearchManager::uniqueSymbol2) { + SearchManager::m_mutexDir.lock(); + if(m_uniqueSymbol == SearchManager::uniqueSymbolDir) { m_search_result->enqueue(path); - SearchManager::m_mutex1.unlock(); + SearchManager::m_mutexDir.unlock(); } else { - SearchManager::m_mutex1.unlock(); + SearchManager::m_mutexDir.unlock(); return -1; } break; case 0: - SearchManager::m_mutex2.lock(); - if(m_uniqueSymbol == SearchManager::uniqueSymbol1) { + SearchManager::m_mutexFile.lock(); + if(m_uniqueSymbol == SearchManager::uniqueSymbolFile) { m_search_result->enqueue(path); - SearchManager::m_mutex2.unlock(); + SearchManager::m_mutexFile.unlock(); } else { - SearchManager::m_mutex2.unlock(); + SearchManager::m_mutexFile.unlock(); return -1; } break; @@ -233,9 +243,11 @@ FileContentSearch::~FileContentSearch() { } void FileContentSearch::run() { + SearchManager::m_mutexContent.lock(); if(!m_search_result->isEmpty()) { m_search_result->clear(); } + SearchManager::m_mutexContent.unlock(); int resultCount = 0; int total = 0; @@ -375,14 +387,14 @@ int FileContentSearch::getResult(Xapian::MSet &result, std::string &keyWord) { // QString().swap(snippet); // } - SearchManager::m_mutex3.lock(); - if(m_uniqueSymbol == SearchManager::uniqueSymbol3) { + SearchManager::m_mutexContent.lock(); + if(m_uniqueSymbol == SearchManager::uniqueSymbolContent) { m_search_result->enqueue(qMakePair(path, snippets)); - SearchManager::m_mutex3.unlock(); + SearchManager::m_mutexContent.unlock(); snippets.clear(); QStringList().swap(snippets); } else { - SearchManager::m_mutex3.unlock(); + SearchManager::m_mutexContent.unlock(); return -1; } // searchResult.insert(path,snippets); @@ -432,16 +444,16 @@ void DirectSearch::run() { bfs.enqueue(i.absoluteFilePath()); } if(i.fileName().contains(m_keyword, Qt::CaseInsensitive)) { - SearchManager::m_mutex1.lock(); + SearchManager::m_mutexFile.lock(); // qWarning() << i.fileName() << m_keyword; - if(m_uniqueSymbol == SearchManager::uniqueSymbol1) { + if(m_uniqueSymbol == SearchManager::uniqueSymbolFile) { // TODO if(i.isDir() && m_searchResultDir->length() < 51) { m_searchResultDir->enqueue(i.absoluteFilePath()); } else if(m_searchResultFile->length() < 51) { m_searchResultFile->enqueue(i.absoluteFilePath()); } - SearchManager::m_mutex1.unlock(); + SearchManager::m_mutexFile.unlock(); if(m_searchResultDir->length() > 49 && m_searchResultFile->length() > 49) { return; } @@ -450,7 +462,7 @@ void DirectSearch::run() { // More suitable method? m_searchResultFile->clear(); m_searchResultDir->clear(); - SearchManager::m_mutex1.unlock(); + SearchManager::m_mutexFile.unlock(); return; } } diff --git a/libsearch/index/search-manager.h b/libsearch/index/search-manager.h index c715306..4e0c5da 100644 --- a/libsearch/index/search-manager.h +++ b/libsearch/index/search-manager.h @@ -57,12 +57,12 @@ public: static int getCurrentIndexCount(); - static size_t uniqueSymbol1; - static size_t uniqueSymbol2; - static size_t uniqueSymbol3; - static QMutex m_mutex1; - static QMutex m_mutex2; - static QMutex m_mutex3; + static size_t uniqueSymbolFile; + static size_t uniqueSymbolDir; + static size_t uniqueSymbolContent; + static QMutex m_mutexFile; + static QMutex m_mutexDir; + static QMutex m_mutexContent; public Q_SLOTS: void onKeywordSearch(QString keyword, QQueue *searchResultFile, QQueue *searchResultDir, QQueue> *searchResultContent); diff --git a/src/search-result.cpp b/src/search-result.cpp index fdf5d7e..00e87ca 100644 --- a/src/search-result.cpp +++ b/src/search-result.cpp @@ -44,31 +44,31 @@ void SearchResult::run() { int emptyLists = 0; while(!isInterruptionRequested()) { emptyLists = 0; - m_mainwindow->m_searcher->m_mutex1.lock(); + m_mainwindow->m_searcher->m_mutexFile.lock(); if(!m_mainwindow->m_search_result_file->isEmpty()) { Q_EMIT this->searchResultFile(m_mainwindow->m_search_result_file->dequeue()); - m_mainwindow->m_searcher->m_mutex1.unlock(); + m_mainwindow->m_searcher->m_mutexFile.unlock(); } else { emptyLists ++; - m_mainwindow->m_searcher->m_mutex1.unlock(); + m_mainwindow->m_searcher->m_mutexFile.unlock(); } - m_mainwindow->m_searcher->m_mutex2.lock(); + m_mainwindow->m_searcher->m_mutexDir.lock(); if(!m_mainwindow->m_search_result_dir->isEmpty()) { Q_EMIT this->searchResultDir(m_mainwindow->m_search_result_dir->dequeue()); - m_mainwindow->m_searcher->m_mutex2.unlock(); + m_mainwindow->m_searcher->m_mutexDir.unlock(); } else { emptyLists ++; - m_mainwindow->m_searcher->m_mutex2.unlock(); + m_mainwindow->m_searcher->m_mutexDir.unlock(); } - m_mainwindow->m_searcher->m_mutex3.lock(); + m_mainwindow->m_searcher->m_mutexContent.lock(); // if (!m_mainwindow->m_search_result_content->isEmpty()) // qDebug() << m_mainwindow->m_search_result_content->head(); if(!m_mainwindow->m_search_result_content->isEmpty()) { Q_EMIT this->searchResultContent(m_mainwindow->m_search_result_content->dequeue()); - m_mainwindow->m_searcher->m_mutex3.unlock(); + m_mainwindow->m_searcher->m_mutexContent.unlock(); } else { emptyLists ++; - m_mainwindow->m_searcher->m_mutex3.unlock(); + m_mainwindow->m_searcher->m_mutexContent.unlock(); } if(m_timer->isActive() && m_timer->remainingTime() < 0.01) { this->requestInterruption();