sem_post沒有優先考慮在其他線程sem_wait調用

Question

我正在我的項目中，它是至關重要的，一切正常清理，所有緩衝的日誌消息保存到一個文件等..我打電話退出另一個線程和我正在考慮在程序完全退出之前使用信號量等待主線程中的清理。問題是，當我從退出處理程序調用sem_post並使用atexit註冊時，sem_wait不會立即減少信號量，因此，退出處理程序中的sem_wait將立即退出，因爲信號量大於零。

#include <stdio.h> 
#include <stdlib.h> 
#include <unistd.h> 
#include <semaphore.h> 
#include <pthread.h> 

static sem_t sem; 

void *do_work(void *arg) { 
    // if something fails 
    printf("Something failed!\n"); 
    exit(1); 
} 

void exit_handler() { 
    sem_post(&sem); // wake up main thread 
    sem_wait(&sem); // wait for cleanup in main 
    sem_destroy(&sem); 
} 

int main() { 
    pthread_t worker_thread; 

    sem_init(&sem, 0, 0); 
    atexit(exit_handler); 

    pthread_create(&worker_thread, NULL, do_work, NULL); 

    sem_wait(&sem); // block this thread until work is done 

    // simulate some cleanup 
    usleep(1000000); 
    printf("This never gets called!\n"); 

    sem_post(&sem); // destroy semaphore 
} 

這個example演示我的問題。有沒有解決這個問題的方法？我不能把清理放到我的退出處理程序中，因爲我的主函數中有很多本地資源需要在我的實際程序中清理。

Answer 1

在調用sem_post()之後，您無法控制哪個線程將從sem_wait()返回。你需要使用兩個信號量：

#include <stdio.h> 
#include <stdlib.h> 
#include <unistd.h> 
#include <semaphore.h> 
#include <pthread.h> 

static sem_t wakeupMain; 
static sem_t cleanupDone; 

void *do_work(void *arg) { 
    // if something fails 
    printf("Something failed!\n"); 
    exit(1); 
} 

void exit_handler() { 
    sem_post(&wakeupMain); // wake up main thread 
    sem_wait(&cleanupDone); // wait for cleanup in main 
    sem_destroy(&wakeupMain); 
    sem_destroy(&cleanupDone); 
} 

int main() { 
    pthread_t worker_thread; 

    sem_init(&wakeupMain, 0, 0); 
    sem_init(&cleanupDone, 0, 0); 
    atexit(exit_handler); 

    pthread_create(&worker_thread, NULL, do_work, NULL); 

    sem_wait(&wakeupMain); // block this thread until work is done 

    // simulate some cleanup 
    usleep(1000000); 
    printf("This never gets called!\n"); 

    sem_post(&cleanupDone); // destroy semaphore 
}