amis-rpc-design/node_modules/react-native/ReactCommon/react/renderer/leakchecker/WeakFamilyRegistry.h
2023-10-07 19:42:30 +08:00

40 lines
971 B
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/renderer/core/ReactPrimitives.h>
#include <react/renderer/core/ShadowNodeFamily.h>
#include <unordered_map>
#include <vector>
namespace facebook {
namespace react {
class WeakFamilyRegistry final {
public:
using WeakFamilies = std::vector<ShadowNodeFamily::Weak>;
void add(ShadowNodeFamily::Shared const &shadowNodeFamily) const;
void removeFamiliesWithSurfaceId(SurfaceId surfaceId) const;
WeakFamilies weakFamiliesForSurfaceId(SurfaceId surfaceId) const;
private:
/**
* Mutex protecting `families_` property.
*/
mutable std::mutex familiesMutex_;
/**
* A map of ShadowNodeFamily used on surface.
*/
mutable std::unordered_map<SurfaceId, WeakFamilies> families_{};
};
} // namespace react
} // namespace facebook