sem_post 不优先在其他线程中调用 sem_wait

Question

我正在处理我的项目，正确清理所有内容并将所有缓冲的日志消息保存到文件等是至关重要的。我正在从另一个线程调用退出并且我正在考虑使用信号量来在程序完全退出之前等待主线程中的清理发生。问题是，当我从使用 atexit 注册的退出处理程序调用 sem_post 时，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 证明了我的问题。这个问题有什么解决办法吗？我不能把清理放在我的退出处理程序中，因为我在 main 函数中有很多本地资源需要在我的实际程序中清理。

Answer 1

您无法控制在调用 sem_post() 后哪个线程将从 sem_wait() return。您需要使用两个信号量：

#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
}

sem_post 不优先在其他线程中调用 sem_wait

sem_post does not prioritise sem_wait call in other thread

c

multithreading

semaphore

pthreads

exit