TestInterfaceAsyncIterableDouble.cpp (3562B)
1 /* -*- Mode: C++; tab-width: 8; indent-tabs-mode: nil; c-basic-offset: 2 -*- */ 2 /* vim: set ts=8 sts=2 et sw=2 tw=80: */ 3 /* This Source Code Form is subject to the terms of the Mozilla Public 4 * License, v. 2.0. If a copy of the MPL was not distributed with this file, 5 * You can obtain one at http://mozilla.org/MPL/2.0/. */ 6 7 #include "mozilla/dom/TestInterfaceAsyncIterableDouble.h" 8 9 #include "mozilla/dom/BindingUtils.h" 10 #include "mozilla/dom/TestInterfaceJSMaplikeSetlikeIterableBinding.h" 11 #include "nsPIDOMWindow.h" 12 13 namespace mozilla::dom { 14 15 NS_IMPL_CYCLE_COLLECTION_WRAPPERCACHE(TestInterfaceAsyncIterableDouble, mParent) 16 17 NS_IMPL_CYCLE_COLLECTING_ADDREF(TestInterfaceAsyncIterableDouble) 18 NS_IMPL_CYCLE_COLLECTING_RELEASE(TestInterfaceAsyncIterableDouble) 19 20 NS_INTERFACE_MAP_BEGIN_CYCLE_COLLECTION(TestInterfaceAsyncIterableDouble) 21 NS_WRAPPERCACHE_INTERFACE_MAP_ENTRY 22 NS_INTERFACE_MAP_ENTRY(nsISupports) 23 NS_INTERFACE_MAP_END 24 25 TestInterfaceAsyncIterableDouble::TestInterfaceAsyncIterableDouble( 26 nsPIDOMWindowInner* aParent) 27 : mParent(aParent) { 28 mValues.AppendElement(std::pair<nsString, nsString>(u"a"_ns, u"b"_ns)); 29 mValues.AppendElement(std::pair<nsString, nsString>(u"c"_ns, u"d"_ns)); 30 mValues.AppendElement(std::pair<nsString, nsString>(u"e"_ns, u"f"_ns)); 31 } 32 33 // static 34 already_AddRefed<TestInterfaceAsyncIterableDouble> 35 TestInterfaceAsyncIterableDouble::Constructor(const GlobalObject& aGlobal, 36 ErrorResult& aRv) { 37 nsCOMPtr<nsPIDOMWindowInner> window = 38 do_QueryInterface(aGlobal.GetAsSupports()); 39 if (!window) { 40 aRv.Throw(NS_ERROR_FAILURE); 41 return nullptr; 42 } 43 44 RefPtr<TestInterfaceAsyncIterableDouble> r = 45 new TestInterfaceAsyncIterableDouble(window); 46 return r.forget(); 47 } 48 49 JSObject* TestInterfaceAsyncIterableDouble::WrapObject( 50 JSContext* aCx, JS::Handle<JSObject*> aGivenProto) { 51 return TestInterfaceAsyncIterableDouble_Binding::Wrap(aCx, this, aGivenProto); 52 } 53 54 nsPIDOMWindowInner* TestInterfaceAsyncIterableDouble::GetParentObject() const { 55 return mParent; 56 } 57 58 already_AddRefed<Promise> 59 TestInterfaceAsyncIterableDouble::GetNextIterationResult(Iterator* aIterator, 60 ErrorResult& aRv) { 61 RefPtr<Promise> promise = Promise::Create(mParent->AsGlobal(), aRv); 62 if (NS_WARN_IF(aRv.Failed())) { 63 return nullptr; 64 } 65 66 NS_DispatchToMainThread(NewRunnableMethod<RefPtr<Iterator>, RefPtr<Promise>>( 67 "TestInterfaceAsyncIterableDouble::GetNextIterationResult", this, 68 &TestInterfaceAsyncIterableDouble::ResolvePromise, aIterator, promise)); 69 70 return promise.forget(); 71 } 72 73 void TestInterfaceAsyncIterableDouble::ResolvePromise(Iterator* aIterator, 74 Promise* aPromise) { 75 IteratorData& data = aIterator->Data(); 76 77 // Test data: ['a', 'b'], ['c', 'd'], ['e', 'f'] 78 uint32_t idx = data.mIndex; 79 if (idx >= mValues.Length()) { 80 iterator_utils::ResolvePromiseForFinished(aPromise); 81 } else { 82 switch (aIterator->GetIteratorType()) { 83 case IterableIteratorBase::IteratorType::Keys: 84 aPromise->MaybeResolve(mValues[idx].first); 85 break; 86 case IterableIteratorBase::IteratorType::Values: 87 aPromise->MaybeResolve(mValues[idx].second); 88 break; 89 case IterableIteratorBase::IteratorType::Entries: 90 iterator_utils::ResolvePromiseWithKeyAndValue( 91 aPromise, mValues[idx].first, mValues[idx].second); 92 break; 93 } 94 95 data.mIndex++; 96 } 97 } 98 99 } // namespace mozilla::dom