2014年7月11日 星期五

[轉] Linux中的工作隊列

工作隊列(work queue)Linux kernel中將工作推後執行的一種機制。這種機制和BHTasklets不同之處在於工作隊列是把推後的工作交由一個內核線程去執行,因此工作隊列的優勢就在於它允許重新調度甚至睡眠。
工作隊列是2.6內核開始引入的機制,在2.6.20之後,工作隊列的數據結構發生了一些變化,因此本文分成兩個部分對2.6.20之前和之後的版本分別做介紹。

I2.6.0~2.6.19


數據結構:
1
2
3
4
5
6
7
8
struct work_struct {
    unsigned long pending;
    struct list_head entry;
    void (*func)(void *);
    void *data;
    void *wq_data;
    struct timer_list timer;
};
pending是用來記錄工作是否已經掛在隊列上;
entry是循環鏈表結構;
func作為函數指針,由用戶實現;
data用來存儲用戶的私人數據,此數據即是func的參數;
wq_data一般用來指向工作者線程(工作者線程參考下文);
timer是推後執行的定時器。
work_struct的這些變量裡,funcdata是用戶使用的,其他是內部變量,我們可以不用太過關心。

API
1
2
3
4
5
INIT_WORK(_work, _func, _data);
int schedule_work(struct work_struct *work);
int schedule_delayed_work(struct work_struct *work, unsigned long delay);
void flush_scheduled_work(void);
int cancel_delayed_work(struct work_struct *work);
1、初始化指定工作,目的是把用戶指定的函數_func_func需要的參數_data賦給work_structfuncdata變量。
2、對工作進行調度,即把給定工作的處理函數提交給缺省的工作隊列和工作者線程。工作者線程本質上是一個普通的內核線程,在默認情況下,每個CPU均有一個類型為「events」的工作者線程,當調用schedule_work時,這個工作者線程會被喚醒去執行工作鏈表上的所有工作。
3、延遲執行工作,與schedule_work類似。
4、刷新缺省工作隊列。此函數會一直等待,直到隊列中的所有工作都被執行。
5、flush_scheduled_work並不取消任何延遲執行的工作,因此,如果要取消延遲工作,應該調用cancel_delayed_work

以上均是採用缺省工作者線程來實現工作隊列,其優點是簡單易用,缺點是如果缺省工作隊列負載太重,執行效率會很低,這就需要我們創建自己的工作者線程和工作隊列。
API
1
2
3
4
5
struct workqueue_struct *create_workqueue(const char *name);
int queue_work(struct workqueue_struct *wq, struct work_struct *work);
int queue_delayed_work(struct workqueue_struct *wq, struct work_struct *work, unsigned long delay);
void flush_workqueue(struct workqueue_struct *wq);
void destroy_workqueue(struct workqueue_struct *wq);
1、創建新的工作隊列和相應的工作者線程,name用於該內核線程的命名。
2、類似於schedule_work,區別在於queue_work把給定工作提交給創建的工作隊列wq而不是缺省隊列。
3、延遲執行工作。
4、刷新指定工作隊列。
5、釋放創建的工作隊列。

下面一段代碼可以看作一個簡單的實作:
1
2
3
4
5
6
7
8
9
10
11
12
13
void my_func(void *data)
{
    char *name = (char *)data;
    printk(KERN_INFO 「Hello world, my name is %s!\n」, name);
}
struct workqueue_struct *my_wq = create_workqueue(「my wq」);
struct work_struct my_work;
INIT_WORK(&my_work, my_func, 「Jack」);
queue_work(my_wq, &my_work);
destroy_workqueue(my_wq);


II2.6.20~2.6.??


2.6.20起,工作隊列的數據結構發生了一些變化,使用時不能沿用舊的方法。

數據結構:
1
2
3
4
5
6
7
typedef void (*work_func_t)(struct work_struct *work);
struct work_struct {
    atomic_long_t data;
    struct list_head entry;
    work_func_t func;
};
2.6.19之前的版本相比,work_struct瘦身不少。粗粗一看,entry和之前的版本相同,funcdata發生了變化,另外並無其他的變量。
entry我們不去過問,這個和以前的版本完全相同。data的類型是atomic_long_t,這個類型從字面上看可以知道是一個原子類型。第一次看到這個變量時,很容易誤認為和以前的data是同樣的用法,只不過類型變了而已,其實不然,這裡的data是之前版本的pendingwq_data的複合體,起到了以前的pendingwq_data的作用。
func的參數是一個work_struct指針,指向的數據就是定義funcwork_struct
看到這裡,會有兩個疑問,第一,如何把用戶的數據作為參數傳遞給func呢?以前有void *data來作為參數,現在好像完全沒有辦法做到;第二,如何實現延遲工作?目前版本的work_struct並沒有定義timer

解決第一個問題,需要換一種思路。2.6.20版本之後使用工作隊列需要把work_struct定義在用戶的數據結構中,然後通過container_of來得到用戶數據。具體用法可以參考稍後的實作。

對於第二個問題,新的工作隊列把timer拿掉的用意是使得work_struct更加單純。首先回憶一下之前版本,只有在需要延遲執行工作時才會用到timer,普通情況下timer是沒有意義的,所以之前的做法在一定程度上有些浪費資源。所以新版本中,將timerwork_struct中拿掉,然後又定義了一個新的結構delayed_work用於處理延遲執行:
1
2
3
4
struct delayed_work {
    struct work_struct work;
    struct timer_list timer;
};

下面把API羅列一下,每個函數的解釋可參考之前版本的介紹或者之後的實作:
1
2
3
4
5
6
7
8
9
10
11
INIT_WORK(struct work_struct *work, work_func_t func);
INIT_DELAYED_WORK(struct delayed_work *work, work_func_t func);
int schedule_work(struct work_struct *work);
int schedule_delayed_work(struct delayed_work *work, unsigned long delay);
struct workqueue_struct *create_workqueue(const char *name);
int queue_work(struct workqueue_struct *wq, struct work_struct *work);
int queue_delayed_work(struct workqueue_struct *wq, struct delayed_work *work, unsigned long delay);
void flush_scheduled_work(void);
void flush_workqueue(struct workqueue_struct *wq);
int cancel_delayed_work(struct delayed_work *work);
void destroy_workqueue(struct workqueue_struct *wq);
其中,1、2、4、7和以前略有區別,其他用法完全一樣。

實作:
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
struct my_struct_t {
    char *name;
    struct work_struct my_work;
};
void my_func(struct work_struct *work)
{
    struct my_struct_t *my_name = container_of(work, struct my_struct_t, my_work);
    printk(KERN_INFO 「Hello world, my name is %s!\n」, my_name->name);
}
struct workqueue_struct *my_wq = create_workqueue(「my wq」);
struct my_struct_t my_name;
my_name.name = 「Jack」;
INIT_WORK(&(my_name.my_work), my_func);
queue_work(my_wq, &(my_name.my_work));
destroy_workqueue(my_wq);

沒有留言:

張貼留言