linux下多定時器的實現-(multimer implementation under linux)

概論

定時器屬於基本的基礎組件,不管是用戶空間的程序開發,還是內核空間的程序開發,很多時候都需要有定時器作爲基礎組件的支持,但使用場景的不同,對定時器的實現考慮也不盡相同,本文討論了在 Linux 環境下,應用層和內核層的定時器的各種實現方法,並分析了各種實現方法的利弊以及適宜的使用環境。

首先,給出一個基本模型,定時器的實現,需要具備以下幾個行爲,這也是在後面評判各種定時器實現的一個基本模型 [1]:

StartTimer(Interval, TimerId, ExpiryAction)

註冊一個時間間隔爲 Interval 後執行 ExpiryAction 的定時器實例,其中,返回 TimerId 以區分在定時器系統中的其他定時器實例。

StopTimer(TimerId)

根據 TimerId 找到註冊的定時器實例並執行 Stop 。

PerTickBookkeeping()

在一個 Tick 內,定時器系統需要執行的動作,它最主要的行爲,就是檢查定時器系統中,是否有定時器實例已經到期。注意,這裏的 Tick 實際上已經隱含了一個時間粒度 (granularity) 的概念。

ExpiryProcessing()

在定時器實例到期之後,執行預先註冊好的 ExpiryAction 行爲。

上面說了基本的定時器模型,但是針對實際的使用情況,又有以下 2 種基本行爲的定時器:

Single-Shot Timer

這種定時器,從註冊到終止,僅僅只執行一次。

Repeating Timer

這種定時器,在每次終止之後,會自動重新開始。本質上,可以認爲 Repeating Timer 是在 Single-Shot Timer 終止之後,再次註冊到定時器系統裏的 Single-Shot Timer,因此,在支持 Single-Shot Timer 的基礎上支持 Repeating Timer 並不算特別的複雜。

————————————————————————————————————————————————————————————————————————————

基於鏈表和信號實現定時器(2.4版內核情況下)

在 2.4 的內核中,並沒有提供 POSIX timer [ 2 ]的支持,要在進程環境中支持多個定時器,只能自己來實現,好在 Linux 提供了 setitimer(2) 的接口。它是一個具有間隔功能的定時器 (interval timer),但如果想在進程環境中支持多個計時器,不得不自己來管理所有的計時器。 setitimer(2) 的定義如下:

清單 1. setitimer的原型

#include <sys/time.h> 

 int setitimer(int which, const struct itimerval *new_value,struct itimerval *old_value);

setitimer 能夠在 Timer 到期之後,自動再次啓動自己,因此,用它來解決 Single-Shot Timer 和 Repeating Timer 的問題顯得很簡單。該函數可以工作於 3 種模式:

ITIMER_REAL 以實時時間 (real time) 遞減,在到期之後發送 SIGALRM 信號

ITIMER_VIRTUAL 僅進程在用戶空間執行時遞減,在到期之後發送 SIGVTALRM 信號

ITIMER_PROF 進程在用戶空間執行以及內核爲該進程服務時 ( 典型如完成一個系統調用 ) 都會遞減,與 ITIMER_VIRTUAL 共用時可度量該應用在內核空間和用戶空間的時間消耗情況,在到期之後發送 SIGPROF 信號

定時器的值由下面的結構定義:

清單 2. setitimer 定時器的值定義

				
struct itimerval { 
 struct timeval it_interval; /* next value */ 
 struct timeval it_value;     /* current value */ 
 }; 

 struct timeval { 
        long tv_sec;                /* seconds */ 
        long tv_usec;               /* microseconds */ 
 };

setitimer() 以 new_value 設置特定的定時器,如果 old_value 非空,則它返回 which 類型時間間隔定時器的前一個值。定時器從 it_value 遞減到零,然後產生一個信號,並重新設置爲 it_interval,如果此時 it_interval 爲零,則該定時器停止。任何時候,只要 it_value 設置爲零,該定時器就會停止。

