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);
}
}
说明:
根据传进的描述符,删除连接对应的事件,并不回收资源,复用描述符