mirror of
https://github.com/mozilla/gecko-dev.git
synced 2024-11-27 14:52:16 +00:00
93 lines
3.2 KiB
C++
93 lines
3.2 KiB
C++
/* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */
|
|
/* vim: set ts=8 sts=2 et sw=2 tw=80: */
|
|
/* This Source Code Form is subject to the terms of the Mozilla Public
|
|
* License, v. 2.0. If a copy of the MPL was not distributed with this file,
|
|
* You can obtain one at http://mozilla.org/MPL/2.0/. */
|
|
|
|
#ifndef DOM_QUOTA_INITIALIZATIONUTILS_H_
|
|
#define DOM_QUOTA_INITIALIZATIONUTILS_H_
|
|
|
|
#include <functional>
|
|
|
|
#include "mozilla/MozPromise.h"
|
|
#include "mozilla/RefPtr.h"
|
|
#include "mozilla/dom/quota/DirectoryLock.h"
|
|
#include "mozilla/dom/quota/DirectoryLockInlines.h"
|
|
#include "mozilla/dom/quota/ForwardDecls.h"
|
|
#include "mozilla/dom/quota/QuotaManager.h"
|
|
|
|
namespace mozilla::dom::quota {
|
|
|
|
template <typename UninitChecker, typename PromiseArrayIter>
|
|
RefPtr<UniversalDirectoryLock> CreateDirectoryLockForInitialization(
|
|
QuotaManager& aQuotaManager, const PersistenceScope& aPersistenceScope,
|
|
const OriginScope& aOriginScope, const bool aAlreadyInitialized,
|
|
UninitChecker&& aUninitChecker, PromiseArrayIter&& aPromiseArrayIter) {
|
|
RefPtr<UniversalDirectoryLock> directoryLock =
|
|
aQuotaManager.CreateDirectoryLockInternal(aPersistenceScope, aOriginScope,
|
|
Nullable<Client::Type>(),
|
|
/* aExclusive */ false);
|
|
|
|
if (aAlreadyInitialized &&
|
|
!std::forward<UninitChecker>(aUninitChecker)(directoryLock)) {
|
|
return nullptr;
|
|
}
|
|
|
|
auto iter = std::forward<PromiseArrayIter>(aPromiseArrayIter);
|
|
*iter = directoryLock->Acquire();
|
|
++iter;
|
|
|
|
return directoryLock;
|
|
}
|
|
|
|
template <typename Callable>
|
|
class MaybeInitializeHelper final {
|
|
public:
|
|
MaybeInitializeHelper(RefPtr<UniversalDirectoryLock> aDirectoryLock,
|
|
Callable&& aCallable)
|
|
: mDirectoryLock(std::move(aDirectoryLock)),
|
|
mCallable(std::move(aCallable)) {}
|
|
|
|
RefPtr<BoolPromise> operator()(
|
|
const BoolPromise::ResolveOrRejectValue& aValue) {
|
|
if (aValue.IsReject()) {
|
|
SafeDropDirectoryLockIfNotDropped(mDirectoryLock);
|
|
|
|
return BoolPromise::CreateAndReject(aValue.RejectValue(), __func__);
|
|
}
|
|
|
|
if (!mDirectoryLock) {
|
|
return BoolPromise::CreateAndResolve(true, __func__);
|
|
}
|
|
|
|
return mCallable(std::move(mDirectoryLock));
|
|
}
|
|
|
|
private:
|
|
RefPtr<UniversalDirectoryLock> mDirectoryLock;
|
|
Callable mCallable;
|
|
};
|
|
|
|
template <typename Callable>
|
|
auto MaybeInitialize(RefPtr<UniversalDirectoryLock> aDirectoryLock,
|
|
Callable&& aCallable) {
|
|
return MaybeInitializeHelper(std::move(aDirectoryLock),
|
|
std::forward<Callable>(aCallable));
|
|
}
|
|
|
|
auto MaybeInitialize(RefPtr<UniversalDirectoryLock> aDirectoryLock,
|
|
RefPtr<QuotaManager> aQuotaManager,
|
|
RefPtr<BoolPromise> (QuotaManager::*aMethod)(
|
|
RefPtr<UniversalDirectoryLock>)) {
|
|
return MaybeInitializeHelper(
|
|
std::move(aDirectoryLock),
|
|
[quotaManager = std::move(aQuotaManager),
|
|
method = aMethod](RefPtr<UniversalDirectoryLock> aDirectoryLock) {
|
|
return (quotaManager->*method)(std::move(aDirectoryLock));
|
|
});
|
|
}
|
|
|
|
} // namespace mozilla::dom::quota
|
|
|
|
#endif // DOM_QUOTA_INITIALIZATIONUTILS_H_
|