由於 setitimer() 不支持在同一進程中同時使用多次以支持多個定時器,因此,如果需要同時支持多個定時實例的話,需要由實現者來管理所有的實例。用 setitimer() 和鏈表,可以構造一個在進程環境下支持多個定時器實例的 Timer,在一般的實現中的 PerTickBookkeeping 時,會遞增每個定時器的 elapse 值,直到該值遞增到最初設定的 interval 則表示定時器到期。

基於鏈表實現的定時器可以定義爲:

清單 3. 基於鏈表的定時器定義

typedef int timer_id; 

 /** 
 * The type of callback function to be called by timer scheduler when a timer 
 * has expired. 
 * 
 * @param id                The timer id. 
 * @param user_data        The user data. 
 * $param len               The length of user data. 
 */ 
 typedef int timer_expiry(timer_id id, void *user_data, int len); 

 /** 
 * The type of the timer 
 */ 
 struct timer { 
        LIST_ENTRY(timer) entries;/**< list entry               */ 

        timer_id id;                /**< timer id                  */ 

        int interval;               /**< timer interval(second) */ 
        int elapse;                 /**< 0 -> interval             */ 

        timer_expiry *cb;          /**< call if expiry            */ 
        void *user_data;           /**< callback arg               */ 
        int len; 	                    /**< user_data length          */ 
 };

定時器的時間間隔以 interval 表示,而 elapse 則在 PerTickBookkeeping() 時遞增,直到 interval 表示定時器中止,此時調用回調函數 cb 來執行相關的行爲,而 user_data 和 len 爲用戶可以傳遞給回調函數的參數。

所有的定時器實例以鏈表來管理:

清單 4. 定時器鏈表

/** 
 * The timer list 
 */ 
 struct timer_list { 
        LIST_HEAD(listheader, timer) header;  /**< list header         */ 
        int num; 	                                   /**< timer entry number */ 
        int max_num;                               /**< max entry number    */ 

        void (*old_sigfunc)(int);                /**< save previous signal handler */ 
        void (*new_sigfunc)(int);                /**< our signal handler              */ 

        struct itimerval ovalue;                 /**< old timer value */ 
        struct itimerval value;                  /**< our internal timer value */ 
 };
這裏關於鏈表的實現使用了 BSD 風格關於鏈表的一組宏,避免了再造輪子;該結構中,old_sigfunc 在 init_timer 初始定時器鏈表時候用來保存系統對 SIGALRM 的處理函數,在定時器系統 destory 時用來恢復到之前的處理函數; ovalue 的用途與此類似。

清單 5. 定時器鏈表的創建和Destroy

/** 
 * Create a timer list. 
 * 
 * @param count  The maximum number of timer entries to be supported initially. 
 * 
 * @return        0 means ok, the other means fail. 
 */ 
 int init_timer(int count) 
 { 
        int ret = 0; 

        if(count <=0 || count > MAX_TIMER_NUM) { 
               printf("the timer max number MUST less than %d.\n", MAX_TIMER_NUM); 
               return -1; 
        } 

        memset(&timer_list, 0, sizeof(struct timer_list)); 
        LIST_INIT(&timer_list.header); 
        timer_list.max_num = count; 

        /* Register our internal signal handler and store old signal handler */ 
        if ((timer_list.old_sigfunc = signal(SIGALRM, sig_func)) == SIG_ERR) { 
                return -1; 
        } 
        timer_list.new_sigfunc = sig_func; 

     /*Setting our interval timer for driver our mutil-timer and store old timer value*/ 
        timer_list.value.it_value.tv_sec = TIMER_START; 
        timer_list.value.it_value.tv_usec = 0; 
        timer_list.value.it_interval.tv_sec = TIMER_TICK; 
        timer_list.value.it_interval.tv_usec = 0; 
        ret = setitimer(ITIMER_REAL, &timer_list.value, &timer_list.ovalue); 

        return ret; 
 } 


 /** 
 * Destroy the timer list. 
 * 
 * @return          0 means ok, the other means fail. 
 */ 
 int destroy_timer(void) 
 { 
        struct timer *node = NULL; 

        if ((signal(SIGALRM, timer_list.old_sigfunc)) == SIG_ERR) { 
                return -1; 
        } 

        if((setitimer(ITIMER_REAL, &timer_list.ovalue, &timer_list.value)) < 0) { 
                return -1; 
        } 

        while (!LIST_EMPTY(&timer_list.header)) {     /* Delete. */ 
		 node = LIST_FIRST(&timer_list.header); 
                LIST_REMOVE(node, entries); 
                /* Free node */ 
		 printf("Remove id %d\n", node->id); 
                free(node->user_data); 
                free(node); 
        } 

        memset(&timer_list, 0, sizeof(struct timer_list)); 

        return 0; 
 }

