学步园

最近看了一下定时器的实现方式主要有链表，最小堆，还有就是Timing-Wheel定时器方式，自己实现了一个Timing-Wheel定时器，求拍砖

主要数据结构：

/*定时器节点*/

struct timer_node{

timer_id id; /*定时器id号*/       

int interval; /*定时器超时值*/

int solt; /*触发定时器需要的时间片*/

timer_expiry *cb; /*定时器回调函数*/

void *user_data; /*定时器传入参数*/

int len; /*定时器传入参数长度*/

struct timer_node *next;     

};

/*定时器管理*/

struct timer{

int   time_solt; /*定时器时间片大小*/

int   solt_num; /*定时器最大数量*/      

int   cur_solt; /*定时器当前时间片大小*/

int   solt_count;               /*定时器累计时间片大小*/

long  cur_time; /*定时器当前时间*/

long  pre_time; /*定时器上一个时间*/

get_time *gt; /*定时器得到当前时间*/

struct timer_node **timer_list;

} T;

定时器回调函数 ：

/*定时器超时函数*/

typedef int timer_expiry(timer_id id, void *user_data, int len);

/*获得当前时间*/

typedef long get_time(void);

timer_node结构中timer_expiry *cb为定时器超时调用的函数，返回0结束定时器，返回1继续循环定时器，这样定时器执行次数控制就非常方便了

timer结构体中get_time *gt为获得当前时间回调函数，如果返回秒即为秒定时器，返回毫秒即为毫秒定时器

具体代码如下：

time.h

#ifndef __TIMER_H__<br /> #define __TIMER_H__<br /> typedef unsigned int timer_id;<br /> /*定时器超时函数*/<br /> typedef int timer_expiry(timer_id id, void *user_data, int len);<br /> /*获得当前时间*/<br /> typedef long get_time(void);<br /> /*初始化定时器*/<br /> int init_timer(int num, int solt, get_time *gt);<br /> /*销毁所有定时器*/<br /> void destroy_timer();<br /> /*增加一个定时器*/<br /> int add_timer(timer_id id, int interval, timer_expiry *cb, void *user_data, int len);<br /> /*删除一个定时器*/<br /> int delete_timer(timer_id id);<br /> /*运行定时器*/<br /> void timer_runonce();<br /> #endif  

time.c

