iw5-mod/src/module/scheduler.cpp

#include <std_include.hpp>
#include "scheduler.hpp"

std::mutex scheduler::mutex_;
std::vector<std::function<void()>> scheduler::callbacks_;
std::vector<std::function<void()>> scheduler::single_callbacks_;

void scheduler::on_frame(const std::function<void()>& callback)
{
	std::lock_guard _(mutex_);
	callbacks_.push_back(callback);
}

void scheduler::once(const std::function<void()>& callback)
{
	std::lock_guard _(mutex_);
	single_callbacks_.push_back(callback);
}

void scheduler::execute()
{
	std::vector<std::function<void()>> callbacks_copy;
	std::vector<std::function<void()>> single_callbacks_copy;

	{
		std::lock_guard _(mutex_);
		callbacks_copy = callbacks_;
		single_callbacks_copy = single_callbacks_;
		single_callbacks_.clear();
	}

	for (const auto& callback : callbacks_copy)
	{
		callback();
	}

	for (const auto& callback : single_callbacks_copy)
	{
		callback();
	}
}

void scheduler::pre_destroy()
{
	std::lock_guard _(mutex_);
	callbacks_.clear();
	single_callbacks_.clear();
}

REGISTER_MODULE(scheduler);