간단한 타이머

이 기사에서는 루프 배열을 기반으로 간단한 타임 휠 타이머를 구현합니다.

타이머는 싱글 샷 또는 주기적으로 트리거 될 수 있습니다. 타이머는 매초마다 그리드로 내려가 이벤트가 트리거되도록 어레이 노드를 스캔합니다. 그리드에 여러 노드가있을 수 있으며 수준이 0 인 노드가 트리거됩니다. 이 예제는 비교적 간단하며 경보 시스템 호출을 통해 구현되며 경보는 SIGALRM 신호를 보내 틱 기능의 실행을 트리거합니다.

#include <signal.h>
#include <stdio.h>
#include <unistd.h>

#define TIME_WHEEL_SIZE 8          // 轮子大小
typedef void (*func_t)(int data);  // 回调函数类型

// 时间轮节点
struct timer_node {
    int rotation;             //轮数，有多少轮
    func_t func;              // 回调函数
    int data;                 // 回调函数的参数
    struct timer_node *next;  // 下一个节点
};

// 时间轮
struct timer_wheel {
    struct timer_node *slot[TIME_WHEEL_SIZE];  //节点数组
    int current;                               // 当前节点
};

// 初始化时间轮
struct timer_wheel timer = {
   
   {0}, 0};

// 时间轮最小单元，每一秒检查一下时间轮
void tick(int signo) {
    struct timer_node **cur = &timer.slot[timer.current];  //当前slot
    while (*cur) {
        struct timer_node *curr = *cur;
        if (curr->rotation > 0) {
            curr->rotation--;
            cur = &curr->next;
        } else {
            curr->func(curr->data);
            *cur = curr->next;
            free(curr);
        }
    }
    timer.current = (timer.current + 1) % TIME_WHEEL_SIZE;
    alarm(1);  // 如果要多次触发，通常的做法是在这行加入alarm
}

// 添加节点
void add_timer(int len, func_t func) {
    int pos = (len + timer.current) % TIME_WHEEL_SIZE;  // 新增节点的位置
    struct timer_node *node = malloc(sizeof(struct timer_node));

    node->next = timer.slot[pos];
    timer.slot[pos] = node;
    node->rotation = len / TIME_WHEEL_SIZE;  // 新增节点处于第几轮
    node->data = 0;
    node->func = func;
}
int g_sec = 0;

void do_time1(int data) {
    printf("timer %s, %d\n", __FUNCTION__, g_sec++);
    add_timer(1, do_time1);  // 循环添加
}

// 只打印一次
void do_time2(int data) { printf("timer %s, %d\n", __FUNCTION__, g_sec++); }

int main(int argc, char const *argv[]) {
    signal(SIGALRM, tick);
    alarm(1);
    add_timer(1, do_time1);
    add_timer(3, do_time2);
    add_timer(15, do_time2);
    // alarm(1);// signal只生效一次，之后恢复默认，多次调用alarm不会回调
    while (1) pause();
    return 0;
}

참고:

[1] Linux C / C ++ 타이머의 구현 원리 및 사용 방법

추천