用于轮询工作线程任务完成状态的 C 文件描述符

Question

我有一个 C 项目，它有 2 个线程，main_thread 和 worker_thread。下面的伪代码对此进行了解释。

TASK g_task[MAX_TASK];
worker_thread()
{
    do {
        listen_for_task on g_task
        do_task(0);
    } while(0);
}

int post_task(msg)
{
    post on g_task
    /* I am planning for a kind of file descriptor to return from here */
    /* on which main thread polls along with sock_fd */
    return fd;
}

main_thread()
{
    int sock_fd;

    /* sock_fd listens on well known TCP port for msgs from clients */
    /* Using epoll */
    ev.events = EPOLIN;
    ev.data.fd = sock_fd;
    epoll_ctl(epollfd, EPOLL_CTL_ADD, sock_fd, &ev);
    do {
        nfs = epoll_wait(...)
        if (nfs > 0) {
            /* I read the msg from from sock_fd and post to g_task */
            post_task(msg)
        }
    } while (0);
}

On main_thread 我在一个 tcp 套接字上等待来自不同客户端的消息。我正在处理并将要完成的工作发布到另一个线程 worker_thread。

现在我希望 worker_thread 在完成发布给它的每个任务后通知 main_thread。我期待事件应该来到一个新的文件描述符。这意味着我期望 post_task 应该 return 一个我可以添加到 epollfd 的新 fd。 worker_thread 应该唤醒它。

实现此目标的任何建议是？

Answer 1

既然你提到使用epoll，那你一定是在使用Linux。这意味着您可以使用 eventfd().

基本上，您的 post_task 函数会为每个线程创建一个新的 eventfd 描述符，并且主线程会监视它以确保 epoll 中的可读性。然后线程在完成后将一个 64 位整数写入其描述符。然后主线程从 epoll 中唤醒，看到事件是可读的，从中读取一个 64 位整数，并在知道相关线程已完成任务后做任何它需要做的事情。（在非 linux 系统上，您可以使用 pipe() 获得类似的效果；我编写了带有事件创建函数的代码，该函数根据 OS 的目标在两者之间切换）

如果您需要传递的信息不仅仅是 "I'm done" 状态，我喜欢使用使用 socketpair() 和 reading/writing 结构创建的 Unix 域数据报套接字，其中包含所有需要的数据。如果包含线程 ID，所有线程甚至可以共享同一个套接字。

用于轮询工作线程任务完成状态的 C 文件描述符

C File Descriptor for polling task completion status of worker thread

c

events

multithreading

epoll

polling