timer_worker, timer with private stack/context

Author: lioramr

include/ribs.h

@@ -48,6 +48,7 @@
 #include "sleep.h"
 #include "malloc.h"
 #include "timer.h"
+#include "timer_worker.h"
 #include "base64.h"
 
 #endif // _RIBS__H_

include/timer_worker.h

@@ -0,0 +1,29 @@
+/*
+    This file is part of RIBS2.0 (Robust Infrastructure for Backend Systems).
+    RIBS is an infrastructure for building great SaaS applications (but not
+    limited to).
+
+    Copyright (C)  Adap.tv, Inc.
+
+    RIBS is free software: you can redistribute it and/or modify
+    it under the terms of the GNU Lesser General Public License as published by
+    the Free Software Foundation, version 2.1 of the License.
+
+    RIBS is distributed in the hope that it will be useful,
+    but WITHOUT ANY WARRANTY; without even the implied warranty of
+    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+    GNU Lesser General Public License for more details.
+
+    You should have received a copy of the GNU Lesser General Public License
+    along with RIBS.  If not, see <http://www.gnu.org/licenses/>.
+*/
+#ifndef _TIMER_WORKER__H_
+#define _TIMER_WORKER__H_
+
+#include "ribs_defs.h"
+
+void *timer_worker_init(size_t stack_size, size_t reserved_size, time_t usec_initial, void (*user_func)(void));
+int timer_worker_schedule_next(time_t usec);
+void *timer_worker_get_user_data(void);
+
+#endif // _TIMER_WORKER__H_

src/ribs2_common.mk

@@ -1,3 +1,3 @@
-SRC=context.c epoll_worker.c ctx_pool.c http_server.c hashtable.c mime_types.c http_client_pool.c timeout_handler.c ribify.c logger.c daemonize.c http_headers.c http_cookies.c file_mapper.c ds_var_field.c file_utils.c lhashtable.c search.c json.c memalloc.c mempool.c sleep.c timer.c ringbuf.c ringfile.c sendemail.c ds_loader.c heap.c vmallocator.c base64.c http_file_server.c http_vhost.c thashtable.c json_dom.c vmbuf.c
+SRC=context.c epoll_worker.c ctx_pool.c http_server.c hashtable.c mime_types.c http_client_pool.c timeout_handler.c ribify.c logger.c daemonize.c http_headers.c http_cookies.c file_mapper.c ds_var_field.c file_utils.c lhashtable.c search.c json.c memalloc.c mempool.c sleep.c timer.c timer_worker.c ringbuf.c ringfile.c sendemail.c ds_loader.c heap.c vmallocator.c base64.c http_file_server.c http_vhost.c thashtable.c json_dom.c vmbuf.c
 ASM=context_asm.S
 CFLAGS+= -I ../include

src/timer_worker.c

@@ -0,0 +1,86 @@
+/*
+    This file is part of RIBS2.0 (Robust Infrastructure for Backend Systems).
+    RIBS is an infrastructure for building great SaaS applications (but not
+    limited to).
+
+    Copyright (C)  Adap.tv, Inc.
+
+    RIBS is free software: you can redistribute it and/or modify
+    it under the terms of the GNU Lesser General Public License as published by
+    the Free Software Foundation, version 2.1 of the License.
+
+    RIBS is distributed in the hope that it will be useful,
+    but WITHOUT ANY WARRANTY; without even the implied warranty of
+    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+    GNU Lesser General Public License for more details.
+
+    You should have received a copy of the GNU Lesser General Public License
+    along with RIBS.  If not, see <http://www.gnu.org/licenses/>.
+*/
+#include "timer_worker.h"
+#include "ribs.h"
+#include <sys/timerfd.h>
+
+struct timer_worker_context {
+    int tfd;
+    void (*user_func)(void);
+    char user_data[];
+};
+
+static int _timer_worker_schedule_next(int tfd, time_t usec) {
+    struct itimerspec timerspec = {{0,0},{usec/1000000,(usec%1000000)*1000}};
+    if (0 > timerfd_settime(tfd, 0, &timerspec, NULL))
+        return LOGGER_PERROR("timerfd: %d", tfd), -1;
+    return 0;
+}
+
+static inline struct timer_worker_context *_timer_worker_get_context(void) {
+    struct timer_worker_context *twc = (struct timer_worker_context *)current_ctx->reserved;
+    return twc;
+}
+
+int timer_worker_schedule_next(time_t usec) {
+    struct timer_worker_context *twc = _timer_worker_get_context();
+    return _timer_worker_schedule_next(twc->tfd, usec);
+}
+
+static void _timer_worker_wrapper(void) {
+    struct timer_worker_context *twc = _timer_worker_get_context();
+    twc->user_func();
+}
+
+static void _timer_worker_trigger(void) {
+    struct ribs_context **worker_ctx_ref = (struct ribs_context **)current_ctx->reserved;
+    struct ribs_context *worker_ctx = *worker_ctx_ref;
+    for (;;yield()) {
+        uint64_t num_exp;
+        if (sizeof(num_exp) != read(last_epollev.data.fd, &num_exp, sizeof(num_exp))) {
+            if (EAGAIN == errno)
+                continue;
+            abort();
+        }
+        ribs_makecontext(worker_ctx, event_loop_ctx, _timer_worker_wrapper);
+        ribs_swapcurcontext(worker_ctx);
+    }
+}
+
+void *timer_worker_init(size_t stack_size, size_t reserved_size, time_t usec_initial, void (*user_func)(void)) {
+    int tfd = timerfd_create(CLOCK_MONOTONIC, TFD_NONBLOCK | TFD_CLOEXEC);
+    if (0 > tfd)
+        return LOGGER_PERROR("timerfd"), NULL;
+    if (0 > _timer_worker_schedule_next(tfd, usec_initial))
+        return close(tfd), NULL;
+    struct ribs_context *ctx = small_ctx_for_fd(tfd, sizeof(struct ribs_context *), _timer_worker_trigger);
+    struct ribs_context *worker_ctx = ribs_context_create(stack_size, sizeof(struct timer_worker_context) + reserved_size, NULL);
+    struct ribs_context **ctx_ref = (struct ribs_context **)ctx->reserved;
+    *ctx_ref = worker_ctx;
+    struct timer_worker_context *twc = (struct timer_worker_context *)worker_ctx->reserved;
+    twc->tfd = tfd;
+    twc->user_func = user_func;
+    return twc->user_data;
+}
+
+void *timer_worker_get_user_data(void) {
+    struct timer_worker_context *twc = _timer_worker_get_context();
+    return twc->user_data;
+}