diff options
Diffstat (limited to '')
-rw-r--r-- | src/nxt_timer.c | 317 |
1 files changed, 317 insertions, 0 deletions
diff --git a/src/nxt_timer.c b/src/nxt_timer.c new file mode 100644 index 00000000..95e8605d --- /dev/null +++ b/src/nxt_timer.c @@ -0,0 +1,317 @@ + +/* + * Copyright (C) Igor Sysoev + * Copyright (C) NGINX, Inc. + */ + +#include <nxt_main.h> + + +/* + * Timer operations are batched to improve instruction and data + * cache locality of rbtree operations. + * + * nxt_timer_add() adds a timer to the changes array to add or to + * modify the timer. The changes are processed by nxt_timer_find(). + * + * nxt_timer_disable() disables a timer. The disabled timer may + * however present in rbtree for a long time and may be eventually removed + * by nxt_timer_find() or nxt_timer_expire(). + * + * nxt_timer_delete() removes a timer at once from both the rbtree and + * the changes array and should be used only if the timer memory must be freed. + */ + +static intptr_t nxt_timer_rbtree_compare(nxt_rbtree_node_t *node1, + nxt_rbtree_node_t *node2); +static void nxt_timer_change(nxt_timers_t *timers, nxt_timer_t *timer, + nxt_msec_t time); +static void nxt_commit_timer_changes(nxt_timers_t *timers); +static void nxt_timer_drop_changes(nxt_timers_t *timers, nxt_timer_t *timer); + + +nxt_int_t +nxt_timers_init(nxt_timers_t *timers, nxt_uint_t mchanges) +{ + nxt_rbtree_init(&timers->tree, nxt_timer_rbtree_compare); + + timers->mchanges = mchanges; + + timers->changes = nxt_malloc(sizeof(nxt_timer_change_t) * mchanges); + + if (nxt_fast_path(timers->changes != NULL)) { + return NXT_OK; + } + + return NXT_ERROR; +} + + +static intptr_t +nxt_timer_rbtree_compare(nxt_rbtree_node_t *node1, nxt_rbtree_node_t *node2) +{ + nxt_timer_t *timer1, *timer2; + + timer1 = (nxt_timer_t *) node1; + timer2 = (nxt_timer_t *) node2; + + /* + * Timer values are distributed in small range, usually several minutes + * and overflow every 49 days if nxt_msec_t is stored in 32 bits. + * This signed comparison takes into account that overflow. + */ + /* timer1->time < timer2->time */ + return nxt_msec_diff(timer1->time, timer2->time); +} + + +void +nxt_timer_add(nxt_event_engine_t *engine, nxt_timer_t *timer, + nxt_msec_t timeout) +{ + int32_t diff; + uint32_t time; + + time = engine->timers.now + timeout; + + if (nxt_timer_is_in_tree(timer)) { + + diff = nxt_msec_diff(time, timer->time); + + /* + * Use the previous timer if difference between it and the + * new timer is less than required precision milliseconds: + * this decreases rbtree operations for fast connections. + */ + + if (nxt_abs(diff) < timer->precision) { + nxt_log_debug(timer->log, "timer previous: %D: %d:%M", + timer->ident, timer->state, time); + + if (timer->state == NXT_TIMER_DISABLED) { + timer->state = NXT_TIMER_ACTIVE; + } + + return; + } + + nxt_log_debug(timer->log, "timer change: %D: %d:%M", + timer->ident, timer->state, timer->time); + + } else { + /* + * The timer's time is updated here just to log a correct + * value by debug logging in nxt_timer_disable(). + * It could be updated only in nxt_commit_timer_changes() + * just before nxt_rbtree_insert(). + */ + timer->time = time; + + nxt_log_debug(timer->log, "timer add: %D: %M:%M", + timer->ident, timeout, time); + } + + nxt_timer_change(&engine->timers, timer, time); +} + + +static void +nxt_timer_change(nxt_timers_t *timers, nxt_timer_t *timer, nxt_msec_t time) +{ + nxt_timer_change_t *ch; + + if (timers->nchanges >= timers->mchanges) { + nxt_commit_timer_changes(timers); + } + + timer->state = NXT_TIMER_ACTIVE; + + ch = &timers->changes[timers->nchanges]; + ch->time = time; + ch->timer = timer; + timers->nchanges++; +} + + +#if (NXT_DEBUG) + +void +nxt_timer_disable(nxt_timer_t *timer) +{ + nxt_debug(timer->task, "timer disable: %D: %d:%M", + timer->ident, timer->state, timer->time); + + timer->state = NXT_TIMER_DISABLED; +} + +#endif + + +void +nxt_timer_delete(nxt_event_engine_t *engine, nxt_timer_t *timer) +{ + if (nxt_timer_is_in_tree(timer)) { + nxt_log_debug(timer->log, "timer delete: %D: %d:%M", + timer->ident, timer->state, timer->time); + + nxt_rbtree_delete(&engine->timers.tree, &timer->node); + nxt_timer_in_tree_clear(timer); + timer->state = NXT_TIMER_DISABLED; + } + + nxt_timer_drop_changes(&engine->timers, timer); +} + + +static void +nxt_timer_drop_changes(nxt_timers_t *timers, nxt_timer_t *timer) +{ + nxt_timer_change_t *dst, *src, *end; + + dst = timers->changes; + end = dst + timers->nchanges; + + for (src = dst; src < end; src++) { + + if (src->timer == timer) { + continue; + } + + if (dst != src) { + *dst = *src; + } + + dst++; + } + + timers->nchanges -= end - dst; +} + + +static void +nxt_commit_timer_changes(nxt_timers_t *timers) +{ + nxt_timer_t *timer; + nxt_timer_change_t *ch, *end; + + nxt_thread_log_debug("timers changes: %ui", timers->nchanges); + + ch = timers->changes; + end = ch + timers->nchanges; + + while (ch < end) { + timer = ch->timer; + + if (timer->state != NXT_TIMER_DISABLED) { + + if (nxt_timer_is_in_tree(timer)) { + nxt_log_debug(timer->log, "timer delete: %D: %d:%M", + timer->ident, timer->state, timer->time); + + nxt_rbtree_delete(&timers->tree, &timer->node); + + timer->time = ch->time; + } + + nxt_log_debug(timer->log, "timer add: %D: %M", + timer->ident, timer->time); + + nxt_rbtree_insert(&timers->tree, &timer->node); + nxt_timer_in_tree_set(timer); + } + + ch++; + } + + timers->nchanges = 0; +} + + +nxt_msec_t +nxt_timer_find(nxt_event_engine_t *engine) +{ + int32_t time; + nxt_timer_t *timer; + nxt_rbtree_node_t *node, *next; + + if (engine->timers.nchanges != 0) { + nxt_commit_timer_changes(&engine->timers); + } + + for (node = nxt_rbtree_min(&engine->timers.tree); + nxt_rbtree_is_there_successor(&engine->timers.tree, node); + node = next) + { + next = nxt_rbtree_node_successor(&engine->timers.tree, node); + + timer = (nxt_timer_t *) node; + + if (timer->state != NXT_TIMER_DISABLED) { + + if (timer->state == NXT_TIMER_BLOCKED) { + nxt_log_debug(timer->log, "timer blocked: %D: %M", + timer->ident, timer->time); + continue; + } + + time = nxt_msec_diff(timer->time, engine->timers.now); + + return (nxt_msec_t) nxt_max(time, 0); + } + + /* Delete disabled timer. */ + + nxt_log_debug(timer->log, "timer delete: %D: 0:%M", + timer->ident, timer->time); + + nxt_rbtree_delete(&engine->timers.tree, &timer->node); + nxt_timer_in_tree_clear(timer); + } + + return NXT_INFINITE_MSEC; +} + + +void +nxt_timer_expire(nxt_thread_t *thr, nxt_msec_t now) +{ + nxt_timer_t *timer; + nxt_rbtree_t *tree; + nxt_rbtree_node_t *node, *next; + + thr->engine->timers.now = now; + tree = &thr->engine->timers.tree; + + for (node = nxt_rbtree_min(tree); + nxt_rbtree_is_there_successor(tree, node); + node = next) + { + timer = (nxt_timer_t *) node; + + /* timer->time > now */ + if (nxt_msec_diff(timer->time, now) > 0) { + return; + } + + next = nxt_rbtree_node_successor(tree, node); + + if (timer->state == NXT_TIMER_BLOCKED) { + nxt_log_debug(timer->log, "timer blocked: %D: %M", + timer->ident, timer->time); + continue; + } + + nxt_log_debug(timer->log, "timer delete: %D: %d:%M", + timer->ident, timer->state, timer->time); + + nxt_rbtree_delete(tree, &timer->node); + nxt_timer_in_tree_clear(timer); + + if (timer->state != NXT_TIMER_DISABLED) { + timer->state = NXT_TIMER_DISABLED; + + nxt_work_queue_add(timer->work_queue, timer->handler, timer->task, + timer, NULL); + } + } +} |