tdlight/tdactor/td/actor/impl/ConcurrentScheduler.h

94 lines
2.4 KiB
C
Raw Normal View History

//
// Copyright Aliaksei Levin (levlam@telegram.org), Arseny Smirnov (arseny30@gmail.com) 2014-2018
//
// Distributed under the Boost Software License, Version 1.0. (See accompanying
// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
//
#pragma once
#include "td/actor/impl/Scheduler-decl.h"
#include "td/utils/common.h"
#include "td/utils/logging.h"
#include "td/utils/port/thread.h"
#include "td/utils/Slice.h"
#include <atomic>
#include <functional>
#include <mutex>
#include <utility>
namespace td {
class ConcurrentScheduler : private Scheduler::Callback {
public:
void init(int32 threads_n);
void finish_async() {
schedulers_[0]->finish();
}
void wakeup() {
schedulers_[0]->wakeup();
}
SchedulerGuard get_current_guard() {
return schedulers_[0]->get_guard();
}
void test_one_thread_run();
bool is_finished() {
return is_finished_.load(std::memory_order_relaxed);
}
void start();
bool run_main(double timeout);
void finish();
template <class ActorT, class... Args>
ActorOwn<ActorT> create_actor_unsafe(int32 sched_id, Slice name, Args &&... args) {
#if TD_THREAD_UNSUPPORTED || TD_EVENTFD_UNSUPPORTED
sched_id = 0;
#endif
CHECK(0 <= sched_id && sched_id < static_cast<int32>(schedulers_.size()));
auto guard = schedulers_[sched_id]->get_guard();
return schedulers_[sched_id]->create_actor<ActorT>(name, std::forward<Args>(args)...);
}
template <class ActorT>
ActorOwn<ActorT> register_actor_unsafe(int32 sched_id, Slice name, ActorT *actor) {
#if TD_THREAD_UNSUPPORTED || TD_EVENTFD_UNSUPPORTED
sched_id = 0;
#endif
CHECK(0 <= sched_id && sched_id < static_cast<int32>(schedulers_.size()));
auto guard = schedulers_[sched_id]->get_guard();
return schedulers_[sched_id]->register_actor<ActorT>(name, actor);
}
private:
enum class State { Start, Run };
State state_;
std::vector<unique_ptr<Scheduler>> schedulers_;
std::atomic<bool> is_finished_;
std::mutex at_finish_mutex_;
std::vector<std::function<void()>> at_finish_;
#if !TD_THREAD_UNSUPPORTED && !TD_EVENTFD_UNSUPPORTED
std::vector<thread> threads_;
#endif
void on_finish() override {
is_finished_.store(true, std::memory_order_relaxed);
for (auto &it : schedulers_) {
it->wakeup();
}
}
void register_at_finish(std::function<void()> f) override {
std::lock_guard<std::mutex> lock(at_finish_mutex_);
at_finish_.push_back(std::move(f));
}
};
} // namespace td