mirror of
https://github.com/facebook/react-native.git
synced 2025-11-01 09:14:26 +00:00
Samuel Susla
841756b150
Summary: Changelog: [internal] Implement `RuntimeScheduler::getCurrentPriorityLevel`. JavaScript implementation: https://github.com/facebook/react/blob/master/packages/scheduler/src/forks/SchedulerNoDOM.js#L63 Reviewed By: ShikaSD Differential Revision: D27998510 fbshipit-source-id: 634c09185f9eae8f7afcdb6acd9b74effd587da7
77 lines
2.2 KiB
C++
77 lines
2.2 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 "RuntimeScheduler.h"
|
|
|
|
namespace facebook::react {
|
|
|
|
RuntimeScheduler::RuntimeScheduler(RuntimeExecutor const &runtimeExecutor)
|
|
: RuntimeScheduler(runtimeExecutor, RuntimeSchedulerClock::now) {}
|
|
|
|
RuntimeScheduler::RuntimeScheduler(
|
|
RuntimeExecutor const &runtimeExecutor,
|
|
std::function<RuntimeSchedulerTimePoint()> now)
|
|
: runtimeExecutor_(runtimeExecutor), now_(now) {}
|
|
|
|
std::shared_ptr<Task> RuntimeScheduler::scheduleTask(
|
|
SchedulerPriority priority,
|
|
jsi::Function callback) {
|
|
auto expirationTime = now() + timeoutForSchedulerPriority(priority);
|
|
auto task =
|
|
std::make_shared<Task>(priority, std::move(callback), expirationTime);
|
|
taskQueue_.push(task);
|
|
|
|
if (!isCallbackScheduled_) {
|
|
isCallbackScheduled_ = true;
|
|
runtimeExecutor_([this](jsi::Runtime &runtime) {
|
|
isCallbackScheduled_ = false;
|
|
auto previousPriority = currentPriority_;
|
|
while (!taskQueue_.empty()) {
|
|
auto topPriorityTask = taskQueue_.top();
|
|
auto now = now_();
|
|
auto didUserCallbackTimeout = topPriorityTask->expirationTime <= now;
|
|
|
|
if (!didUserCallbackTimeout && shouldYield_) {
|
|
// This task hasn't expired and we need to yield.
|
|
break;
|
|
}
|
|
currentPriority_ = topPriorityTask->priority;
|
|
auto result = topPriorityTask->execute(runtime);
|
|
|
|
if (result.isObject() &&
|
|
result.getObject(runtime).isFunction(runtime)) {
|
|
topPriorityTask->callback =
|
|
result.getObject(runtime).getFunction(runtime);
|
|
} else {
|
|
taskQueue_.pop();
|
|
}
|
|
}
|
|
currentPriority_ = previousPriority;
|
|
});
|
|
}
|
|
|
|
return task;
|
|
}
|
|
|
|
void RuntimeScheduler::cancelTask(const std::shared_ptr<Task> &task) {
|
|
task->callback.reset();
|
|
}
|
|
|
|
bool RuntimeScheduler::getShouldYield() const {
|
|
return shouldYield_;
|
|
}
|
|
|
|
SchedulerPriority RuntimeScheduler::getCurrentPriorityLevel() const {
|
|
return currentPriority_;
|
|
}
|
|
|
|
RuntimeSchedulerTimePoint RuntimeScheduler::now() const {
|
|
return now_();
|
|
}
|
|
|
|
} // namespace facebook::react
|