rckrdmrd
97f407c661
- Prefijo v2: MCH - TRACEABILITY-MASTER.yml creado - Listo para integracion como submodulo Workspace: v2.0.0 | SIMCO: v4.0.0
111 lines
2.8 KiB
C++
111 lines
2.8 KiB
C++
/*
|
|
* Copyright (c) Meta Platforms, Inc. and affiliates.
|
|
*
|
|
* This source code is licensed under the MIT license found in the
|
|
* LICENSE file in the root directory of this source tree.
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include <react/bridging/Error.h>
|
|
#include <react/bridging/Function.h>
|
|
#include <react/bridging/LongLivedObject.h>
|
|
|
|
#include <mutex>
|
|
#include <optional>
|
|
|
|
namespace facebook::react {
|
|
|
|
template <typename... T>
|
|
class AsyncPromise {
|
|
static_assert(sizeof...(T) <= 1, "AsyncPromise must have at most one argument");
|
|
|
|
public:
|
|
AsyncPromise(jsi::Runtime &rt, const std::shared_ptr<CallInvoker> &jsInvoker)
|
|
: state_(std::make_shared<SharedState>())
|
|
{
|
|
auto constructor = rt.global().getPropertyAsFunction(rt, "Promise");
|
|
|
|
auto promise = constructor.callAsConstructor(
|
|
rt,
|
|
bridging::toJs(
|
|
rt,
|
|
// Safe to capture this since this is called synchronously.
|
|
[this](AsyncCallback<T...> resolve, const AsyncCallback<Error> &reject) {
|
|
state_->resolve = std::move(resolve);
|
|
state_->reject = std::move(reject);
|
|
},
|
|
jsInvoker));
|
|
|
|
auto promiseHolder = std::make_shared<PromiseHolder>(rt, promise.asObject(rt));
|
|
LongLivedObjectCollection::get(rt).add(promiseHolder);
|
|
|
|
// The shared state can retain the promise holder weakly now.
|
|
state_->promiseHolder = promiseHolder;
|
|
}
|
|
|
|
void resolve(T... value)
|
|
{
|
|
std::lock_guard<std::mutex> lock(state_->mutex);
|
|
|
|
if (state_->resolve) {
|
|
state_->resolve->call(std::forward<T>(value)...);
|
|
state_->resolve.reset();
|
|
state_->reject.reset();
|
|
}
|
|
}
|
|
|
|
void reject(Error error)
|
|
{
|
|
std::lock_guard<std::mutex> lock(state_->mutex);
|
|
|
|
if (state_->reject) {
|
|
state_->reject->call(std::move(error));
|
|
state_->reject.reset();
|
|
state_->resolve.reset();
|
|
}
|
|
}
|
|
|
|
jsi::Object get(jsi::Runtime &rt) const
|
|
{
|
|
if (auto holder = state_->promiseHolder.lock()) {
|
|
return jsi::Value(rt, holder->promise).asObject(rt);
|
|
} else {
|
|
throw jsi::JSError(rt, "Failed to get invalidated promise");
|
|
}
|
|
}
|
|
|
|
private:
|
|
struct PromiseHolder : LongLivedObject {
|
|
PromiseHolder(jsi::Runtime &runtime, jsi::Object p) : LongLivedObject(runtime), promise(std::move(p)) {}
|
|
|
|
jsi::Object promise;
|
|
};
|
|
|
|
struct SharedState {
|
|
~SharedState()
|
|
{
|
|
if (auto holder = promiseHolder.lock()) {
|
|
holder->allowRelease();
|
|
}
|
|
}
|
|
|
|
std::mutex mutex;
|
|
std::weak_ptr<PromiseHolder> promiseHolder;
|
|
std::optional<AsyncCallback<T...>> resolve;
|
|
std::optional<AsyncCallback<Error>> reject;
|
|
};
|
|
|
|
std::shared_ptr<SharedState> state_;
|
|
};
|
|
|
|
template <typename... T>
|
|
struct Bridging<AsyncPromise<T...>> {
|
|
static jsi::Object toJs(jsi::Runtime &rt, const AsyncPromise<T...> &promise)
|
|
{
|
|
return promise.get(rt);
|
|
}
|
|
};
|
|
|
|
} // namespace facebook::react
|