mirror of
https://github.com/esphome/esphome.git
synced 2024-12-03 20:24:14 +01:00
86c0e6114f
* Cosmetic fixes to scheduler code * Add generic Mutex API * Lock scheduler items while modifying them * Always defer MQTT callbacks on Arduino
84 lines
2.4 KiB
C++
84 lines
2.4 KiB
C++
#pragma once
|
|
|
|
#include <vector>
|
|
#include <memory>
|
|
|
|
#include "esphome/core/component.h"
|
|
#include "esphome/core/helpers.h"
|
|
|
|
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);
|
|
|
|
void set_retry(Component *component, const std::string &name, uint32_t initial_wait_time, uint8_t max_attempts,
|
|
std::function<RetryResult(uint8_t)> func, float backoff_increase_factor = 1.0f);
|
|
bool cancel_retry(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()> callback;
|
|
bool remove;
|
|
uint8_t last_execution_major;
|
|
|
|
inline uint32_t next_execution() { return this->last_execution + this->timeout; }
|
|
inline uint8_t next_execution_major() {
|
|
uint32_t next_exec = this->next_execution();
|
|
uint8_t next_exec_major = this->last_execution_major;
|
|
if (next_exec < this->last_execution)
|
|
next_exec_major++;
|
|
return next_exec_major;
|
|
}
|
|
|
|
static bool cmp(const std::unique_ptr<SchedulerItem> &a, const std::unique_ptr<SchedulerItem> &b);
|
|
const char *get_type_str() {
|
|
switch (this->type) {
|
|
case SchedulerItem::INTERVAL:
|
|
return "interval";
|
|
case SchedulerItem::TIMEOUT:
|
|
return "timeout";
|
|
default:
|
|
return "";
|
|
}
|
|
}
|
|
};
|
|
|
|
uint32_t millis_();
|
|
void cleanup_();
|
|
void pop_raw_();
|
|
void push_(std::unique_ptr<SchedulerItem> item);
|
|
bool cancel_item_(Component *component, const std::string &name, SchedulerItem::Type type);
|
|
bool empty_() {
|
|
this->cleanup_();
|
|
return this->items_.empty();
|
|
}
|
|
|
|
Mutex lock_;
|
|
std::vector<std::unique_ptr<SchedulerItem>> items_;
|
|
std::vector<std::unique_ptr<SchedulerItem>> to_add_;
|
|
uint32_t last_millis_{0};
|
|
uint8_t millis_major_{0};
|
|
uint32_t to_remove_{0};
|
|
};
|
|
|
|
} // namespace esphome
|