62 lines
1.8 KiB
C++
62 lines
1.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 <jsi/jsi.h>
|
|
#include <memory>
|
|
#include <mutex>
|
|
#include <unordered_set>
|
|
|
|
namespace facebook::react {
|
|
|
|
/**
|
|
* A simple wrapper class that can be registered to a collection that keep it
|
|
* alive for extended period of time. This object can be removed from the
|
|
* collection when needed.
|
|
*
|
|
* The subclass of this class must be created using std::make_shared<T>().
|
|
* After creation, add it to the `LongLivedObjectCollection`. When done with the
|
|
* object, call `allowRelease()` to reclaim its memory.
|
|
*
|
|
* When using LongLivedObject to keep JS values alive, ensure you only hold weak
|
|
* references to the object outside the JS thread to avoid accessing deallocated
|
|
* values when the JS VM is shutdown.
|
|
*/
|
|
class LongLivedObject {
|
|
public:
|
|
virtual void allowRelease();
|
|
|
|
protected:
|
|
explicit LongLivedObject(jsi::Runtime& runtime) : runtime_(runtime) {}
|
|
virtual ~LongLivedObject() = default;
|
|
jsi::Runtime& runtime_;
|
|
};
|
|
|
|
/**
|
|
* A singleton, thread-safe, write-only collection for the `LongLivedObject`s.
|
|
*/
|
|
class LongLivedObjectCollection {
|
|
public:
|
|
static LongLivedObjectCollection& get(jsi::Runtime& runtime);
|
|
|
|
LongLivedObjectCollection() = default;
|
|
LongLivedObjectCollection(const LongLivedObjectCollection&) = delete;
|
|
void operator=(const LongLivedObjectCollection&) = delete;
|
|
|
|
void add(std::shared_ptr<LongLivedObject> o);
|
|
void remove(const LongLivedObject* o);
|
|
void clear();
|
|
size_t size() const;
|
|
|
|
private:
|
|
std::unordered_set<std::shared_ptr<LongLivedObject>> collection_;
|
|
mutable std::mutex collectionMutex_;
|
|
};
|
|
|
|
} // namespace facebook::react
|