mirror of
https://github.com/esphome/esphome.git
synced 2025-02-18 09:04:43 +01:00
56 lines
1.4 KiB
C++
56 lines
1.4 KiB
C++
#pragma once
|
|
|
|
#include "esphome/core/component.h"
|
|
#include <vector>
|
|
|
|
namespace esphome {
|
|
|
|
class Component;
|
|
|
|
class Scheduler {
|
|
public:
|
|
void set_timeout(Component *component, const std::string &name, uint32_t timeout, std::function<void()> &&func);
|
|
bool cancel_timeout(Component *component, const std::string &name);
|
|
void set_interval(Component *component, const std::string &name, uint32_t interval, std::function<void()> &&func);
|
|
bool cancel_interval(Component *component, const std::string &name);
|
|
|
|
optional<uint32_t> next_schedule_in();
|
|
|
|
void call();
|
|
|
|
void process_to_add();
|
|
|
|
protected:
|
|
struct SchedulerItem {
|
|
Component *component;
|
|
std::string name;
|
|
enum Type { TIMEOUT, INTERVAL } type;
|
|
union {
|
|
uint32_t interval;
|
|
uint32_t timeout;
|
|
};
|
|
uint32_t last_execution;
|
|
std::function<void()> f;
|
|
bool remove;
|
|
uint8_t last_execution_major;
|
|
|
|
static bool cmp(SchedulerItem *a, SchedulerItem *b);
|
|
};
|
|
|
|
uint32_t millis_();
|
|
void cleanup_();
|
|
void pop_raw_();
|
|
void push_(SchedulerItem *item);
|
|
bool cancel_item_(Component *component, const std::string &name, SchedulerItem::Type type);
|
|
bool empty_() {
|
|
this->cleanup_();
|
|
return this->items_.empty();
|
|
}
|
|
|
|
std::vector<SchedulerItem *> items_;
|
|
std::vector<SchedulerItem *> to_add_;
|
|
uint32_t last_millis_{0};
|
|
uint8_t millis_major_{0};
|
|
};
|
|
|
|
} // namespace esphome
|