添加定時器的動作非常的簡單,本質只是一個鏈表的插入而已:

清單 6. 向定時器鏈表中添加定時器

/** 
 * Add a timer to timer list. 
 * 
 * @param interval  The timer interval(second). 
 * @param cb  	    When cb!= NULL and timer expiry, call it. 
 * @param user_data Callback's param. 
 * @param len  	    The length of the user_data. 
 * 
 * @return          The timer ID, if == INVALID_TIMER_ID, add timer fail. 
 */ 
 timer_id  add_timer(int interval, timer_expiry *cb, void *user_data, int len) 
 { 
        struct timer *node = NULL; 

        if (cb == NULL || interval <= 0) { 
                return INVALID_TIMER_ID; 
        } 

	 if(timer_list.num < timer_list.max_num) { 
		 timer_list.num++; 
	 } else { 
		 return INVALID_TIMER_ID; 
	 } 

	 if((node = malloc(sizeof(struct timer))) == NULL) { 
		 return INVALID_TIMER_ID; 
	 } 
	 if(user_data != NULL || len != 0) { 
		 node->user_data = malloc(len); 
		 memcpy(node->user_data, user_data, len); 
		 node->len = len; 
	 } 

	 node->cb = cb; 
	 node->interval = interval; 
	 node->elapse = 0; 
	 node->id = timer_list.num; 

	 LIST_INSERT_HEAD(&timer_list.header, node, entries); 

    return node->id; 
 }
註冊的信號處理函數則用來驅動定時器系統:


清單 7. 信號處理函數驅動定時器

				
/* Tick Bookkeeping */ 
 static void sig_func(int signo) 
 { 
        struct timer *node = timer_list.header.lh_first; 
        for ( ; node != NULL; node = node->entries.le_next) { 
                node->elapse++; 
                if(node->elapse >= node->interval) { 
                        node->elapse = 0; 
                        node->cb(node->id, node->user_data, node->len); 
                } 
        } 
 }

它主要是在每次收到 SIGALRM 信號時,執行定時器鏈表中的每個定時器 elapse 的自增操作,並與 interval 相比較,如果相等,代表註冊的定時器已經超時,這時則調用註冊的回調函數。

上面的實現,有很多可以優化的地方:考慮另外一種思路,在定時器系統內部將維護的相對 interval 轉換成絕對時間,這樣,在每 PerTickBookkeeping 時,只需將當前時間與定時器的絕對時間相比較,就可以知道是否該定時器是否到期。這種方法,把遞增操作變爲了比較操作。並且上面的實現方式,效率也不高,在執行 StartTimer,StopTimer,PerTickBookkeeping 時,算法複雜度分別爲 O(1),O(n),O(n),可以對上面的實現做一個簡單的改進,在 StartTimer 時,即在添加 Timer 實例時,對鏈表進行排序,這樣的改進,可以使得在執行 StartTimer,StopTimer,PerTickBookkeeping 時,算法複雜度分別爲 O(n),O(1),O(1) 。改進後的定時器系統如下圖 1:

圖 1. 基於排序鏈表的定時器

基於排序鏈表的定時器

————————————————————————————————————————————————————————————————————————————

基於2.6版本內核定時器的實現(Posix實時定時器)

Linux 自 2.6 開始,已經開始支持 POSIX timer [ 2 ]所定義的定時器,它主要由下面的接口構成 :

