TaskTree: Hide Storage internals in cpp

Change-Id: Iaa37aabd7e982302569291612b21a879da11c1a1
Reviewed-by: hjk <hjk@qt.io>
This commit is contained in:
Jarek Kobus
2023-11-07 00:28:06 +01:00
parent 0938e6cf4d
commit 89ab44c32d
2 changed files with 80 additions and 87 deletions

View File

@@ -5,11 +5,15 @@
#include <QEventLoop>
#include <QFutureWatcher>
#include <QHash>
#include <QMutex>
#include <QPromise>
#include <QPointer>
#include <QSet>
#include <QTimer>
#include <atomic>
using namespace std::chrono;
namespace Tasking {
@@ -842,23 +846,64 @@ static DoneResult toDoneResult(DoneWith doneWith)
return doneWith == DoneWith::Success ? DoneResult::Success : DoneResult::Error;
}
bool TreeStorageBase::isValid() const
class StorageThreadData
{
return m_storageData && m_storageData->m_constructor && m_storageData->m_destructor;
Q_DISABLE_COPY_MOVE(StorageThreadData)
public:
StorageThreadData(StorageData *storageData);
~StorageThreadData();
int createStorage();
bool deleteStorage(int id); // Returns true if it was the last storage.
void activateStorage(int id);
void *activeStorageVoid() const;
int activeStorageId() const { return m_activeStorage; }
private:
StorageData *m_storageData = nullptr;
QHash<int, void *> m_storageHash;
int m_activeStorage = 0; // 0 means no active storage.
};
class StorageData
{
public:
~StorageData();
StorageThreadData &threadData() {
QMutexLocker lock(&m_threadDataMutex);
return m_threadDataMap.emplace(QThread::currentThread(), this).first->second;
}
TreeStorageBase::TreeStorageBase(StorageConstructor ctor, StorageDestructor dtor)
: m_storageData(new StorageData{ctor, dtor})
{}
void deleteStorage(int id)
{
if (!threadData().deleteStorage(id))
return;
TreeStorageBase::ThreadData::ThreadData(TreeStorageBase::StorageData *storageData)
QMutexLocker lock(&m_threadDataMutex);
m_threadDataMap.erase(
m_threadDataMap.find(QThread::currentThread()));
}
const TreeStorageBase::StorageConstructor m_constructor = {};
const TreeStorageBase::StorageDestructor m_destructor = {};
QMutex m_threadDataMutex = {};
// Use std::map on purpose, so that it doesn't invalidate references on modifications.
// Don't optimize it by using std::unordered_map.
std::map<QThread *, StorageThreadData> m_threadDataMap = {};
std::atomic_int m_storageInstanceCounter = 0; // Bumped on each creation.
};
StorageThreadData::StorageThreadData(StorageData *storageData)
: m_storageData(storageData)
{
QTC_CHECK(m_storageData->m_constructor);
QTC_CHECK(m_storageData->m_destructor);
}
TreeStorageBase::ThreadData::~ThreadData()
StorageThreadData::~StorageThreadData()
{
QTC_CHECK(m_storageHash.isEmpty());
// TODO: Issue a warning about the leak instead?
@@ -866,7 +911,7 @@ TreeStorageBase::ThreadData::~ThreadData()
m_storageData->m_destructor(ptr);
}
int TreeStorageBase::ThreadData::createStorage()
int StorageThreadData::createStorage()
{
QTC_ASSERT(m_activeStorage == 0, return 0); // TODO: should be allowed?
const int newId = m_storageData->m_storageInstanceCounter.fetch_add(1) + 1;
@@ -874,7 +919,7 @@ int TreeStorageBase::ThreadData::createStorage()
return newId;
}
bool TreeStorageBase::ThreadData::deleteStorage(int id)
bool StorageThreadData::deleteStorage(int id)
{
QTC_ASSERT(m_activeStorage == 0, return false); // TODO: should be allowed?
const auto it = m_storageHash.constFind(id);
@@ -884,24 +929,8 @@ bool TreeStorageBase::ThreadData::deleteStorage(int id)
return m_storageHash.empty();
}
int TreeStorageBase::createStorage() const
{
return threadData().createStorage();
}
void TreeStorageBase::deleteStorage(int id) const
{
if (!threadData().deleteStorage(id))
return;
QMutexLocker lock(&m_storageData->m_threadDataMutex);
m_storageData->m_threadDataMap.erase(
m_storageData->m_threadDataMap.find(QThread::currentThread()));
}
// passing 0 deactivates currently active storage
void TreeStorageBase::ThreadData::activateStorage(int id)
void StorageThreadData::activateStorage(int id)
{
if (id == 0) {
QTC_ASSERT(m_activeStorage, return);
@@ -915,7 +944,7 @@ void TreeStorageBase::ThreadData::activateStorage(int id)
m_activeStorage = id;
}
void *TreeStorageBase::ThreadData::activeStorageVoid() const
void *StorageThreadData::activeStorageVoid() const
{
QTC_ASSERT(m_activeStorage, qWarning(
"The referenced storage is not reachable in the running tree. "
@@ -927,22 +956,24 @@ void *TreeStorageBase::ThreadData::activeStorageVoid() const
return it.value();
}
int TreeStorageBase::ThreadData::activeStorageId() const
{
return m_activeStorage;
}
TreeStorageBase::StorageData::~StorageData()
StorageData::~StorageData()
{
QMutexLocker lock(&m_threadDataMutex);
QTC_CHECK(m_threadDataMap.empty());
}
TreeStorageBase::ThreadData &TreeStorageBase::threadData() const
bool TreeStorageBase::isValid() const
{
QMutexLocker lock(&m_storageData->m_threadDataMutex);
return m_storageData->m_threadDataMap.emplace(QThread::currentThread(),
m_storageData.get()).first->second;
return m_storageData && m_storageData->m_constructor && m_storageData->m_destructor;
}
TreeStorageBase::TreeStorageBase(StorageConstructor ctor, StorageDestructor dtor)
: m_storageData(new StorageData{ctor, dtor})
{}
void *TreeStorageBase::activeStorageVoid() const
{
return m_storageData->threadData().activeStorageVoid();
}
void GroupItem::addChildren(const QList<GroupItem> &children)
@@ -1058,7 +1089,7 @@ public:
const StorageHandler storageHandler = *it;
// TODO: we don't necessarily need to activate the storage here, it's enough to
// get a pointer to the relevant storage instance.
auto &threadData = storage.threadData();
auto &threadData = storage.m_storageData->threadData();
threadData.activateStorage(storageId);
if (storageHandler.*ptr)
(storageHandler.*ptr)(threadData.activeStorageVoid());
@@ -1212,7 +1243,7 @@ public:
ExecutionContextActivator(TaskContainer *container) { activateContext(container); }
~ExecutionContextActivator() {
for (int i = m_activeStorages.size() - 1; i >= 0; --i) // iterate in reverse order
m_activeStorages[i].threadData().activateStorage(0);
m_activeStorages[i].m_storageData->threadData().activateStorage(0);
}
private:
@@ -1222,7 +1253,7 @@ private:
const TaskContainer::ConstData &constData = container->m_constData;
for (int i = 0; i < constData.m_storageList.size(); ++i) {
const TreeStorageBase &storage = constData.m_storageList[i];
auto &threadData = storage.threadData();
auto &threadData = storage.m_storageData->threadData();
if (threadData.activeStorageId())
continue; // Storage shadowing: The storage is already active, skipping it...
m_activeStorages.append(storage);
@@ -1276,7 +1307,7 @@ QList<int> TaskContainer::RuntimeData::createStorages(const TaskContainer::Const
{
QList<int> storageIdList;
for (const TreeStorageBase &storage : constData.m_storageList) {
const int storageId = storage.createStorage();
const int storageId = storage.m_storageData->threadData().createStorage();
storageIdList.append(storageId);
constData.m_taskTreePrivate->callSetupHandler(storage, storageId);
}
@@ -1313,7 +1344,7 @@ TaskContainer::RuntimeData::~RuntimeData()
const int storageId = m_storageIdList.value(i);
if (m_callStorageDoneHandlersOnDestruction)
m_constData.m_taskTreePrivate->callDoneHandler(storage, storageId);
storage.deleteStorage(storageId);
storage.m_storageData->deleteStorage(storageId);
}
}

View File

@@ -5,12 +5,9 @@
#include "tasking_global.h"
#include <QHash>
#include <QMutex>
#include <QObject>
#include <QSharedPointer>
#include <atomic>
#include <memory>
QT_BEGIN_NAMESPACE
@@ -23,6 +20,7 @@ namespace Tasking {
Q_NAMESPACE_EXPORT(TASKING_EXPORT)
class ExecutionContextActivator;
class StorageData;
class TaskContainer;
class TaskTreePrivate;
@@ -46,40 +44,15 @@ protected:
class TASKING_EXPORT TreeStorageBase
{
public:
bool isValid() const;
private:
using StorageConstructor = std::function<void *(void)>;
using StorageDestructor = std::function<void(void *)>;
TreeStorageBase(StorageConstructor ctor, StorageDestructor dtor);
struct StorageData;
struct TASKING_EXPORT ThreadData
{
Q_DISABLE_COPY_MOVE(ThreadData)
ThreadData(StorageData *storageData);
~ThreadData();
int createStorage();
bool deleteStorage(int id); // Returns true if it was the last storage.
void activateStorage(int id);
void *activeStorageVoid() const;
int activeStorageId() const;
bool isValid() const;
private:
StorageData *m_storageData = nullptr;
QHash<int, void *> m_storageHash;
int m_activeStorage = 0; // 0 means no active storage.
};
TreeStorageBase(StorageConstructor ctor, StorageDestructor dtor);
ThreadData &threadData() const;
int createStorage() const;
void deleteStorage(int id) const;
void *activeStorageVoid() const;
friend bool operator==(const TreeStorageBase &first, const TreeStorageBase &second)
{ return first.m_storageData == second.m_storageData; }
@@ -90,17 +63,6 @@ private:
friend size_t qHash(const TreeStorageBase &storage, uint seed = 0)
{ return size_t(storage.m_storageData.get()) ^ seed; }
struct StorageData
{
~StorageData();
const StorageConstructor m_constructor = {};
const StorageDestructor m_destructor = {};
QMutex m_threadDataMutex = {};
// Use std::map on purpose, so that it doesn't invalidate references on modifications.
// Don't optimize it by using std::unordered_map.
std::map<QThread *, ThreadData> m_threadDataMap = {};
std::atomic_int m_storageInstanceCounter = 0; // Bumped on each creation.
};
QSharedPointer<StorageData> m_storageData;
template <typename StorageStruct> friend class TreeStorage;
@@ -117,7 +79,7 @@ public:
StorageStruct &operator*() const noexcept { return *activeStorage(); }
StorageStruct *operator->() const noexcept { return activeStorage(); }
StorageStruct *activeStorage() const {
return static_cast<StorageStruct *>(threadData().activeStorageVoid()); // HERE
return static_cast<StorageStruct *>(activeStorageVoid());
}
private: