pthread_setcancelstate概述

隨着多線程編程的普及，如何管理線程的狀態和組織線程成為編程難點之一。而pthread_setcancelstate函數的作用是設置線程被取消的狀態。

一、pthread_setcancelstate的基本用法

#include <pthread.h>

int pthread_setcancelstate(int state, int *oldstate);

其中state參數為期望的線程取消狀態，可選值為：

• PTHREAD_CANCEL_ENABLE：允許線程被取消
• PTHREAD_CANCEL_DISABLE：禁止線程被取消

oldstate參數為之前的線程取消狀態，如果此參數為非NULL，則之前的狀態會被保存在oldstate所指向的內存中。

二、pthread_setcancelstate的注意事項

1、pthread_setcancelstate函數只會影響當前線程的取消狀態，不會影響其他線程的取消狀態。

2、當線程處於被取消狀態時，若需要清理資源等工作，可以通過在線程入口函數中調用pthread_cleanup_push和pthread_cleanup_pop實現。

void cleanup_func(void *arg) {
    // 資源清理等工作
}

void *thread_func(void *arg) {
    pthread_cleanup_push(cleanup_func, NULL);
    // 線程主邏輯
    pthread_cleanup_pop(0);
}

其中cleanup_func是需要調用的清理函數，arg為傳入清理函數的參數。

三、pthread_setcancelstate的適用場景

當線程需要長時間運行或需要處理耗時工作時，建議設置為允許被取消狀態，以防止其它線程長時間等待它的完成。而當數據不穩定或處理不可重複時，建議採用禁止被取消狀態，以保證程序不發生不可預期的錯誤。

四、pthread_setcancelstate的錯誤處理

1、如果state參數不是PTHREAD_CANCEL_ENABLE或PTHREAD_CANCEL_DISABLE，函數會返回EINVAL錯誤。

2、如果oldstate參數是NULL，函數會忽略之前的狀態，不會返回錯誤。

3、如果調用pthread_setcancelstate函數失敗，可以通過errno全局變量獲取錯誤號。

五、示例代碼

#include <stdio.h>
#include <pthread.h>

volatile int is_running = 1;

void *thread_func(void *arg) {
    printf("Thread started\n");
    int oldstate;
    pthread_setcancelstate(PTHREAD_CANCEL_DISABLE, &oldstate);
    
    while (is_running) {
        // 長時間運行的邏輯
    }
    
    printf("Thread finished\n");
    pthread_cleanup_push(free, arg);
    pthread_cleanup_pop(0);
}

int main() {
    pthread_t tid;
    if (pthread_create(&tid, NULL, thread_func, NULL) < 0) {
        printf("Failed to create thread\n");
        return -1;
    }
    
    printf("Waiting for thread to finish\n");
    sleep(1);
    
    pthread_cancel(tid);
    is_running = 0;
    
    if (pthread_join(tid, NULL) < 0) {
        printf("Failed to join thread\n");
        return -1;
    }
    
    printf("Thread joined\n");
    return 0;
}

該代碼演示了如何使用pthread_setcancelstate函數設置線程的取消狀態。其中，在主線程中通過pthread_cancel函數取消線程，並通過設置is_running變量為0讓線程退出。