/* * Copyright (c) Meta Platforms, Inc. and affiliates. * * Licensed under the Apache License, Version 2.0 (the "License"); * you may not use this file except in compliance with the License. * You may obtain a copy of the License at * * http://www.apache.org/licenses/LICENSE-2.0 * * Unless required by applicable law or agreed to in writing, software * distributed under the License is distributed on an "AS IS" BASIS, * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. * See the License for the specific language governing permissions and * limitations under the License. */ #include #include #include #include #include namespace folly { TimedDrivableExecutor::TimedDrivableExecutor() = default; TimedDrivableExecutor::~TimedDrivableExecutor() noexcept { // Drain on destruction so that if work is added here during the collapse // of a future train, it will propagate. drain(); } void TimedDrivableExecutor::add(Func callback) { queue_.enqueue(std::move(callback)); } void TimedDrivableExecutor::drive() noexcept { wait(); run(); } bool TimedDrivableExecutor::try_drive() noexcept { return try_wait() && run() > 0; } size_t TimedDrivableExecutor::run() noexcept { size_t count = 0; size_t n = queue_.size(); // If we have waited already, then func_ may have a value if (func_) { auto f = std::move(func_); f(); count = 1; } while (count < n && queue_.try_dequeue(func_)) { auto f = std::move(func_); f(); ++count; } return count; } size_t TimedDrivableExecutor::drain() noexcept { size_t tasksRun = 0; size_t tasksForSingleRun = 0; while ((tasksForSingleRun = run()) != 0) { tasksRun += tasksForSingleRun; } return tasksRun; } void TimedDrivableExecutor::wait() noexcept { if (!func_) { queue_.dequeue(func_); } } bool TimedDrivableExecutor::try_wait() noexcept { return func_ || queue_.try_dequeue(func_); } } // namespace folly