头文件:
#ifndef __MY_TIMER_LIST_H__
#define __MY_TIMER_LIST_H__
struct timer;
struct timer*
schedule(void(*on_timer)(void* udata),
void* udata,
int after_msec,
int repeat_count);
void
cancel_timer(struct timer* t);
struct timer*
schedule_once(void(*on_timer)(void* udata),
void* udata,
int after_msec);
#endif
c文件(实现文件)
#include <stdio.h>
#include <string.h>
#include <stdlib.h>
#include "uv.h"
#include "time_list.h"
#define my_malloc malloc
#define my_free free
struct timer {
uv_timer_t uv_timer; // libuv timer handle
void(*on_timer)(void* udata);
void* udata;
int repeat_count; // -1一直循环;
};
static struct timer*
alloc_timer(void(*on_timer)(void* udata),
void* udata, int repeat_count) {
struct timer* t = my_malloc(sizeof(struct timer));
memset(t, 0, sizeof(struct timer));
t->on_timer = on_timer;
t->repeat_count = repeat_count;
t->udata = udata;
uv_timer_init(uv_default_loop(), &t->uv_timer);
return t;
}
static void
free_timer(struct timer* t) {
my_free(t);
}
static void
on_uv_timer(uv_timer_t* handle) {
struct timer* t = handle->data;
if (t->repeat_count < 0) { // 不断的触发;
t->on_timer(t->udata);
}
else {
t->repeat_count --;
t->on_timer(t->udata);
if (t->repeat_count == 0) { // 函数time结束
uv_timer_stop(&t->uv_timer); // 停止这个timer
free_timer(t);
}
}
}
struct timer*
schedule(void(*on_timer)(void* udata),
void* udata,
int after_msec,
int repeat_count) {
struct timer* t = alloc_timer(on_timer, udata, repeat_count);
// 启动一个timer;
t->uv_timer.data = t;
uv_timer_start(&t->uv_timer, on_uv_timer, after_msec, after_msec);
// end
return t;
}
void
cancel_timer(struct timer* t) {
if (t->repeat_count == 0) { // 全部触发完成,;
return;
}
uv_timer_stop(&t->uv_timer);
free_timer(t);
}
struct timer*
schedule_once(void(*on_timer)(void* udata),
void* udata,
int after_msec) {
return schedule(on_timer, udata, after_msec, 1);
}