130 lines
4.7 KiB
C++
130 lines
4.7 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/. */
|
|
|
|
#include "mozilla/dom/TestInterfaceAsyncIterableSingle.h"
|
|
#include "mozilla/dom/TestInterfaceJSMaplikeSetlikeIterableBinding.h"
|
|
#include "nsPIDOMWindow.h"
|
|
#include "mozilla/dom/BindingUtils.h"
|
|
#include "mozilla/dom/IterableIterator.h"
|
|
#include "mozilla/dom/Promise-inl.h"
|
|
#include "nsThreadUtils.h"
|
|
|
|
namespace mozilla::dom {
|
|
|
|
NS_IMPL_CYCLE_COLLECTION_WRAPPERCACHE(TestInterfaceAsyncIterableSingle, mParent)
|
|
|
|
NS_IMPL_CYCLE_COLLECTING_ADDREF(TestInterfaceAsyncIterableSingle)
|
|
NS_IMPL_CYCLE_COLLECTING_RELEASE(TestInterfaceAsyncIterableSingle)
|
|
|
|
NS_INTERFACE_MAP_BEGIN_CYCLE_COLLECTION(TestInterfaceAsyncIterableSingle)
|
|
NS_WRAPPERCACHE_INTERFACE_MAP_ENTRY
|
|
NS_INTERFACE_MAP_ENTRY(nsISupports)
|
|
NS_INTERFACE_MAP_END
|
|
|
|
TestInterfaceAsyncIterableSingle::TestInterfaceAsyncIterableSingle(
|
|
nsPIDOMWindowInner* aParent, bool aFailToInit)
|
|
: mParent(aParent), mFailToInit(aFailToInit) {}
|
|
|
|
// static
|
|
already_AddRefed<TestInterfaceAsyncIterableSingle>
|
|
TestInterfaceAsyncIterableSingle::Constructor(
|
|
const GlobalObject& aGlobal,
|
|
const TestInterfaceAsyncIterableSingleOptions& aOptions, ErrorResult& aRv) {
|
|
nsCOMPtr<nsPIDOMWindowInner> window =
|
|
do_QueryInterface(aGlobal.GetAsSupports());
|
|
if (!window) {
|
|
aRv.Throw(NS_ERROR_FAILURE);
|
|
return nullptr;
|
|
}
|
|
|
|
RefPtr<TestInterfaceAsyncIterableSingle> r =
|
|
new TestInterfaceAsyncIterableSingle(window, aOptions.mFailToInit);
|
|
return r.forget();
|
|
}
|
|
|
|
JSObject* TestInterfaceAsyncIterableSingle::WrapObject(
|
|
JSContext* aCx, JS::Handle<JSObject*> aGivenProto) {
|
|
return TestInterfaceAsyncIterableSingle_Binding::Wrap(aCx, this, aGivenProto);
|
|
}
|
|
|
|
nsPIDOMWindowInner* TestInterfaceAsyncIterableSingle::GetParentObject() const {
|
|
return mParent;
|
|
}
|
|
|
|
void TestInterfaceAsyncIterableSingle::InitAsyncIteratorData(
|
|
IteratorData& aData, Iterator::IteratorType aType, ErrorResult& aError) {
|
|
if (mFailToInit) {
|
|
aError.ThrowTypeError("Caller asked us to fail");
|
|
return;
|
|
}
|
|
|
|
// Nothing else to do.
|
|
MOZ_ASSERT(aData.mIndex == 0);
|
|
MOZ_ASSERT(aData.mMultiplier == 1);
|
|
}
|
|
|
|
already_AddRefed<Promise>
|
|
TestInterfaceAsyncIterableSingle::GetNextIterationResult(Iterator* aIterator,
|
|
ErrorResult& aRv) {
|
|
return GetNextIterationResult(aIterator, aIterator->Data(), aRv);
|
|
}
|
|
|
|
already_AddRefed<Promise>
|
|
TestInterfaceAsyncIterableSingle::GetNextIterationResult(
|
|
IterableIteratorBase* aIterator, IteratorData& aData, ErrorResult& aRv) {
|
|
RefPtr<Promise> promise = Promise::Create(mParent->AsGlobal(), aRv);
|
|
if (NS_WARN_IF(aRv.Failed())) {
|
|
return nullptr;
|
|
}
|
|
|
|
nsCOMPtr<nsIRunnable> callResolvePromise =
|
|
NewRunnableMethod<RefPtr<IterableIteratorBase>, IteratorData&,
|
|
RefPtr<Promise>>(
|
|
"TestInterfaceAsyncIterableSingle::GetNextIterationResult", this,
|
|
&TestInterfaceAsyncIterableSingle::ResolvePromise, aIterator, aData,
|
|
promise);
|
|
if (aData.mBlockingPromisesIndex < aData.mBlockingPromises.Length()) {
|
|
aData.mBlockingPromises[aData.mBlockingPromisesIndex]
|
|
->AddCallbacksWithCycleCollectedArgs(
|
|
[](JSContext* aCx, JS::Handle<JS::Value> aValue, ErrorResult& aRv,
|
|
nsIRunnable* aCallResolvePromise) {
|
|
NS_DispatchToMainThread(aCallResolvePromise);
|
|
},
|
|
[](JSContext* aCx, JS::Handle<JS::Value> aValue, ErrorResult& aRv,
|
|
nsIRunnable* aCallResolvePromise) {},
|
|
std::move(callResolvePromise));
|
|
++aData.mBlockingPromisesIndex;
|
|
} else {
|
|
NS_DispatchToMainThread(callResolvePromise);
|
|
}
|
|
|
|
return promise.forget();
|
|
}
|
|
|
|
void TestInterfaceAsyncIterableSingle::ResolvePromise(
|
|
IterableIteratorBase* aIterator, IteratorData& aData, Promise* aPromise) {
|
|
if (aData.mIndex >= 10) {
|
|
iterator_utils::ResolvePromiseForFinished(aPromise);
|
|
} else {
|
|
aPromise->MaybeResolve(int32_t(aData.mIndex * 9 % 7 * aData.mMultiplier));
|
|
|
|
aData.mIndex++;
|
|
}
|
|
}
|
|
|
|
void TestInterfaceAsyncIterableSingle::IteratorData::Traverse(
|
|
nsCycleCollectionTraversalCallback& cb) {
|
|
TestInterfaceAsyncIterableSingle::IteratorData* tmp = this;
|
|
NS_IMPL_CYCLE_COLLECTION_TRAVERSE(mBlockingPromises);
|
|
NS_IMPL_CYCLE_COLLECTION_TRAVERSE(mThrowFromReturn);
|
|
}
|
|
void TestInterfaceAsyncIterableSingle::IteratorData::Unlink() {
|
|
TestInterfaceAsyncIterableSingle::IteratorData* tmp = this;
|
|
NS_IMPL_CYCLE_COLLECTION_UNLINK(mBlockingPromises);
|
|
NS_IMPL_CYCLE_COLLECTION_UNLINK(mThrowFromReturn);
|
|
}
|
|
|
|
} // namespace mozilla::dom
|