1 files changed, 139 insertions, 0 deletions
diff --git a/src/nxt_timer.h b/src/nxt_timer.h
new file mode 100644
index 00000000..de11e9e3
--- /dev/null
+++ b/src/nxt_timer.h
@@ -0,0 +1,139 @@
+
+/*
+ * Copyright (C) Igor Sysoev
+ * Copyright (C) NGINX, Inc.
+ */
+
+#ifndef _NXT_TIMER_H_INCLUDED_
+#define _NXT_TIMER_H_INCLUDED_
+
+
+/* Valid values are between 1ms to 255ms. */
+#define NXT_TIMER_DEFAULT_PRECISION  100
+//#define NXT_TIMER_DEFAULT_PRECISION  1
+
+
+#if (NXT_DEBUG)
+#define NXT_TIMER             { NXT_RBTREE_NODE_INIT, 0, 0, 0,                \
+                                NULL, NULL, NULL, NULL, -1 }
+
+#else
+#define NXT_TIMER             { NXT_RBTREE_NODE_INIT, 0, 0, 0,                \
+                                NULL, NULL, NULL, NULL }
+#endif
+
+
+typedef struct {
+    /* The rbtree node must be the first field. */
+    NXT_RBTREE_NODE           (node);
+
+    uint8_t                   state;
+    uint8_t                   precision;
+    nxt_msec_t                time;
+
+    nxt_work_queue_t          *work_queue;
+    nxt_work_handler_t        handler;
+
+    nxt_task_t                *task;
+    nxt_log_t                 *log;
+#if (NXT_DEBUG)
+    int32_t                   ident;
+#endif
+} nxt_timer_t;
+
+
+typedef struct {
+    nxt_msec_t                time;
+    nxt_timer_t               *timer;
+} nxt_timer_change_t;
+
+
+typedef struct {
+    nxt_rbtree_t              tree;
+
+    /* An overflown milliseconds counter. */
+    nxt_msec_t                now;
+
+    nxt_uint_t                mchanges;
+    nxt_uint_t                nchanges;
+
+   nxt_timer_change_t         *changes;
+} nxt_timers_t;
+
+
+#define nxt_timer_data(obj, type, timer)                                      \
+    nxt_container_of(obj, type, timer)
+
+
+/*
+ * When timer resides in rbtree all links of its node are not NULL.
+ * A parent link is the nearst to other timer flags.
+ */
+
+#define nxt_timer_is_in_tree(timer)                                           \
+    ((timer)->node.parent != NULL)
+
+#define nxt_timer_in_tree_set(timer)
+    /* Noop, because rbtree insertion sets a node's parent link. */
+
+#define nxt_timer_in_tree_clear(timer)                                        \
+    (timer)->node.parent = NULL
+
+
+#define NXT_TIMER_DISABLED  0
+#define NXT_TIMER_BLOCKED   1
+#define NXT_TIMER_ACTIVE    2
+
+
+#if (NXT_DEBUG)
+
+#define nxt_timer_ident(timer, val)                                           \
+    (timer)->ident = (val)
+
+#else
+
+#define nxt_timer_ident(timer, val)
+
+#endif
+
+
+nxt_inline nxt_timer_t *
+nxt_timer_create(int32_t ident)
+{
+    nxt_timer_t  *timer;
+
+    timer = nxt_zalloc(sizeof(nxt_timer_t));
+    if (timer == NULL) {
+        return NULL;
+    }
+
+    timer->precision = NXT_TIMER_DEFAULT_PRECISION;
+#if (NXT_DEBUG)
+    timer->ident = ident;
+#endif
+
+    return timer;
+}
+
+
+nxt_int_t nxt_timers_init(nxt_timers_t *timers, nxt_uint_t mchanges);
+NXT_EXPORT void nxt_timer_add(nxt_event_engine_t *engine, nxt_timer_t *timer,
+    nxt_msec_t timeout);
+NXT_EXPORT void nxt_timer_delete(nxt_event_engine_t *engine,
+    nxt_timer_t *timer);
+nxt_msec_t nxt_timer_find(nxt_event_engine_t *engine);
+void nxt_timer_expire(nxt_thread_t *thr, nxt_msec_t now);
+
+#if (NXT_DEBUG)
+
+NXT_EXPORT void nxt_timer_disable(nxt_timer_t *timer);
+
+#else
+
+#define nxt_timer_disable(timer)                                              \
+    (timer)->state = NXT_TIMER_DISABLED
+
+#endif
+
+
+#endif /* _NXT_TIMER_H_INCLUDED_ */