vn_event_timer.c

/*
 * Copyright (C) Chenyang Li
 * Copyright (C) Vino
 */
#include <stdlib.h>
#include <sys/time.h>
#include "vn_event_timer.h"
#include "vn_priority_queue.h"
#include "vn_palloc.h"
#include "vn_logger.h"
#include "vn_error.h"

extern vn_priority_queue_t pq;

int vn_event_timer_init(void) {
    vn_pq_init(&pq);
    vn_event_time_update();
    return 0;
}

void vn_event_time_update(void) {
    struct timeval tv;
    vn_sec_t sec;
    vn_msec_t msec;

    if (gettimeofday(&tv, NULL) < 0) {
        err_sys("[vn_time_update] gettimeofday error");
    }
    sec = tv.tv_sec;
    msec = tv.tv_usec / 1000;

    vn_current_msec = sec * 1000 + msec;
}

vn_msec_t vn_event_find_timer(void) {
    long timer;
    vn_priority_queue_node_t *node;

    if (vn_pq_isempty(&pq)) {
        return 0;
    }

    node = vn_pq_min(&pq);
    timer = node->key - vn_current_msec;
    return timer > 0 ? timer : 0;
}

void vn_event_expire_timers(void) {
    vn_priority_queue_node_t *node;
    vn_http_connection_t *conn;

    while (!vn_pq_isempty(&pq)) {
        node = vn_pq_min(&pq);

        if (node->deleted == VN_PQ_DELETED) {
            vn_pq_delete_min(&pq);
            continue;
        }

        /* 
         * If node.key < vn_current_msec, timeout.
         * Be careful, the gap should be cast to Long type.
         */
        if ((long) (node->key - vn_current_msec) > 0) {
            return;
        }

        conn = (vn_http_connection_t *) node->data;
    
        if (conn->handler) {
#ifdef DEBUG
            DEBUG_PRINT("The timer has invoked handler to close the connection, [fd = %d]", conn->fd);
#endif
            (*conn->handler)(conn);
        }
        vn_pq_delete_min(&pq);
    }
}

void vn_event_add_timer(vn_http_connection_t *conn, vn_msec_t timer) {
    vn_priority_queue_node_t *node;
    vn_msec_t key;

    if ((node = (vn_priority_queue_node_t *) vn_palloc(conn->pool, sizeof(vn_priority_queue_node_t))) == NULL) {
        err_sys("[vn_event_add_timer] vn_palloc error");
    }

    vn_event_time_update();
    key = vn_current_msec + timer;
    node->key = key;
    node->data = (void *) conn;
    node->deleted = VN_PQ_NOT_DELETED;
    vn_pq_insert(&pq, node);

    conn->pq_node = node;
}