searchtaskmanager.h

/*
    SPDX-FileCopyrightText: 2013, 2014 Daniel Vrátil <[email protected]>
 
    SPDX-License-Identifier: LGPL-2.0-or-later
*/
 
#pragma once
 
#include "akthread.h"
 
#include "agentmanagerinterface.h"
#include "exception.h"
#include <QMap>
#include <QMutex>
#include <QSet>
#include <QStringList>
#include <QVector>
#include <QWaitCondition>
 
namespace Akonadi
{
namespace Server
{
class Connection;
class AgentSearchInstance;
 
class SearchTask
{
public:
    QByteArray id;
    QString query;
    QStringList mimeTypes;
    QVector<qint64> collections;
    bool complete;
 
    QMutex sharedLock;
    QWaitCondition notifier;
 
    QVector<QPair<QString /* resource */, qint64 /* collection */>> queries;
    QSet<qint64> pendingResults;
};
 
class SearchTaskManager : public AkThread
{
    Q_OBJECT
 
public:
    explicit SearchTaskManager();
    ~SearchTaskManager() override;
 
    void registerInstance(const QString &id);
    void unregisterInstance(const QString &id);
 
    void addTask(SearchTask *task);
 
    void pushResults(const QByteArray &searchId, const QSet<qint64> &ids, Connection *connection);
 
private Q_SLOTS:
    void searchLoop();
 
private:
    class ResourceTask
    {
    public:
        QString resourceId;
        qint64 collectionId;
        SearchTask *parentTask;
        QSet<qint64> results;
 
        qint64 timestamp;
    };
 
    using TasksMap = QMap<QString, ResourceTask *>;
 
    bool mShouldStop;
 
    TasksMap::Iterator cancelRunningTask(TasksMap::Iterator &iter);
    bool allResourceTasksCompleted(SearchTask *agentSearchTask) const;
 
    QMap<QString, AgentSearchInstance *> mInstances;
    QMutex mInstancesLock;
 
    QWaitCondition mWait;
    QMutex mLock;
 
    QVector<SearchTask *> mTasklist;
 
    QMap<QString /* resource */, ResourceTask *> mRunningTasks;
    QVector<ResourceTask *> mPendingResults;
};
 
AKONADI_EXCEPTION_MAKE_INSTANCE(SearchException);
 
} // namespace Server
} // namespace Akonadi