#include "timer.h"<br /> #include <malloc.h><br /> #include <string.h><br /> /*定时器节点*/<br /> struct timer_node{<br /> timer_id id; /*定时器id号*/<br /> int interval; /*定时器超时值*/<br /> int solt; /*触发定时器需要的时间片*/<br /> timer_expiry *cb; /*定时器回调函数*/<br /> void *user_data; /*定时器传入参数*/<br /> int len; /*定时器传入参数长度*/<br /> struct timer_node *next;<br /> };<br /> /*定时器管理*/<br /> struct timer{<br /> int time_solt; /*定时器时间片大小*/<br /> int solt_num; /*定时器最大数量*/<br /> int cur_solt; /*定时器当前时间片大小*/<br /> int solt_count; /*定时器累计时间片大小*/<br /> long cur_time; /*定时器当前时间*/<br /> long pre_time; /*定时器上一个时间*/<br /> get_time *gt; /*定时器得到当前时间*/<br /> struct timer_node **timer_list;<br /> } T;<br /> /*初始化定时器*/<br /> /*num:定时器个数*/<br /> /*solt:定时器时间片大小*/<br /> /*gt:获得当前时间函数*/<br /> /*返回值:0成功,-1失败*/<br /> int<br /> init_timer(int num, int solt, get_time *gt){</p> <p> int i = 0;<br /> memset(&T, 0, sizeof(struct timer));<br /> T.gt = gt;<br /> T.timer_list = (struct timer_node **)malloc(num*sizeof(struct timer_node *));<br /> if (0 == T.timer_list){<br /> return -1;<br /> }<br /> while (i < num){<br /> T.timer_list[i] = 0;<br /> i++;<br /> }<br /> T.time_solt = solt;<br /> T.solt_num = num;<br /> return 0;<br /> }<br /> /*销毁所有定时器*/<br /> /*返回值:无*/<br /> void<br /> destroy_timer(){<br /> int i = 0;<br /> struct timer_node *p=0, *q=0;<br /> while(i<T.solt_num){<br /> p = T.timer_list[i];<br /> if(p){<br /> while(p){<br /> q = p;<br /> p = p->next;<br /> free(q);<br /> }<br /> }<br /> i++;<br /> }<br /> free(T.timer_list);<br /> return;<br /> }<br /> /*添加一个定时器*/<br /> /*p:定时器节点指针*/<br /> /*返回值:无*/<br /> void add_timer(struct timer_node *p){<br /> int index = (T.cur_solt + p->interval/T.time_solt)%T.solt_num;<br /> p->solt = T.solt_count + p->interval/T.time_solt;<br /> if (T.timer_list[index]){<br /> p->next = T.timer_list[index];<br /> }<br /> T.timer_list[index] = p;<br /> return;<br /> }<br /> /*添加一个定时器*/<br /> /*id:定时器id用户指定,程序不保证id唯一*/<br /> /*interval:定时器超时值*/<br /> /*cb:定时器超时回调函数*/<br /> /*user_data:传入参数*/<br /> /*len:传入参数长度*/<br /> /*返回值:0成功,-1失败*/<br /> int<br /> add_timer(timer_id id, int interval, timer_expiry *cb, void *user_data, int len){<br /> struct timer_node *p = (struct timer_node *)malloc(sizeof(struct timer_node));<br /> if (0 == p){<br /> return -1;<br /> }<br /> memset(p, 0, sizeof(struct timer_node));<br /> p->cb = cb;<br /> p->id = id;<br /> p->len = len;<br /> p->interval = interval;<br /> p->user_data = user_data;<br /> add_timer(p);<br /> return 0;<br /> }<br /> /*删除一个定时器*/<br /> /*id:定时器id*/<br /> /*返回值:0成功,-1失败*/<br /> int<br /> delete_timer(timer_id id){<br /> return -1;<br /> }<br /> /*循环定时器*/<br /> /*返回值:无*/<br /> void<br /> timer_runonce(){<br /> long diff;<br /> struct timer_node *p=0, *q=0;<br /> T.cur_time = T.gt();<br /> if (0 == T.pre_time){<br /> T.pre_time = T.cur_time;<br /> }<br /> diff = (T.cur_time-T.pre_time)/T.time_solt;<br /> if (diff > 0){<br /> T.pre_time = T.cur_time;<br /> T.cur_solt += diff;<br /> T.solt_count += diff;<br /> if (T.cur_solt>=T.solt_num){<br /> T.cur_solt = 0;<br /> }<br /> }<br /> else{<br /> return;<br /> }<br /> q = T.timer_list[T.cur_solt];<br /> while (q){<br /> p = q;<br /> q = q->next;<br /> if (T.solt_count >= p->solt){<br /> T.timer_list[T.cur_solt] = q;<br /> if (p->cb(p->id, p->user_data, p->len)){<br /> p->next = 0;<br /> add_timer(p);<br /> }<br /> else{<br /> free(p);<br /> }<br /> }<br /> }<br /> return;<br /> } 

测试代码testtimer

// timer.cpp : 定义控制台应用程序的入口点。<br /> //<br /> #include "stdafx.h"<br /> #include "timer.h"<br /> #include <stdio.h><br /> #include <time.h><br /> #include <windows.h><br /> #include "stdlib.h"<br /> int g_counter = 0;<br /> long<br /> gt(){<br /> return time(0);<br /> }<br /> int<br /> cb(timer_id id, void *user_data, int len){<br /> printf("%d", id);<br /> return 1;<br /> }<br /> void test_sleep()<br /> {<br /> init_timer(10, 1, gt);<br /> add_timer(1, 5, cb, 0, 0);</p> <p> while(1){<br /> timer_runonce();<br /> Sleep(2000);<br /> }<br /> destroy_timer();<br /> }<br /> int _tmain(int argc, _TCHAR* argv[])<br /> {<br /> detect_memory_leaks(true);<br /> int num = 1024*1024 - rand()%100;<br /> init_timer(num, 1, gt);<br /> // clock_ms tm;<br /> for (int i=1; i<num; i++){<br /> int interval = rand()%30+1;<br /> add_timer(i, interval, cb, 0, 0);<br /> }<br /> // printf("add_timer: %d/n", tm.elapsed());<br /> while (1){<br /> timer_runonce();<br /> }<br /> destroy_timer();<br /> return 0;<br /> }<br />