由alarm和setitimer函数设置的实时闹钟一旦超时,将触发SIGALRM信号。
因此,我们可以利用该信号的信号处理函数来处理定时任务,但是,如果要处理多个定时任务,就需要不断触发此信号,并在其信号处理函数中执行到期的任务。
一般而言, SIGALRM信号按照固定的频率生成,即由alarm和setitimer函数设置的定时周期T保持不变,如果某个定时任务的超时时间不是T的整数倍,那么它实际被执行的时间和预期的时间将略有偏差,因此定时周期T反映了定时的精度。
此次,我们实现一个简单的基于升序链表的定时器,来处理非活动连接。
1. 代码
//lst_timer.h
#ifndef LST_TIMER
#define LST_TIMER
#include <time.h>
#include <sys/types.h>
#include <sys/socket.h>
#include <netinet/in.h>
#include <arpa/inet.h>
#include <stdio.h>
#define BUFFER_SIZE 64
class util_timer;
//用户数据结构
struct client_data
{
sockaddr_in address; //客户端地址
int sockfd; //客户端socket
char buf[BUFFER_SIZE]; //接收到的客户端数据
util_timer *timer; //定时器
};
//定时器类
class util_timer
{
public:
util_timer():prev(NULL),next(NULL) {}
public:
time_t expire; //任务超时时间,这里使用绝对时间
void (*cb_func)(client_data*); //任务回调函数
client_data *user_data; //客户端数据
util_timer *prev; //指向上一个定时器
util_timer *next; //指向下一个定时器
};
//定时器链表
class sort_timer_lst
{
public:
sort_timer_lst():head(NULL), tail(NULL) {}
~sort_timer_lst();
//将目标定时器添加到链表中
void add_timer(util_timer *timer);
//当某个定时任务发生变化时,调整其在定时器链表中位置
//这个函数只考虑被调整的定时器的超时延长情况,即该定时器需要往链表的尾部移动
void adjust_timer(util_timer * timer);
//将目标定时器从链表中删除
void del_timer(util_timer *timer);
//SIGALRM信号每次被触发时,就在其信号处理函数中执行一次(如果使用统一事件源,则是主函数),
//以处理链表上到期的任务
void tick();
private:
//重载的辅助函数,被add_timer和adjust_timer调用,
//将目标定时器添加到lst_head之后的部分链表中
void add_timer(util_timer *timer, util_timer *lst_head);
private:
util_timer *head; //链表头
util_timer *tail; //链表尾
};
#endif
//lst_timer.cpp
#include "lst_timer.h"
sort_timer_lst::~sort_timer_lst()
{
util_timer *temp = head;
//删除所有定时器
while (temp) {
head = temp->next;
delete temp;
temp = head;
}
}
void sort_timer_lst::add_timer(util_timer *timer)
{
if (!timer)
return;
if (!head) {
head = tail = timer;
return;
}
//如果目标定时器的超时时间比头节点的超时时间小,
//则把该定时器插入头部,作为头节点。
//否则插入合适的位置,保证升序特性
if (timer->expire < head->expire) {
timer->next = head;
head->prev = timer;
head = timer;