4 连接接口设计-CSDN博客

4 连接接口设计

1 数据结构

1.1 连接实体 `connections_s`

struct connections_s {
    event_t *read;
    event_t *write;

    ukey_pool_t *m_cpool;
};

说明：
每个连接有一个内存块，并且连接分为读事件、写事件。

2 函数

2.1 `conn_init`

函数签名：
void conn_init(connections_t *c, int size)

函数实体：

void conn_init(connections_t *c, int size) {    
    int i;
    for (i = 0; i < size; i++) {
        c[i].m_cpool = ukey_create_pool(1024);
        c[i].read = NULL;
        c[i].write = NULL;
    }
}

说明：

初始化每个连接，给每个连接申请内存区域，初始化读事件、写事件

2.2 `conn_set`

函数签名：
int conn_set(int fd, int events, void (call_back)(int, int, void ))

函数实体：

int conn_set(int fd, int events, void (*call_back)(int, int, void *)) {
    if (fd > g_manager.nconnections) {
        fprintf(stderr, "fd max limits\n");
        return -1;
    }

    connections_t *new_conn = &g_manager.conn[fd];
    if (new_conn == NULL) {
        fprintf(stderr, "conn set error\n");
        return -1;
    }
    event_t *new_ev = (event_t *)ukey_palloc(new_conn->m_cpool, sizeof(event_t));
    if (new_ev == NULL) {
        fprintf(stderr, "alloc from coon m_cpoll error\n");
        return -1;
    }
    void *arg = (void *)new_ev;
    event_set(new_ev, fd, call_back, arg);
    event_add(g_manager.ep_fd, events, new_ev, 1);
    if (events & EPOLLIN) {
        new_conn->read = new_ev;
    }
    if (events & EPOLLOUT) {
        new_conn->write = new_ev;
    }
    
    return 0;
}

说明：

根据传进的 fd ,找到对应的连接，设置连接的事件，并且注册回调函数

2.3 `conn_free`

函数签名：
void conn_free(int fd)

函数实体：

void conn_free(int fd) {
    
    if (fd == -1)
        return;
    connections_t *c = &g_manager.conn[fd];
    if (c == NULL)
        return ;
    if (c->read != NULL) {
        event_del(g_manager.ep_fd, c->read);
        event_set(c->read, -1, NULL, 0);
        
    } else if (c->write != NULL) {
        event_del(g_manager.ep_fd, c->write);
        event_set(c->write, -1, NULL, 0);
    }
}

说明：

根据传进的描述符，删除连接对应的事件，并不回收资源，复用描述符

转载于:https://www.cnblogs.com/eemjwu/p/10609640.html