清單 8. POSIX timer接口

				
#include <signal.h> 
 #include <time.h> 

 int timer_create(clockid_t clockid, struct sigevent *evp, 
 timer_t *timerid); 
 int timer_settime(timer_t timerid, int flags, 
 const struct itimerspec *new_value, 
 struct itimerspec * old_value); 
 int timer_gettime(timer_t timerid, struct itimerspec *curr_value); 
 int timer_getoverrun(timer_t timerid); 
 int timer_delete(timer_t timerid);

這套接口是爲了讓操作系統對實時有更好的支持,在鏈接時需要指定 -lrt 。

timer_create(2): 創建了一個定時器。

timer_settime(2): 啓動或者停止一個定時器。

timer_gettime(2): 返回到下一次到期的剩餘時間值和定時器定義的時間間隔。出現該接口的原因是,如果用戶定義了一個 1ms 的定時器,可能當時系統負荷很重,導致該定時器實際山 10ms 後才超時,這種情況下,overrun=9ms 。

timer_getoverrun(2): 返回上次定時器到期時超限值。

timer_delete(2): 停止並刪除一個定時器。

上面最重要的接口是 timer_create(2),其中,clockid 表明了要使用的時鐘類型,在 POSIX 中要求必須實現 CLOCK_REALTIME 類型的時鐘。 evp 參數指明瞭在定時到期後,調用者被通知的方式。該結構體定義如下 :

清單 9. POSIX timer接口中的信號和事件定義

union sigval { 
 int sival_int; 
 void *sival_ptr; 
 }; 

 struct sigevent { 
 int sigev_notify; /* Notification method */ 
 int sigev_signo; /* Timer expiration signal */ 
 union sigval sigev_value; /* Value accompanying signal or 
 passed to thread function */ 
 void (*sigev_notify_function) (union sigval); 
 /* Function used for thread 
 notifications (SIGEV_THREAD) */ 
 void *sigev_notify_attributes; 
 /* Attributes for notification thread 
 (SIGEV_THREAD) */ 
 pid_t sigev_notify_thread_id; 
 /* ID of thread to signal (SIGEV_THREAD_ID) */ 
 };

其中,sigev_notify 指明瞭通知的方式 :

SIGEV_NONE

當定時器到期時,不發送異步通知,但該定時器的運行進度可以使用 timer_gettime(2) 監測。

SIGEV_SIGNAL

當定時器到期時,發送 sigev_signo 指定的信號。

SIGEV_THREAD

當定時器到期時,以 sigev_notify_function 開始一個新的線程。該函數使用 sigev_value 作爲其參數,當 sigev_notify_attributes 非空,則制定該線程的屬性。注意,由於 Linux 上線程的特殊性,這個功能實際上是由 glibc 和內核一起實現的。

SIGEV_THREAD_ID (Linux-specific)

僅推薦在實現線程庫時候使用。

如果 evp 爲空的話,則該函數的行爲等效於:sigev_notify = SIGEV_SIGNAL,sigev_signo = SIGVTALRM,sigev_value.sival_int = timer ID 。

由於 POSIX timer [ 2 ]接口支持在一個進程中同時擁有多個定時器實例,所以在上面的基於 setitimer() 和鏈表的 PerTickBookkeeping 動作就交由 Linux 內核來維護,這大大減輕了實現定時器的負擔。由於 POSIX timer [ 2 ]接口在定時器到期時,有更多的控制能力,因此,可以使用實時信號避免信號的丟失問題,並將 sigev_value.sival_int 值指定爲 timer ID,這樣,就可以將多個定時器一起管理了。需要注意的是,POSIX timer [ 2 ]接口只在進程環境下才有意義 (fork(2) 和 exec(2) 也需要特殊對待 ),並不適合多線程環境。與此相類似的,Linux 提供了基於文件描述符的相關定時器接口:


清單 10. linux提供的基於文件描述符的定時器接口

#include <sys/timerfd.h> 

 int timerfd_create(int clockid, int flags); 
 int timerfd_settime(int fd, int flags, 
			 const struct itimerspec *new_value, 
			 struct itimerspec *old_value); 
 int timerfd_gettime(int fd, struct itimerspec *curr_value);
