2020-12-05 19:26:32 +08:00
|
|
|
#include "timer.h"
|
|
|
|
#include "arch.h"
|
|
|
|
|
2022-05-15 21:29:34 +08:00
|
|
|
#define MG_TIMER_CALLED 4
|
|
|
|
|
2022-04-12 21:14:55 +08:00
|
|
|
void mg_timer_init(struct mg_timer **head, struct mg_timer *t, uint64_t ms,
|
|
|
|
unsigned flags, void (*fn)(void *), void *arg) {
|
2022-07-17 19:52:18 +08:00
|
|
|
struct mg_timer tmp = {0U, ms, 0U, 0U, flags, fn, arg, *head};
|
2020-12-05 19:26:32 +08:00
|
|
|
*t = tmp;
|
2022-04-12 21:14:55 +08:00
|
|
|
*head = t;
|
2020-12-05 19:26:32 +08:00
|
|
|
}
|
|
|
|
|
2022-04-12 21:14:55 +08:00
|
|
|
void mg_timer_free(struct mg_timer **head, struct mg_timer *t) {
|
2020-12-05 19:26:32 +08:00
|
|
|
while (*head && *head != t) head = &(*head)->next;
|
|
|
|
if (*head) *head = t->next;
|
|
|
|
}
|
|
|
|
|
2022-04-12 21:14:55 +08:00
|
|
|
void mg_timer_poll(struct mg_timer **head, uint64_t now_ms) {
|
2020-12-05 19:26:32 +08:00
|
|
|
// If time goes back (wrapped around), reset timers
|
|
|
|
struct mg_timer *t, *tmp;
|
2022-04-12 21:14:55 +08:00
|
|
|
for (t = *head; t != NULL; t = tmp) {
|
2020-12-05 19:26:32 +08:00
|
|
|
tmp = t->next;
|
2022-04-12 21:14:55 +08:00
|
|
|
if (t->prev_ms > now_ms) t->expire = 0; // Handle time wrap
|
|
|
|
t->prev_ms = now_ms;
|
2022-05-15 21:29:34 +08:00
|
|
|
if (t->expire == 0 && (t->flags & MG_TIMER_RUN_NOW) &&
|
|
|
|
!(t->flags & MG_TIMER_CALLED)) {
|
|
|
|
// Handle MG_TIMER_NOW only once
|
|
|
|
} else if (t->expire == 0) {
|
|
|
|
t->expire = now_ms + t->period_ms;
|
|
|
|
}
|
2020-12-05 19:26:32 +08:00
|
|
|
if (t->expire > now_ms) continue;
|
2022-05-15 21:29:34 +08:00
|
|
|
if ((t->flags & MG_TIMER_REPEAT) || !(t->flags & MG_TIMER_CALLED)) {
|
|
|
|
t->fn(t->arg);
|
|
|
|
}
|
|
|
|
t->flags |= MG_TIMER_CALLED;
|
2020-12-05 19:26:32 +08:00
|
|
|
// Try to tick timers with the given period as accurate as possible,
|
|
|
|
// even if this polling function is called with some random period.
|
2021-12-22 05:50:18 +08:00
|
|
|
t->expire = now_ms - t->expire > t->period_ms ? now_ms + t->period_ms
|
|
|
|
: t->expire + t->period_ms;
|
2020-12-05 19:26:32 +08:00
|
|
|
}
|
|
|
|
}
|