You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
104 lines
2.9 KiB
104 lines
2.9 KiB
/*
|
|
* 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 <jsi/jsi.h>
|
|
#include <react/bridging/LongLivedObject.h>
|
|
|
|
#include <memory>
|
|
|
|
namespace facebook {
|
|
namespace react {
|
|
|
|
// Helper for passing jsi::Function arg to other methods.
|
|
class CallbackWrapper : public LongLivedObject {
|
|
private:
|
|
CallbackWrapper(
|
|
jsi::Function &&callback,
|
|
jsi::Runtime &runtime,
|
|
std::shared_ptr<CallInvoker> jsInvoker)
|
|
: longLivedObjectCollection_(),
|
|
callback_(std::move(callback)),
|
|
runtime_(runtime),
|
|
jsInvoker_(std::move(jsInvoker)) {}
|
|
|
|
CallbackWrapper(
|
|
std::shared_ptr<LongLivedObjectCollection> longLivedObjectCollection,
|
|
jsi::Function &&callback,
|
|
jsi::Runtime &runtime,
|
|
std::shared_ptr<CallInvoker> jsInvoker)
|
|
: longLivedObjectCollection_(longLivedObjectCollection),
|
|
callback_(std::move(callback)),
|
|
runtime_(runtime),
|
|
jsInvoker_(std::move(jsInvoker)) {}
|
|
|
|
// Use a weak_ptr to avoid a retain cycle: LongLivedObjectCollection owns all
|
|
// CallbackWrappers. So, CallbackWrapper cannot own its
|
|
// LongLivedObjectCollection.
|
|
std::weak_ptr<LongLivedObjectCollection> longLivedObjectCollection_;
|
|
jsi::Function callback_;
|
|
jsi::Runtime &runtime_;
|
|
std::shared_ptr<CallInvoker> jsInvoker_;
|
|
|
|
public:
|
|
static std::weak_ptr<CallbackWrapper> createWeak(
|
|
jsi::Function &&callback,
|
|
jsi::Runtime &runtime,
|
|
std::shared_ptr<CallInvoker> jsInvoker) {
|
|
auto wrapper = std::shared_ptr<CallbackWrapper>(
|
|
new CallbackWrapper(std::move(callback), runtime, jsInvoker));
|
|
LongLivedObjectCollection::get().add(wrapper);
|
|
return wrapper;
|
|
}
|
|
|
|
static std::weak_ptr<CallbackWrapper> createWeak(
|
|
std::shared_ptr<LongLivedObjectCollection> longLivedObjectCollection,
|
|
jsi::Function &&callback,
|
|
jsi::Runtime &runtime,
|
|
std::shared_ptr<CallInvoker> jsInvoker) {
|
|
auto wrapper = std::shared_ptr<CallbackWrapper>(new CallbackWrapper(
|
|
longLivedObjectCollection, std::move(callback), runtime, jsInvoker));
|
|
longLivedObjectCollection->add(wrapper);
|
|
return wrapper;
|
|
}
|
|
|
|
// Delete the enclosed jsi::Function
|
|
void destroy() {
|
|
allowRelease();
|
|
}
|
|
|
|
jsi::Function &callback() {
|
|
return callback_;
|
|
}
|
|
|
|
jsi::Runtime &runtime() {
|
|
return runtime_;
|
|
}
|
|
|
|
CallInvoker &jsInvoker() {
|
|
return *(jsInvoker_);
|
|
}
|
|
|
|
std::shared_ptr<CallInvoker> jsInvokerPtr() {
|
|
return jsInvoker_;
|
|
}
|
|
|
|
void allowRelease() override {
|
|
if (auto longLivedObjectCollection = longLivedObjectCollection_.lock()) {
|
|
if (longLivedObjectCollection != nullptr) {
|
|
longLivedObjectCollection->remove(this);
|
|
return;
|
|
}
|
|
}
|
|
LongLivedObject::allowRelease();
|
|
}
|
|
};
|
|
|
|
} // namespace react
|
|
} // namespace facebook
|