mirror of
https://github.com/zhigang1992/react-native.git
synced 2026-06-18 20:26:01 +08:00
25a58d7bbb
Summary: Replace `folly::SharedMutex` with `better::shared_mutex`, consolidate the shared_mutex. cc. shergin . [General] [Changed] - Consolidate shared_mutex with better::shared_mutex Pull Request resolved: https://github.com/facebook/react-native/pull/24075 Differential Revision: D14559213 Pulled By: shergin fbshipit-source-id: 934c7cd7db9ce60031d6b007faeebb353860268f
44 lines
1.1 KiB
C++
44 lines
1.1 KiB
C++
// Copyright (c) Facebook, Inc. and its affiliates.
|
|
|
|
// This source code is licensed under the MIT license found in the
|
|
// LICENSE file in the root directory of this source tree.
|
|
|
|
#include "ShadowTreeRegistry.h"
|
|
|
|
namespace facebook {
|
|
namespace react {
|
|
|
|
void ShadowTreeRegistry::add(std::unique_ptr<ShadowTree> &&shadowTree) const {
|
|
std::unique_lock<better::shared_mutex> lock(mutex_);
|
|
|
|
registry_.emplace(shadowTree->getSurfaceId(), std::move(shadowTree));
|
|
}
|
|
|
|
std::unique_ptr<ShadowTree> ShadowTreeRegistry::remove(
|
|
SurfaceId surfaceId) const {
|
|
std::unique_lock<better::shared_mutex> lock(mutex_);
|
|
|
|
auto iterator = registry_.find(surfaceId);
|
|
auto shadowTree = std::unique_ptr<ShadowTree>(iterator->second.release());
|
|
registry_.erase(iterator);
|
|
return shadowTree;
|
|
}
|
|
|
|
bool ShadowTreeRegistry::visit(
|
|
SurfaceId surfaceId,
|
|
std::function<void(const ShadowTree &shadowTree)> callback) const {
|
|
std::shared_lock<better::shared_mutex> lock(mutex_);
|
|
|
|
auto iterator = registry_.find(surfaceId);
|
|
|
|
if (iterator == registry_.end()) {
|
|
return false;
|
|
}
|
|
|
|
callback(*iterator->second);
|
|
return true;
|
|
}
|
|
|
|
} // namespace react
|
|
} // namespace facebook
|