62 lines
1.5 KiB
C++
62 lines
1.5 KiB
C++
|
#include <sys/timerfd.h>
|
||
|
|
#include <sys/poll.h>
|
||
|
|
#include <unistd.h>
|
||
|
|
#include "timerfd_interval.h"
|
||
|
|
|
||
|
|
timerfd_interval::timerfd_interval(ring_loop_t *ringloop, int seconds, std::function<void(void)> cb)
|
||
|
|
{
|
||
|
|
wait_state = 0;
|
||
|
|
timerfd = timerfd_create(CLOCK_MONOTONIC, TFD_NONBLOCK);
|
||
|
|
if (timerfd < 0)
|
||
|
|
{
|
||
|
|
throw std::runtime_error(std::string("timerfd_create: ") + strerror(errno));
|
||
|
|
}
|
||
|
|
struct itimerspec exp = {
|
||
|
|
.it_interval = { seconds, 0 },
|
||
|
|
.it_value = { seconds, 0 },
|
||
|
|
};
|
||
|
|
if (timerfd_settime(timerfd, 0, &exp, NULL))
|
||
|
|
{
|
||
|
|
throw std::runtime_error(std::string("timerfd_settime: ") + strerror(errno));
|
||
|
|
}
|
||
|
|
consumer.loop = [this]() { loop(); };
|
||
|
|
ringloop->register_consumer(consumer);
|
||
|
|
this->ringloop = ringloop;
|
||
|
|
this->callback = cb;
|
||
|
|
}
|
||
|
|
|
||
|
|
timerfd_interval::~timerfd_interval()
|
||
|
|
{
|
||
|
|
ringloop->unregister_consumer(consumer);
|
||
|
|
close(timerfd);
|
||
|
|
}
|
||
|
|
|
||
|
|
void timerfd_interval::loop()
|
||
|
|
{
|
||
|
|
if (wait_state == 1)
|
||
|
|
{
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
struct io_uring_sqe *sqe = ringloop->get_sqe();
|
||
|
|
if (!sqe)
|
||
|
|
{
|
||
|
|
wait_state = 0;
|
||
|
|
return;
|
||
|
|
}
|
||
|
|
struct ring_data_t *data = ((ring_data_t*)sqe->user_data);
|
||
|
|
my_uring_prep_poll_add(sqe, timerfd, POLLIN);
|
||
|
|
data->callback = [&](ring_data_t *data)
|
||
|
|
{
|
||
|
|
if (data->res < 0)
|
||
|
|
{
|
||
|
|
throw std::runtime_error(std::string("waiting for timer failed: ") + strerror(-data->res));
|
||
|
|
}
|
||
|
|
uint64_t n;
|
||
|
|
read(timerfd, &n, 8);
|
||
|
|
wait_state = 0;
|
||
|
|
callback();
|
||
|
|
};
|
||
|
|
wait_state = 1;
|
||
|
|
ringloop->submit();
|
||
|
|
}
|