1. 程式人生 > >linux下執行緒的一次性初始化

linux下執行緒的一次性初始化

一、為什麼要使用一次性初始化
    有些事需要且只能執行一次(比如互斥量初始化)。通常當初始化應用程式時,可以比較容易地將其放在main函式中。但當你寫一個庫函式時,就不能在main裡面初始化了,你可以用靜態初始化,但使用一次初始(pthread_once_t)會比較容易些。

二、如何進行一次性初始化
    1、首先要定義一個pthread_once_t變數,這個變數要用巨集PTHREAD_ONCE_INIT初始化。然後建立一個與控制變數相關的初始化函式:

    pthread_once_t once_control = PTHREAD_ONCE_INIT;
    void init_routine()
    {
         //初始化互斥量
         //初始化讀寫鎖
         ......
    }


    2、接下來就可以在任何時刻呼叫pthread_once函式
    int pthread_once(pthread_once_t* once_control, void (*init_routine)(void));
    功能:本函式使用初值為PTHREAD_ONCE_INIT的once_control變數保證init_routine()函式在本程序執行序列中僅執行一次。在多執行緒程式設計環境下,儘管pthread_once()呼叫會出現在多個執行緒中,init_routine()函式僅執行一次,究竟在哪個執行緒中執行是不定的,是由核心排程來決定。
    3、Linux Threads使用互斥鎖和條件變數保證由pthread_once()指定的函式執行且僅執行一次。實際"一次性函式"的執行狀態有三種:
        NEVER(0)、IN_PROGRESS(1)、DONE (2),用once_control來表示pthread_once()的執行狀態:
     1)、如果once_control初值為0,那麼 pthread_once從未執行過,init_routine()函式會執行。
    2)、如果once_control初值設為1,則由於所有pthread_once()都必須等待其中一個激發"已執行一次"訊號, 因此所有pthread_once ()都會陷入永久 的等待中,init_routine()就無法執行
    3)、如果once_control設為2,則表示pthread_once()函式已執行過一次,從而所有pthread_once()都會立即   返回,init_routine()就沒有機會執行,當pthread_once函式成功返回,once_control就會被設定為2。四、例項
1、一次性初始化的驗證

/*DATE:            2015-4-15
 *AUTHOR:        DDDDD
 *DESCRIPTION:    一次性初始化
    int pthread_once(pthread_once_t* once_control, void (*init_routine)(void));
    如果once_control為0,init_routine()就會執行
    pthread_once()成功返回之後,once_control會變為2
 */

#include "apue.h"

pthread_once_t once = 2;
pthread_t tid;

void thread_init()
{
    printf("I'm in thread 0x%x\n", tid);

}


void *thread_fun1(void *arg)
{
    tid = pthread_self();
    printf("I'm thread 0x%x\n", tid);
    printf("once is %d\n", once);
    pthread_once(&once, thread_init);
    printf("once is %d\n", once);

    return NULL;
}

void *thread_fun2(void *arg)
{
    sleep(2);
    tid = pthread_self();
    printf("I'm thread 0x%x\n", tid);
    pthread_once(&once, thread_init);

    return NULL;
}

int main()
{
    pthread_t tid1, tid2;
    int err;

    err = pthread_create(&tid1, NULL, thread_fun1, NULL);
    if(err != 0)
    {
        printf("create new thread 1 failed\n");
        return ;
    }
    err = pthread_create(&tid2, NULL, thread_fun2, NULL);
    if(err != 0)
    {
        printf("create new thread 1 failed\n");
        return ;
    }


    pthread_join(tid1, NULL);
    pthread_join(tid2, NULL);

    return 0;
}

2、將互斥量的初始化,使用pthread_once來實現
 

/*DATA:            2015-4-20
 *AUTHOR;        WJ
 *DESCRIPTION:    使用多執行緒對一個佇列進行增加和減少,增加操作是一個執行緒,刪除操作是一個執行緒
 *    
 */
#include "apue.h"

pthread_mutex_t mutex;
pthread_once_t once = PTHREAD_ONCE_INIT;

struct queue{
    int len;
    int write_pos;
    int read_pos;
    int data[50];
};

//互斥量初始化函式
void mutex_init()
{
    int err;
    err = pthread_mutex_init(&mutex, NULL);
    if(err)
    {
        printf("mutex init failed\n");
        return;
    }
}

//佇列初始化
struct queue *queue_init()
{
    struct queue *que;
    //申請記憶體
    que = (struct queue *)malloc(sizeof(struct queue));
    if(que ==NULL)
    {
        printf("malloc failed\n");
        return;
    }

    //初始化
    que->len = 0;
    que->write_pos = 0;
    que->read_pos = 0;

    return que;
}

void queue_destroy(struct queue *que)
{
    //銷燬互斥量和que
    pthread_mutex_destroy(&mutex);
    free(que);
}

void *queue_add(void *arg)
{
    //對互斥量進行一次性初始化
    pthread_once(&once, mutex_init);
    struct queue *que = (struct queue *)arg;
    int buf=0;
    while(buf<50)
    {
        pthread_mutex_lock(&mutex);
        que->data[que->write_pos] = buf;
        que->write_pos ++;
        que->len ++;
        buf++;
        printf("write data %d to queue\n", que->data[que->write_pos -1]);

        pthread_mutex_unlock(&mutex);
        sleep(1);
    }
}

void *queue_del(void *arg)
{
    //    對互斥量進行一次性初始化
    pthread_once(&once, mutex_init);
    struct queue *que = (struct queue *)arg;
    int buf=0;
    while(1)
    {
        sleep(2);
        pthread_mutex_lock(&mutex);
        buf = que->data[que->read_pos];
        que->read_pos ++;
        if(que->len -- == 0)
        {
            printf("queue is empty\n");
            return;
        }
        buf++;
        printf("read data %d from queue\n", que->data[que->read_pos -1]);
        pthread_mutex_unlock(&mutex);
    }
}

int main()
{
    pthread_t tid1, tid2;
    int err;
    struct queue *que;

    //佇列初始化
    que = queue_init();

    err = pthread_create(&tid1, NULL, queue_add, (void *)que);
    if(err)
    {
        printf("create add thread failed\n");
        queue_destroy(que);
        return;
    }

    err = pthread_create(&tid2, NULL, queue_del, (void *)que);
    if(err)
    {
        printf("create del thread failed\n");
        queue_destroy(que);
        return;
    }

    //等待增加和刪除操作完成
    pthread_join(tid1, NULL);
    pthread_join(tid2, NULL);

    //銷燬
    queue_destroy(que);
}