tdlight/tdactor/td/actor/SchedulerLocalStorage.h
2024-01-01 03:07:21 +03:00

82 lines
1.8 KiB
C++

//
// Copyright Aliaksei Levin (levlam@telegram.org), Arseny Smirnov (arseny30@gmail.com) 2014-2024
//
// 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/actor.h"
#include "td/utils/common.h"
#include "td/utils/optional.h"
#include <functional>
namespace td {
template <class T>
class SchedulerLocalStorage {
public:
SchedulerLocalStorage() : data_(Scheduler::instance()->sched_count()) {
}
T &get() {
return data_[Scheduler::instance()->sched_id()];
}
template <class F>
void for_each(F &&f) {
for (auto &value : data_) {
f(value);
}
}
template <class F>
void for_each(F &&f) const {
for (const auto &value : data_) {
f(value);
}
}
private:
std::vector<T> data_;
};
template <class T>
class LazySchedulerLocalStorage {
public:
LazySchedulerLocalStorage() = default;
explicit LazySchedulerLocalStorage(std::function<T()> create_func) : create_func_(std::move(create_func)) {
}
void set_create_func(std::function<T()> create_func) {
CHECK(!create_func_);
create_func_ = create_func;
}
void set(T &&t) {
auto &optional_value_ = sls_optional_value_.get();
CHECK(!optional_value_);
optional_value_ = std::move(t);
}
T &get() {
auto &optional_value_ = sls_optional_value_.get();
if (!optional_value_) {
CHECK(create_func_);
optional_value_ = create_func_();
}
return *optional_value_;
}
void clear_values() {
sls_optional_value_.for_each([&](auto &optional_value) {
if (optional_value) {
optional_value = optional<T>();
}
});
}
private:
std::function<T()> create_func_;
SchedulerLocalStorage<optional<T>> sls_optional_value_;
};
} // namespace td