pdxx
/
libfastcommon
mirror of https://github.com/happyfish100/libfastcommon.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

set extra_params for socket gracefully

use_iouring
YuQing 2025-01-26 13:02:07 +08:00
parent 5477593ce8
commit e4a9fccddb
1 changed files with 11 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
src/connection_pool.c
View File

@ -166,7 +166,7 @@ static ConnectionManager *find_manager(ConnectionPool *cp,
memcpy(cm->key.str, key->str, key->len + 1); memcpy(cm->key.str, key->str, key->len + 1);
cm->key.len = key->len; cm->key.len = key->len;
//add to chain //add to manager chain
cm->next = bucket->head; cm->next = bucket->head;
bucket->head = cm; bucket->head = cm;
return cm; return cm;
@ -297,17 +297,21 @@ int conn_pool_init_ex1(ConnectionPool *cp, const int connect_timeout,
return result; return result;
} }
if (extra_params != NULL) { if (extra_params != NULL && extra_params->rdma.pd != NULL) {
extra_connection_size = G_RDMA_CONNECTION_CALLBACKS. extra_connection_size = G_RDMA_CONNECTION_CALLBACKS.
get_connection_size(); get_connection_size();
obj_init_func = (fast_mblock_object_init_func)node_init_for_rdma; obj_init_func = (fast_mblock_object_init_func)node_init_for_rdma;
cp->extra_params = *extra_params; cp->extra_params = *extra_params;
} else { } else {
extra_connection_size = 0; extra_connection_size = 0;
cp->extra_params.tls.enabled = false; if (extra_params != NULL) {
cp->extra_params.tls.htable_capacity = 163; cp->extra_params = *extra_params;
cp->extra_params.rdma.buffer_size = 0; } else {
cp->extra_params.rdma.pd = NULL; cp->extra_params.tls.enabled = false;
cp->extra_params.tls.htable_capacity = 163;
cp->extra_params.rdma.buffer_size = 0;
cp->extra_params.rdma.pd = NULL;
}
obj_init_func = (fast_mblock_object_init_func)node_init_for_socket; obj_init_func = (fast_mblock_object_init_func)node_init_for_socket;
} }
if ((result=fast_mblock_init_ex1(&cp->node_allocator, "cpool-node", if ((result=fast_mblock_init_ex1(&cp->node_allocator, "cpool-node",
@ -730,6 +734,7 @@ ConnectionInfo *conn_pool_get_connection_ex(ConnectionPool *cp,
return NULL; return NULL;
} }
//add to thread local hashtable
node = (ConnectionNode *)((char *)ci - sizeof(ConnectionNode)); node = (ConnectionNode *)((char *)ci - sizeof(ConnectionNode));
node->next = *bucket; node->next = *bucket;
*bucket = node; *bucket = node;