mirror of
https://github.com/facebook/react-native.git
synced 2025-11-01 09:14:26 +00:00
Samuel SuslaandFacebook GitHub Bot
5e933fdd2a
Summary: changelog: [internal] With multiple requests to the runtime, we need to make sure they are all granted before React continues with rendering. A boolean is not enough to track this. The first lambda that has VM will set it to false and subsequent requests will have to wait for React to finish rendering. To prevent this, we can count how many lambdas are pending access to the runtime. Reviewed By: ShikaSD Differential Revision: D33792734 fbshipit-source-id: f785fae3575470179dd69acc6a466211b79b633b
70 lines
1.5 KiB
C++
70 lines
1.5 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 <condition_variable>
|
|
#include <queue>
|
|
|
|
class StubQueue {
|
|
public:
|
|
void runOnQueue(std::function<void()> &&func) {
|
|
{
|
|
std::lock_guard<std::mutex> lock(mutex_);
|
|
callbackQueue_.push(func);
|
|
}
|
|
|
|
signal_.notify_one();
|
|
}
|
|
|
|
void flush() {
|
|
while (size() > 0) {
|
|
tick();
|
|
}
|
|
}
|
|
|
|
void tick() {
|
|
std::function<void()> callback;
|
|
{
|
|
std::lock_guard<std::mutex> lock(mutex_);
|
|
if (!callbackQueue_.empty()) {
|
|
callback = callbackQueue_.front();
|
|
callbackQueue_.pop();
|
|
}
|
|
}
|
|
|
|
if (callback) {
|
|
callback();
|
|
}
|
|
}
|
|
|
|
size_t size() const {
|
|
std::lock_guard<std::mutex> lock(mutex_);
|
|
return callbackQueue_.size();
|
|
}
|
|
|
|
bool waitForTask(std::chrono::duration<double> timeout) const {
|
|
std::unique_lock<std::mutex> lock(mutex_);
|
|
return signal_.wait_for(
|
|
lock, timeout, [this]() { return !callbackQueue_.empty(); });
|
|
}
|
|
|
|
bool waitForTasks(
|
|
std::size_t numberOfTasks,
|
|
std::chrono::duration<double> timeout) const {
|
|
std::unique_lock<std::mutex> lock(mutex_);
|
|
return signal_.wait_for(lock, timeout, [this, numberOfTasks]() {
|
|
return numberOfTasks == callbackQueue_.size();
|
|
});
|
|
}
|
|
|
|
private:
|
|
mutable std::condition_variable signal_;
|
|
mutable std::mutex mutex_;
|
|
std::queue<std::function<void()>> callbackQueue_;
|
|
};
|