Source code
Revision control
Copy as Markdown
Other Tools
/* -*- 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,
#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