這樣,由於基於文件描述符,使得該接口可以支持 select(2),poll(2) 等異步接口,使得定時器的實現和使用更加的方便,更重要的是,支持 fork(2),exec(2) 這樣多進程的語義,因此,可以用在多線程環境之中,它們的使用比 POSIX timer [ 2 ]更加的靈活,其根本原因在於定時器的管理統一到了 unix/linux 基本哲學之一 ---- “一切皆文件”之下。

————————————————————————————————————————————————————————————————————————————

最小堆實現的定時器

最小堆指的是滿足除了根節點以外的每個節點都不小於其父節點的堆。這樣,堆中的最小值就存放在根節點中,並且在以某個結點爲根的子樹中,各節點的值都不小於該子樹根節點的值。一個最小堆的例子如下圖 2:

圖 2. 最小堆

最小堆

一個最小堆,一般支持以下幾種操作:

Insert(TimerHeap, Timer): 在堆中插入一個值,並保持最小堆性質,具體對應於定時器的實現,則是把定時器插入到定時器堆中。根據最小堆的插入算法分析,可以知道該操作的時間複雜度爲 O(lgn) 。

Minimum(TimerHeap): 獲取最小堆的中最小值;在定時器系統中,則是返回定時器堆中最先可能終止的定時器。由於是最小堆,只需返回堆的 root 即可。此時的算法複雜度爲 O(1) 。

ExtractMin(TimerHeap): 在定時器到期後,執行相關的動作,它的算法複雜度爲 O(1) 。

最小堆本質上是一種最小優先級隊列 (min-priority queue) 。定時可以作爲最小優先級隊列的一個應用,該優先級隊列把定時器的時間間隔值轉化爲一個絕對時間來處理,ExtractMin 操則是在所有等待的定時器中,找出最先超時的定時器。在任何時候,一個新的定時器實例都可通過 Insert 操作加入到定時器隊列中去。

在 pjsip 項目的基礎庫 pjlib 中,有基於最小堆實現的定時器,它主要提供了以下的幾個接口:

清單 11. pjlib提供的基於最小堆的定時器接口

				
/** 
 * Create a timer heap. 
 */ 
 PJ_DECL(pj_status_t) pj_timer_heap_create( pj_pool_t *pool, 
					   pj_size_t count, 
                                           pj_timer_heap_t **ht); 

 /** 
 * Destroy the timer heap. 
 */ 
 PJ_DECL(void) pj_timer_heap_destroy( pj_timer_heap_t *ht ); 

 /** 
 * Initialize a timer entry. Application should call this function at least 
 * once before scheduling the entry to the timer heap, to properly initialize 
 * the timer entry. 
 */ 
 PJ_DECL(pj_timer_entry*) pj_timer_entry_init( pj_timer_entry *entry, 
                                              int id, 
                                              void *user_data, 
                                              pj_timer_heap_callback *cb ); 

 /** 
 * Schedule a timer entry which will expire AFTER the specified delay. 
 */ 
 PJ_DECL(pj_status_t) pj_timer_heap_schedule( pj_timer_heap_t *ht, 
					     pj_timer_entry *entry, 
					     const pj_time_val *delay); 

 /** 
 * Cancel a previously registered timer. 
 */ 
 PJ_DECL(int) pj_timer_heap_cancel( pj_timer_heap_t *ht, 
				   pj_timer_entry *entry); 

 /** 
 * Poll the timer heap, check for expired timers and call the callback for 
 * each of the expired timers. 
 */ 
 PJ_DECL(unsigned) pj_timer_heap_poll( pj_timer_heap_t *ht, 
                                      pj_time_val *next_delay);
pjlib 中的定時器在內部使用數組的方式實現堆,這樣對於內存空間的使用將更加的緊湊;它的實現還可在定時器的數量超過預先設定的最大數量時會自己增加最大定時器數量。文件 pjlib/src/pjlib-test/timer.c 是它的一個單元測試。與基於鏈表方式的實現相比較,明顯它的時間複雜度要低一些,這樣可以支持更多的定時器實例。

