漫话Redis源码之一百一二十

涛歌依旧

已于 2022-06-19 23:23:54 修改

阅读量294

点赞数 1

分类专栏： S1: C/C++ 文章标签： c++

于 2022-06-19 23:16:48 首次发布

本文链接：https://blog.youkuaiyun.com/stpeace/article/details/125364082

版权

S1: C/C++ 专栏收录该内容

972 篇文章

订阅专栏

本文详细介绍了如何在aeEventLoop中使用mask进行fd的事件关联，通过aeApiAssociate函数实现读写操作的监控，并展示了如何在aeApiAddEvent中整合既有事件。重点在于理解事件驱动编程中fd的管理和事件处理策略。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

掩码的使用，还真得会：

/*
 * Helper function to invoke port_associate for the given fd and mask.
 */
static int aeApiAssociate(const char *where, int portfd, int fd, int mask) {
    int events = 0;
    int rv, err;
 
    if (mask & AE_READABLE)
        events |= POLLIN;
    if (mask & AE_WRITABLE)
        events |= POLLOUT;
 
    if (evport_debug)
        fprintf(stderr, "%s: port_associate(%d, 0x%x) = ", where, fd, events);
 
    rv = port_associate(portfd, PORT_SOURCE_FD, fd, events,
        (void *)(uintptr_t)mask);
    err = errno;
 
    if (evport_debug)
        fprintf(stderr, "%d (%s)\n", rv, rv == 0 ? "no error" : strerror(err));
 
    if (rv == -1) {
        fprintf(stderr, "%s: port_associate: %s\n", where, strerror(err));
 
        if (err == EAGAIN)
            fprintf(stderr, "aeApiAssociate: event port limit exceeded.");
    }
 
    return rv;
}
 
static int aeApiAddEvent(aeEventLoop *eventLoop, int fd, int mask) {
    aeApiState *state = eventLoop->apidata;
    int fullmask, pfd;
 
    if (evport_debug)
        fprintf(stderr, "aeApiAddEvent: fd %d mask 0x%x\n", fd, mask);
 
    /*
     * Since port_associate's "events" argument replaces any existing events, we
     * must be sure to include whatever events are already associated when
     * we call port_associate() again.
     */
    fullmask = mask | eventLoop->events[fd].mask;
    pfd = aeApiLookupPending(state, fd);
 
    if (pfd != -1) {
        /*
         * This fd was recently returned from aeApiPoll.  It should be safe to
         * assume that the consumer has processed that poll event, but we play
         * it safer by simply updating pending_mask.  The fd will be
         * re-associated as usual when aeApiPoll is called again.
         */
        if (evport_debug)
            fprintf(stderr, "aeApiAddEvent: adding to pending fd %d\n", fd);
        state->pending_masks[pfd] |= fullmask;
        return 0;
    }
 
    return (aeApiAssociate("aeApiAddEvent", state->portfd, fd, fullmask));
}