————————————————————————————————————————————————————————————————————————————

基於時間輪(timing-wheel)方式實現的定時器

時間輪 (Timing-Wheel) 算法類似於一以恆定速度旋轉的左輪手槍,槍的撞針則撞擊槍膛,如果槍膛中有子彈,則會被擊發;與之相對應的是:對於 PerTickBookkeeping,其最本質的工作在於以 Tick 爲單位增加時鐘,如果發現有任何定時器到期,則調用相應的 ExpiryProcessing 。設定一個循環爲 N 個 Tick 單元,當前時間是在 S 個循環之後指向元素 i (i>=0 and i<= N - 1),則當前時間 (Current Time)Tc 可以表示爲:Tc = S*N + i ;如果此時插入一個時間間隔 (Time Interval) 爲 Ti 的定時器,設定它將會放入元素 n(Next) 中,則 n = (Tc + Ti)mod N = (S*N + i + Ti) mod N = (i + Ti) mod N 。如果我們的 N 足夠的大,顯然 StartTimer,StopTimer,PerTickBookkeeping 時,算法複雜度分別爲 O(1),O(1),O(1) 。在 [5] 中,給出了一個簡單定時器輪實現的定時。下圖 3 是一個簡單的時間輪定時器:

圖 3. 簡單的時間輪

簡單時間輪

如果需要支持的定時器範圍非常的大,上面的實現方式則不能滿足這樣的需求。因爲這樣將消耗非常可觀的內存,假設需要表示的定時器範圍爲:0 – 2^3-1ticks,則簡單時間輪需要 2^32 個元素空間,這對於內存空間的使用將非常的龐大。也許可以降低定時器的精度,使得每個 Tick 表示的時間更長一些,但這樣的代價是定時器的精度將大打折扣。現在的問題是,度量定時器的粒度,只能使用唯一粒度嗎?想想日常生活中常遇到的水錶,如下圖 4:

圖 4. 水錶

水錶

在上面的水錶中,爲了表示度量範圍,分成了不同的單位,比如 1000,100,10 等等,相似的,表示一個 32bits 的範圍,也不需要 2^32 個元素的數組。實際上,Linux 的內核把定時器分爲 5 組,每組的粒度分別表示爲:1 jiffies,256 jiffies,256*64 jiffies,256*64*64 jiffies,256*64*64*64 jiffies,每組中桶的數量分別爲:256,64,64,64,64,這樣,在 256+64+64+64+64 = 512 個桶中,表示的範圍爲 2^32 。有了這樣的實現,驅動內核定時器的機制也可以通過水錶的例子來理解了,就像水錶,每個粒度上都有一個指針指向當前時間,時間以固定 tick 遞增,而當前時間指針則也依次遞增,如果發現當前指針的位置可以確定爲一個註冊的定時器,就觸發其註冊的回調函數。 Linux 內核定時器本質上是 Single-Shot Timer,如果想成爲 Repeating Timer,可以在註冊的回調函數中再次的註冊自己。內核定時器如下圖 5:

圖 5. linux時間輪

Linux時間輪

————————————————————————————————————————————————————————————————————————————

結論

由上面的分析,可以看到各種定時器實現算法的複雜度:

表 1. 定時器實現的算法複雜度

實現方式 StartTimer StopTimer PerTickBookkeeping
基於鏈表 O(1) O(n) O(n)
基於排序鏈表 O(n) O(1) O(1)
基於最小堆 O(lgn) O(1) O(1)
基於時間輪 O(1) O(1) O(1)

如果需要能在線程環境中使用的定時器,對於基於鏈表的定時器,可能需要很小心的處理信號的問題;而 POSIX timer [ 2 ]接口的定時器,只具有進程的語義,如果想在多線程環境下也 n 能使用,可以使用 Linux 提供的 timerfd_create(2) 接口。如果需要支持的定時器數量非常的大,可以考慮使用基於最小堆和時間輪的方式來實現。

————————————————————————————————————————————————————————————————————————————

Linux 下定時器的實現方式分析

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章